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,7 +1080,6 @@ func TestQueryer(t *testing.T) {
} }
defer rows.Close() defer rows.Close()
n := 1 n := 1
if rows != nil {
for rows.Next() { for rows.Next() {
var id int var id int
err = rows.Scan(&id) err = rows.Scan(&id)
@ -1090,7 +1089,10 @@ func TestQueryer(t *testing.T) {
if id != n { if id != n {
t.Error("Failed to db.Query: not matched results") 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)
} }
} }