107 lines
2.5 KiB
Go
107 lines
2.5 KiB
Go
package store_test
|
|
|
|
import (
|
|
"fmt"
|
|
"gitea.deepak.science/deepak/gogmagog/models"
|
|
"gitea.deepak.science/deepak/gogmagog/store"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestErrorActionMethods(t *testing.T) {
|
|
assert := assert.New(t)
|
|
userID := 2
|
|
str := store.GetErrorStore("error message sample", true)
|
|
str2 := store.GetErrorStore("error message sample", false)
|
|
str3 := store.GetErrorStoreForError(fmt.Errorf("test error"), false)
|
|
|
|
_, err := str.InsertAction(&models.Action{}, userID)
|
|
assert.NotNil(err)
|
|
_, err = str2.InsertAction(&models.Action{}, userID)
|
|
assert.Nil(err)
|
|
_, err = str3.InsertAction(&models.Action{}, userID)
|
|
assert.Nil(err)
|
|
|
|
_, err = str.SelectActionByID(8, userID)
|
|
assert.NotNil(err)
|
|
_, err = str2.SelectActionByID(8, userID)
|
|
assert.NotNil(err)
|
|
|
|
_, err = str.SelectActions(userID)
|
|
assert.NotNil(err)
|
|
|
|
_, err = str.SelectActionsByPlanID(&models.Plan{}, userID)
|
|
assert.NotNil(err)
|
|
|
|
replacementAction := &models.Action{}
|
|
err = str.UpdateAction(replacementAction, userID)
|
|
assert.NotNil(err)
|
|
err = str2.UpdateAction(replacementAction, userID)
|
|
assert.Nil(err)
|
|
|
|
}
|
|
|
|
func TestErrorPlanMethods(t *testing.T) {
|
|
assert := assert.New(t)
|
|
str := store.GetErrorStore("sntahoeu", true)
|
|
str2 := store.GetErrorStore("sntahoeu", false)
|
|
|
|
_, err := str.SelectPlans(3)
|
|
assert.NotNil(err)
|
|
|
|
_, err = str.InsertPlan(&models.Plan{}, 3)
|
|
assert.NotNil(err)
|
|
_, err = str2.InsertPlan(&models.Plan{}, 3)
|
|
assert.Nil(err)
|
|
|
|
_, err = str.SelectPlanByID(5, 3)
|
|
assert.NotNil(err)
|
|
|
|
}
|
|
|
|
func TestErrorLive(t *testing.T) {
|
|
assert := assert.New(t)
|
|
str := store.GetErrorStore("error", true)
|
|
|
|
err := str.ConnectionLive()
|
|
assert.NotNil(err)
|
|
}
|
|
|
|
func TestErrorUserMethods(t *testing.T) {
|
|
assert := assert.New(t)
|
|
str := store.GetErrorStore("error", true)
|
|
str2 := store.GetErrorStore("error", false)
|
|
|
|
u := &models.User{}
|
|
|
|
_, err := str.InsertUser(u)
|
|
assert.NotNil(err)
|
|
_, err = str2.InsertUser(u)
|
|
assert.Nil(err)
|
|
|
|
_, 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)
|
|
}
|