test: Add case for #2644

Author: kyleconroy

internal/endtoend/testdata/cte_recursive_subquery/issue.md

@@ -0,0 +1 @@
+https://github.com/sqlc-dev/sqlc/issues/2644

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/exec.json

@@ -0,0 +1,3 @@
+{
+  "contexts": ["managed-db"]
+}

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/go/db.go

@@ -0,0 +1,33 @@
+-- name: GetLatestVersion :one
+WITH RECURSIVE search_tree(id, chain_id, chain_counter) AS (
+	SELECT base.id, base.id AS chain_id, 0 as chain_counter
+	FROM versions AS base
+	WHERE base.previous_version_id IS NULL
+  	UNION ALL
+	SELECT v.id, search_tree.chain_id, search_tree.chain_counter + 1
+	FROM versions AS v
+	INNER JOIN search_tree ON search_tree.id = v.previous_version_id
+)
+SELECT DISTINCT ON (search_tree.chain_id) 
+	search_tree.id
+FROM search_tree   
+ORDER BY search_tree.chain_id, chain_counter DESC;
+
+-- name: GetLatestVersionWithSubquery :one
+SELECT id
+FROM versions
+WHERE versions.id IN (
+  WITH RECURSIVE search_tree(id, chain_id, chain_counter) AS (
+	SELECT base.id, base.id AS chain_id, 0 as chain_counter
+	FROM versions AS base
+	WHERE versions.previous_version_id IS NULL
+	UNION ALL
+	SELECT v.id, search_tree.chain_id, search_tree.chain_counter + 1
+	FROM versions AS v
+	INNER JOIN search_tree ON search_tree.id = v.previous_version_id 
+  )
+  SELECT DISTINCT ON (search_tree.chain_id) 
+	search_tree.id
+  FROM search_tree   
+  ORDER BY search_tree.chain_id, chain_counter DESC
+);

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/go/models.go

@@ -0,0 +1,5 @@
+CREATE TABLE versions (
+  id   BIGSERIAL PRIMARY KEY,
+  name TEXT,
+  previous_version_id bigint NOT NULL
+);

internal/endtoend/testdata/cte_recursive_subquery/postgresql/pgx/go/query.sql.go

@@ -0,0 +1,14 @@
+version: "2"
+cloud:
+  project: "01HAQMMECEYQYKFJN8MP16QC41"
+sql:
+  - engine: "postgresql"
+    schema: "schema.sql"
+    queries: "query.sql"
+    gen:
+      go:
+        package: "querytest"
+        out: "go"
+        sql_package: "pgx/v5"
+    database:
+      managed: true