fix tests

Author: viewsharp

examples/booktest/mysql/db_test.go

@@ -155,7 +155,9 @@ func TestBooks(t *testing.T) {
 		Title: "my book title",
 		Yr:    2016,
 	})
-	for book := range rows.Iterate() {
+
+	// We need to collect the iterator elements to avoid panic when we make other db queries in the loop.
+	for _, book := range slices.Collect(rows.Iterate()) {
 		t.Logf("Book %d (%s): %s available: %s\n", book.BookID, book.BookType, book.Title, book.Available.Format(time.RFC822Z))
 		author, err := dq.GetAuthor(ctx, book.AuthorID)
 		if err != nil {

examples/booktest/postgresql/db_test.go

@@ -144,7 +144,9 @@ func TestBooks(t *testing.T) {
 		Title: "my book title",
 		Year:  2016,
 	})
-	for book := range rows.Iterate() {
+
+	// We need to collect the iterator elements to avoid panic when we make other db queries in the loop.
+	for _, book := range slices.Collect(rows.Iterate()) {
 		t.Logf("Book %d (%s): %s available: %s\n", book.BookID, book.BookType, book.Title, book.Available.Time.Format(time.RFC822Z))
 		author, err := dq.GetAuthor(ctx, book.AuthorID)
 		if err != nil {

examples/booktest/sqlite/db_test.go

@@ -143,7 +143,9 @@ func TestBooks(t *testing.T) {
 		Title: "my book title",
 		Yr:    2016,
 	})
-	for book := range rows.Iterate() {
+
+	// We need to collect the iterator elements to avoid panic when we make other db queries in the loop.
+	for _, book := range slices.Collect(rows.Iterate()) {
 		t.Logf("Book %d (%s): %s available: %s\n", book.BookID, book.BookType, book.Title, book.Available.Format(time.RFC822Z))
 		author, err := dq.GetAuthor(ctx, book.AuthorID)
 		if err != nil {