add tests

Author: viewsharp

internal/codegen/golang/imports.go

@@ -408,9 +408,12 @@ func (i *importer) queryImports(filename string) fileImports {
 
 	if containIter {
 		std["iter"] = struct{}{}
-		if sqlpkg.IsPGX() {
+		switch sqlpkg {
+		case opts.SQLDriverPGXV4:
+			pkg[ImportSpec{Path: "github.com/jackc/pgx/v4"}] = struct{}{}
+		case opts.SQLDriverPGXV5:
 			pkg[ImportSpec{Path: "github.com/jackc/pgx/v5"}] = struct{}{}
-		} else {
+		default:
 			std["database/sql"] = struct{}{}
 		}
 	}

internal/endtoend/testdata/iter/mysql/go/db.go

@@ -0,0 +1,4 @@
+-- name: IterValues :iter
+SELECT *
+FROM foo
+WHERE b = ?;

internal/endtoend/testdata/iter/mysql/go/models.go

@@ -0,0 +1,2 @@
+CREATE TABLE foo (a text, b integer);
+

internal/endtoend/testdata/iter/mysql/go/query.sql.go

@@ -0,0 +1,12 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "mysql",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/iter/mysql/query.sql

@@ -0,0 +1,4 @@
+-- name: IterValues :iter
+SELECT *
+FROM myschema.foo
+WHERE b = $1;

internal/endtoend/testdata/iter/mysql/schema.sql

@@ -0,0 +1,3 @@
+CREATE SCHEMA myschema;
+CREATE TABLE myschema.foo (a text, b integer);
+

internal/endtoend/testdata/iter/mysql/sqlc.json

@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v4",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}