test: Add case for #1990

sqlc-dev · kyleconroy · Oct 18, 2023 · Oct 16, 2023 · Oct 16, 2023 · Oct 16, 2023
commit 83a925d91d7afaca95f4770e970bcbf07b26fbed
diff --git a/internal/endtoend/testdata/select_subquery_alias/issue.md b/internal/endtoend/testdata/select_subquery_alias/issue.md
@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/1990
diff --git a/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/exec.json b/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/exec.json
@@ -0,0 +1,3 @@
+{
+  "contexts": ["managed-db"]
+}
diff --git a/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/go/db.go b/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/go/db.go
diff --git a/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/go/models.go b/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/go/models.go
diff --git a/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/go/query.sql.go b/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/go/query.sql.go
diff --git a/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/query.sql b/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/query.sql
@@ -0,0 +1,10 @@
+-- name: FindWallets :many
+select id, address, balance, total_balance from 
+(
+	select id, address, balance,
+	  sum(balance) over (order by balance desc rows between unbounded preceding and current row) as total_balance,
+	  sum(balance) over (order by balance desc rows between unbounded preceding and current row) - balance as last_balance
+	from wallets
+	where type=$1
+) amounts
+where amounts.last_balance < $2;
diff --git a/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/schema.sql b/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/schema.sql
@@ -0,0 +1,7 @@
+CREATE TABLE IF NOT EXISTS wallets
+(
+    id BIGSERIAL PRIMARY KEY,
+    address VARCHAR(44) NOT NULL,
+    type VARCHAR(44) NOT NULL,
+    balance DECIMAL(32, 18) NULL
+);
diff --git a/internal/endtoend/testdata/select_subquery_alias/postgresql/pgx/sqlc.yaml b/internal/endtoend/testdata/select_subquery_alias/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"