|
|
|
@ -96,7 +96,15 @@ func AssertExistsAndLoadMap(t assert.TestingT, table string, conditions ...any)
|
|
|
|
|
// GetCount get the count of a bean
|
|
|
|
|
func GetCount(t assert.TestingT, bean any, conditions ...any) int {
|
|
|
|
|
e := db.GetEngine(db.DefaultContext)
|
|
|
|
|
count, err := whereOrderConditions(e, conditions).Count(bean)
|
|
|
|
|
for _, condition := range conditions {
|
|
|
|
|
switch cond := condition.(type) {
|
|
|
|
|
case *testCond:
|
|
|
|
|
e = e.Where(cond.query, cond.args...)
|
|
|
|
|
default:
|
|
|
|
|
e = e.Where(cond)
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
count, err := e.Count(bean)
|
|
|
|
|
assert.NoError(t, err)
|
|
|
|
|
return int(count)
|
|
|
|
|
}
|
|
|
|
|