Fix TestQueryer test to use exec for multistatement insertion

This commit is contained in:
Joshua Hull 2022-08-29 17:47:32 +02:00 committed by mattn
parent d5355d86f9
commit f92b6bb2a1
1 changed files with 18 additions and 10 deletions

View File

@ -1069,30 +1069,38 @@ func TestQueryer(t *testing.T) {
t.Error("Failed to call db.Query:", err) t.Error("Failed to call db.Query:", err)
} }
rows, err := db.Query(` _, err = db.Exec(`
insert into foo(id) values(?); insert into foo(id) values(?);
insert into foo(id) values(?); insert into foo(id) values(?);
insert into foo(id) values(?); insert into foo(id) values(?);
select id from foo order by id;
`, 3, 2, 1) `, 3, 2, 1)
if err != nil {
t.Error("Failed to call db.Exec:", err)
}
rows, err := db.Query(`
select id from foo order by id;
`)
if err != nil { if err != nil {
t.Error("Failed to call db.Query:", err) t.Error("Failed to call db.Query:", err)
} }
defer rows.Close() defer rows.Close()
n := 1 n := 0
for rows.Next() { for rows.Next() {
var id int var id int
err = rows.Scan(&id) err = rows.Scan(&id)
if err != nil { if err != nil {
t.Error("Failed to db.Query:", err) t.Error("Failed to db.Query:", err)
} }
if id != n { if id != n + 1 {
t.Error("Failed to db.Query: not matched results") t.Error("Failed to db.Query: not matched results")
} }
n = n + 1 n = n + 1
} }
if n != 4 { if err := rows.Err(); err != nil {
t.Errorf("Expected 3 rows but retrieved %v", n-1) t.Errorf("Post-scan failed: %v\n", err)
}
if n != 3 {
t.Errorf("Expected 3 rows but retrieved %v", n)
} }
} }