test: Add case for #2152

sqlc-dev · kyleconroy · Oct 18, 2023 · Oct 18, 2023 · Oct 18, 2023 · Oct 18, 2023
commit c5291ae833bb1260b085611869db1228572c7e4a
diff --git a/internal/endtoend/testdata/batch_parameter_type/issue.md b/internal/endtoend/testdata/batch_parameter_type/issue.md
@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/2152
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/exec.json b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/exec.json
@@ -0,0 +1,3 @@
+{
+  "contexts": ["managed-db"]
+}
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/batch.go b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/batch.go
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/db.go b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/db.go
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/models.go b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/query.sql.go b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/go/query.sql.go
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/query.sql b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/query.sql
@@ -0,0 +1,32 @@
+-- name: InsertMappping :batchexec
+WITH
+    table1
+        AS (
+            SELECT
+                version
+            FROM
+                solar_commcard_mapping
+            WHERE
+                "deviceId" = $1
+            ORDER BY
+                "updatedAt" DESC
+            LIMIT
+                1
+        )
+INSERT
+INTO
+    solar_commcard_mapping
+        ("deviceId", version, sn, "updatedAt")
+SELECT
+    $1, @version::text, $3, $4
+WHERE
+    NOT
+        EXISTS(
+            SELECT
+                *
+            FROM
+                table1
+            WHERE
+                table1.version = @version::text
+        )
+    OR NOT EXISTS(SELECT * FROM table1);
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/schema.sql b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/schema.sql
@@ -0,0 +1,14 @@
+CREATE TABLE public.solar_commcard_mapping (
+    id
+        INT8 NOT NULL,
+    "deviceId"
+        INT8 NOT NULL,
+    version
+        VARCHAR(32) DEFAULT ''::VARCHAR NOT NULL,
+    sn
+        VARCHAR(32) DEFAULT ''::VARCHAR NOT NULL,
+    "createdAt"
+        TIMESTAMPTZ DEFAULT now(),
+    "updatedAt"
+        TIMESTAMPTZ DEFAULT now()
+);
diff --git a/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/sqlc.yaml b/internal/endtoend/testdata/batch_parameter_type/postgresql/pgx/sqlc.yaml
@@ -0,0 +1,10 @@
+version: "2"
+sql:
+  - engine: "postgresql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"
+        sql_package: "pgx/v5"