TestQueryer: actually check Rows returned

Fixes a test which did not correctly exercise the multi-statement
Queryer functionality
This commit is contained in:
Oliver Giles 2022-07-01 11:41:50 +12:00 committed by mattn
parent a2e94c9d58
commit f1eef49b3f
1 changed files with 12 additions and 10 deletions

View File

@ -1080,17 +1080,19 @@ func TestQueryer(t *testing.T) {
}
defer rows.Close()
n := 1
if rows != nil {
for rows.Next() {
var id int
err = rows.Scan(&id)
if err != nil {
t.Error("Failed to db.Query:", err)
}
if id != n {
t.Error("Failed to db.Query: not matched results")
}
for rows.Next() {
var id int
err = rows.Scan(&id)
if err != nil {
t.Error("Failed to db.Query:", err)
}
if id != n {
t.Error("Failed to db.Query: not matched results")
}
n = n + 1
}
if n != 3 {
t.Errorf("Expected 3 rows but retrieved %v", n-1)
}
}