test: Add a test case for #1739

sqlc-dev · kyleconroy · Oct 17, 2023 · Oct 16, 2023 · Oct 17, 2023 · Oct 16, 2023
commit 3ff79f2d447e442723ed3d0d37749b84678f1761
diff --git a/internal/endtoend/testdata/column_alias/issue.md b/internal/endtoend/testdata/column_alias/issue.md
@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/1739
diff --git a/internal/endtoend/testdata/column_alias/stdlib/exec.json b/internal/endtoend/testdata/column_alias/stdlib/exec.json
@@ -0,0 +1,3 @@
+{
+    "contexts": ["managed-db"]
+}
diff --git a/internal/endtoend/testdata/column_alias/stdlib/go/db.go b/internal/endtoend/testdata/column_alias/stdlib/go/db.go
diff --git a/internal/endtoend/testdata/column_alias/stdlib/go/models.go b/internal/endtoend/testdata/column_alias/stdlib/go/models.go
diff --git a/internal/endtoend/testdata/column_alias/stdlib/go/query.sql.go b/internal/endtoend/testdata/column_alias/stdlib/go/query.sql.go
diff --git a/internal/endtoend/testdata/column_alias/stdlib/query.sql b/internal/endtoend/testdata/column_alias/stdlib/query.sql
@@ -0,0 +1,21 @@
+-- name: GetUsers :many
+SELECT 
+    users.id,
+    users.fname,
+    users.lname,
+    users.email,
+    users.created_at,
+    rank_email,
+    rank_fname,
+    rank_lname,
+    similarity
+FROM 
+    users, 
+    to_tsvector(users.email || users.fname || users.lname) document,
+    to_tsquery(@search_term::TEXT) query,
+    NULLIF(ts_rank(to_tsvector(users.email), query), 0) rank_email,
+    NULLIF(ts_rank(to_tsvector(users.fname), query), 0) rank_fname,
+    NULLIF(ts_rank(to_tsvector(users.lname), query), 0) rank_lname,
+    SIMILARITY(@search_term::TEXT, users.email || users.fname || users.lname) similarity
+WHERE query @@ document OR similarity > 0
+ORDER BY rank_email, rank_lname, rank_fname, similarity DESC NULLS LAST;
diff --git a/internal/endtoend/testdata/column_alias/stdlib/schema.sql b/internal/endtoend/testdata/column_alias/stdlib/schema.sql
@@ -0,0 +1,12 @@
+CREATE EXTENSION pg_trgm;
+CREATE EXTENSION pgcrypto;
+
+CREATE TABLE users(
+    id INT GENERATED ALWAYS AS IDENTITY NOT NULL,
+    fname VARCHAR(100) NOT NULL,
+    lname VARCHAR(100) NOT NULL,
+    email VARCHAR(100) NOT NULL UNIQUE,
+    enc_passwd TEXT NOT NULL,
+    created_at TIMESTAMP WITH TIME ZONE NOT NULL default (NOW() AT TIME ZONE 'utc'),
+    PRIMARY KEY(id)
+);
diff --git a/internal/endtoend/testdata/column_alias/stdlib/sqlc.yaml b/internal/endtoend/testdata/column_alias/stdlib/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: "database/sql"