diff options
Diffstat (limited to 'models/setup_for_test.go')
| -rw-r--r-- | models/setup_for_test.go | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/models/setup_for_test.go b/models/setup_for_test.go index 91d5442c84..5a17eac789 100644 --- a/models/setup_for_test.go +++ b/models/setup_for_test.go @@ -107,8 +107,9 @@ func AssertSuccessfulInsert(t *testing.T, beans ...interface{}) { assert.NoError(t, err) } -// AssertSuccessfulUpdate assert that bean is successfully updated -func AssertSuccessfulUpdate(t *testing.T, bean interface{}, conditions ...interface{}) { - _, err := x.Update(bean, conditions...) +// AssertCount assert the count of a bean +func AssertCount(t *testing.T, bean interface{}, expected interface{}) { + actual, err := x.Count(bean) assert.NoError(t, err) + assert.EqualValues(t, expected, actual) } |