86 lines
1.8 KiB
Go
86 lines
1.8 KiB
Go
package models_test
|
|
|
|
import (
|
|
"gitea.deepak.science/deepak/gogmagog/models"
|
|
"gitea.deepak.science/deepak/gogmagog/store"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestModelActions(t *testing.T) {
|
|
assert := assert.New(t)
|
|
a1 := &models.Action{ActionID: 3}
|
|
a2 := &models.Action{ActionID: 4}
|
|
userID := 3
|
|
p := &models.Plan{PlanID: 6}
|
|
|
|
str, _ := store.GetInMemoryStore()
|
|
str.InsertAction(a1, userID)
|
|
str.InsertPlan(p, userID)
|
|
m := models.New(str)
|
|
|
|
actions, err := m.Actions(userID)
|
|
assert.Nil(err)
|
|
assert.Equal(1, len(actions))
|
|
|
|
firstAction, err := m.Action(1, userID)
|
|
assert.Nil(err)
|
|
assert.EqualValues(1, firstAction.ActionID)
|
|
|
|
actionID, err := m.AddAction(a2, userID)
|
|
assert.Nil(err)
|
|
assert.EqualValues(2, actionID)
|
|
|
|
err = m.SaveAction(a1, userID)
|
|
assert.Nil(err)
|
|
}
|
|
|
|
func TestModelPlanMethods(t *testing.T) {
|
|
assert := assert.New(t)
|
|
userID := 3
|
|
a1 := &models.Action{ActionID: 3, PlanID: 1}
|
|
a2 := &models.Action{ActionID: 4}
|
|
p := &models.Plan{}
|
|
|
|
str, _ := store.GetInMemoryStore()
|
|
str.InsertPlan(p, userID)
|
|
str.InsertAction(a1, userID)
|
|
str.InsertAction(a2, userID)
|
|
m := models.New(str)
|
|
|
|
plans, err := m.Plans(userID)
|
|
assert.Nil(err)
|
|
assert.Equal(1, len(plans))
|
|
|
|
firstPlan, err := m.Plan(1, userID)
|
|
assert.Nil(err)
|
|
assert.EqualValues(1, firstPlan.PlanID)
|
|
|
|
actions, err := m.GetActions(firstPlan, userID)
|
|
assert.Nil(err)
|
|
assert.Equal(1, len(actions))
|
|
|
|
planId, err := m.AddPlan(&models.Plan{}, userID)
|
|
assert.Nil(err)
|
|
assert.EqualValues(2, planId)
|
|
}
|
|
|
|
func TestModelHealthy(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
str, _ := store.GetInMemoryStore()
|
|
m := models.New(str)
|
|
|
|
err := m.Healthy()
|
|
assert.Nil(err)
|
|
}
|
|
|
|
func TestNilModelUnhealthy(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
m := models.New(nil)
|
|
|
|
err := m.Healthy()
|
|
assert.NotNil(err)
|
|
}
|