From b9cea2347cbee9b5af6978d8ad1b6a8c7cb6afd0 Mon Sep 17 00:00:00 2001
From: Deepak <dmallubhotla+github@gmail.com>
Date: Sun, 31 Jan 2021 12:18:58 -0600
Subject: [PATCH] Adds error store tests

---
 store/errorStore_test.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/store/errorStore_test.go b/store/errorStore_test.go
index 71c8d72..6c075e6 100644
--- a/store/errorStore_test.go
+++ b/store/errorStore_test.go
@@ -82,3 +82,25 @@ func TestErrorUserMethods(t *testing.T) {
 	_, err = str.SelectUserByUsername("snth")
 	assert.NotNil(err)
 }
+
+func TestErrorCurrentPlanMethods(t *testing.T) {
+	assert := assert.New(t)
+	str := store.GetErrorStore("error", true)
+	str2 := store.GetErrorStore("error", false)
+
+	cp := &models.CurrentPlan{}
+
+	_, err := str.SelectCurrentPlan(1)
+	assert.NotNil(err)
+
+	err = str.InsertCurrentPlan(cp, 1)
+	assert.NotNil(err)
+	err = str2.InsertCurrentPlan(cp, 1)
+	assert.Nil(err)
+
+	replace := &models.CurrentPlan{}
+	err = str.UpdateCurrentPlan(replace, 1)
+	assert.NotNil(err)
+	err = str2.UpdateCurrentPlan(replace, 1)
+	assert.Nil(err)
+}