diff --git a/models/status_test.go b/models/status_test.go new file mode 100644 index 0000000..eb44802 --- /dev/null +++ b/models/status_test.go @@ -0,0 +1,31 @@ +package models + +import ( + "testing" + "github.com/stretchr/testify/assert" +) + +func TestGetStatusList(t *testing.T) { + // Create test database + assert.NoError(t, PrepareTestDatabase()) + + // Insert some dummy data + _, err := x.Insert(Status{Name: "new"}) + assert.NoError(t, err) + _, err = x.Insert(Status{Name: "used"}) + assert.NoError(t, err) + _, err = x.Insert(Status{Name: "other"}) + assert.NoError(t, err) + + // Get a status list + list, err := GetStatusList() + assert.NoError(t, err) + assert.Equal(t, "new", list[0].Name) + assert.Equal(t, "used", list[1].Name) + assert.Equal(t, "other", list[2].Name) + + // Get a status by its ID + status, err := GetStatusByID(1) + assert.NoError(t, err) + assert.Equal(t, "new", status.Name) +}