New pgx tests

Author: wickwirew

internal/compiler/output_columns.go

@@ -545,7 +545,7 @@ func (c *Compiler) sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, erro
 					Name:    fn.ReturnType.Name,
 				})
 
-				// Successfully found the table
+				// Failed to find table, check for type with name.
 				if err != nil {
 					table, err = qc.GetCompositeType(fn.ReturnType)
 				}

internal/endtoend/testdata/select_fn/postgresql/pgx/v5/go/db.go

@@ -0,0 +1,5 @@
+-- name: SelectBars :many
+SELECT * FROM bar_fn();
+
+-- name: SelectFoos :many
+SELECT * FROM foo_fn();

internal/endtoend/testdata/select_fn/postgresql/pgx/v5/go/models.go

@@ -0,0 +1,13 @@
+CREATE TYPE foo AS ENUM ('bar', 'baz');
+
+CREATE TYPE bar AS (foo foo, baz INTEGER);
+
+CREATE FUNCTION bar_fn()
+RETURNS SETOF bar LANGUAGE SQL STABLE AS $$
+SELECT * FROM VALUES ('bar', 1);
+$$;
+
+CREATE FUNCTION foo_fn()
+RETURNS SETOF foo LANGUAGE SQL STABLE AS $$
+SELECT * FROM VALUES ('bar');
+$$;

internal/endtoend/testdata/select_fn/postgresql/pgx/v5/go/query.sql.go

@@ -0,0 +1,13 @@
+{
+  "version": "1",
+  "packages": [
+    {
+      "path": "go",
+      "engine": "postgresql",
+      "sql_package": "pgx/v5",
+      "name": "querytest",
+      "schema": "schema.sql",
+      "queries": "query.sql"
+    }
+  ]
+}

internal/endtoend/testdata/select_fn/postgresql/pgx/v5/query.sql

@@ -155,7 +155,6 @@ func (c *Catalog) createCompositeType(stmt *ast.CompositeTypeStmt) error {
 		Name:    stmt.TypeName.Name,
 		Columns: cols,
 	})
-	// schema.Tables = append(schema.Tables, &Table{Rel: tbl, Columns: cols})
 	return nil
 }