adds a copyfrom to the output_file_name tests that support it

sqlc-dev · kyleconroy · Sep 19, 2023 · Sep 14, 2023 · Sep 15, 2023 · Sep 15, 2023
commit d81f148731a6a5b6eef3ce215e4d24c19bd6a0d3
diff --git a/internal/cmd/shim.go b/internal/cmd/shim.go
@@ -105,6 +105,7 @@ func pluginGoCode(s config.SQLGo) *plugin.GoCode {
 		OutputBatchFileName:         s.OutputBatchFileName,
 		OutputModelsFileName:        s.OutputModelsFileName,
 		OutputQuerierFileName:       s.OutputQuerierFileName,
+		OutputCopyfromFileName:      s.OutputCopyFromFileName,
 		OutputFilesSuffix:           s.OutputFilesSuffix,
 		InflectionExcludeTableNames: s.InflectionExcludeTableNames,
 		QueryParameterLimit:         s.QueryParameterLimit,

diff --git a/internal/endtoend/testdata/output_file_names/pgx/v4/go/copyfrom_gen.go b/internal/endtoend/testdata/output_file_names/pgx/v4/go/copyfrom_gen.go
diff --git a/internal/endtoend/testdata/output_file_names/pgx/v4/go/db_gen.go b/internal/endtoend/testdata/output_file_names/pgx/v4/go/db_gen.go
diff --git a/internal/endtoend/testdata/output_file_names/pgx/v4/go/querier_gen.go b/internal/endtoend/testdata/output_file_names/pgx/v4/go/querier_gen.go
diff --git a/internal/endtoend/testdata/output_file_names/pgx/v4/query.sql b/internal/endtoend/testdata/output_file_names/pgx/v4/query.sql
@@ -6,3 +6,9 @@ SELECT "user".* FROM "user";
 -- name: UsersB :batchmany
 SELECT * FROM "user"
 WHERE id = $1;
+
+-- name: UsersC :copyfrom
+INSERT INTO "user"
+(id)
+VALUES
+($1);
diff --git a/internal/endtoend/testdata/output_file_names/pgx/v5/go/copyfrom_gen.go b/internal/endtoend/testdata/output_file_names/pgx/v5/go/copyfrom_gen.go
diff --git a/internal/endtoend/testdata/output_file_names/pgx/v5/go/db_gen.go b/internal/endtoend/testdata/output_file_names/pgx/v5/go/db_gen.go
diff --git a/internal/endtoend/testdata/output_file_names/pgx/v5/go/querier_gen.go b/internal/endtoend/testdata/output_file_names/pgx/v5/go/querier_gen.go
diff --git a/internal/endtoend/testdata/output_file_names/pgx/v5/query.sql b/internal/endtoend/testdata/output_file_names/pgx/v5/query.sql
@@ -6,3 +6,9 @@ SELECT "user".* FROM "user";
 -- name: UsersB :batchmany
 SELECT * FROM "user"
 WHERE id = $1;
+
+-- name: UsersC :copyfrom
+INSERT INTO "user"
+(id)
+VALUES
+    ($1);
diff --git a/protos/plugin/codegen.proto b/protos/plugin/codegen.proto
@@ -86,7 +86,7 @@ message GoCode {
   string output_db_file_name = 15;
   string output_models_file_name = 16;
   string output_querier_file_name = 17;
-  string output_copyfrom_file_name = 28; 
+  string output_copyfrom_file_name = 28;
   string output_files_suffix = 18;
   bool emit_enum_valid_method = 19;
   bool emit_all_enum_values = 20;