add more tests for rows

This commit is contained in:
Lunny Xiao 2017-07-27 14:30:01 +08:00
parent 8849fd919e
commit d04a1a2e3d
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 27 additions and 0 deletions

View File

@ -38,4 +38,31 @@ func TestRows(t *testing.T) {
cnt++
}
assert.EqualValues(t, 1, cnt)
sess := testEngine.NewSession()
defer sess.Close()
rows1, err := sess.Prepare().Rows(new(UserRows))
assert.NoError(t, err)
defer rows1.Close()
cnt = 0
for rows1.Next() {
err = rows1.Scan(user)
assert.NoError(t, err)
cnt++
}
assert.EqualValues(t, 1, cnt)
rows2, err := testEngine.SQL("SELECT * FROM user_rows").Rows(new(UserRows))
assert.NoError(t, err)
defer rows2.Close()
cnt = 0
for rows2.Next() {
err = rows2.Scan(user)
assert.NoError(t, err)
cnt++
}
assert.EqualValues(t, 1, cnt)
}