test: Add case for #2364

sqlc-dev · kyleconroy · Oct 18, 2023 · Oct 18, 2023 · Oct 18, 2023 · Oct 18, 2023
commit fc10954c4b2c556ed1a0d9e47f5082cc0fb3b451
diff --git a/internal/endtoend/testdata/func_star_expansion/issue.md b/internal/endtoend/testdata/func_star_expansion/issue.md
@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/2364
diff --git a/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/exec.json b/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/exec.json
@@ -0,0 +1,3 @@
+{
+  "contexts": ["managed-db"]
+}
diff --git a/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/go/db.go b/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/go/db.go
diff --git a/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/go/models.go b/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/go/query.sql.go b/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/go/query.sql.go
diff --git a/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/query.sql b/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/query.sql
@@ -0,0 +1,8 @@
+-- name: TestFuncGetTime :one
+select * from test_func_get_time();
+
+-- name: TestFuncSelectString :one
+select * from test_func_select_string($1);
+
+-- name: TestFuncSelectBlog :one
+select * from test_func_select_blog($1);
diff --git a/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/schema.sql b/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/schema.sql
@@ -0,0 +1,20 @@
+create function test_func_get_time ()
+    returns timestamp AS $$
+Begin
+    return now();
+End;
+$$ language plpgsql;
+
+create function test_func_select_string (in p_string text)
+    returns text AS $$
+Begin
+    return p_string;
+End;
+$$ language plpgsql;
+
+create function test_func_select_blog(in p_id int)
+    returns table (id int, name text, created_at timestamp, updated_at timestamp) AS $$
+BEGIN RETURN QUERY
+    select id, name, created_at, updated_at from blog where id = p_id;
+END;
+$$ language plpgsql;
diff --git a/internal/endtoend/testdata/func_star_expansion/postgresql/pgx/sqlc.yaml b/internal/endtoend/testdata/func_star_expansion/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"