From 61347439c2aecb1bc542273838aaa7dedbdb90e0 Mon Sep 17 00:00:00 2001 From: Oliver Giles Date: Fri, 1 Jul 2022 11:41:50 +1200 Subject: [PATCH] TestQueryer: actually check Rows returned Fixes a test which did not correctly exercise the multi-statement Queryer functionality --- sqlite3_test.go | 22 ++++++++++++---------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/sqlite3_test.go b/sqlite3_test.go index 9ee87e7e..d948378e 100644 --- a/sqlite3_test.go +++ b/sqlite3_test.go @@ -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) } }