add more tests for rows (#664)
This commit is contained in:
parent
8849fd919e
commit
60c6346a77
27
rows_test.go
27
rows_test.go
|
@ -38,4 +38,31 @@ func TestRows(t *testing.T) {
|
||||||
cnt++
|
cnt++
|
||||||
}
|
}
|
||||||
assert.EqualValues(t, 1, 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)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue