Address reviews

citusdata · Aug 24, 2023 · c7dae63 · c7dae63
1 parent 64ac072
commit c7dae63
Show file tree

Hide file tree

Showing 2 changed files with 50 additions and 0 deletions.
diff --git a/src/test/regress/expected/pg16.out b/src/test/regress/expected/pg16.out
@@ -305,6 +305,32 @@ SELECT * FROM copy_default ORDER BY id;
 (3 rows)
 
 TRUNCATE copy_default;
+\c - - - :worker_1_port
+COPY pg16.copy_default FROM stdin WITH (format csv, default '\D');
+SELECT * FROM pg16.copy_default ORDER BY id;
+ id | text_value |         ts_value
+---------------------------------------------------------------------
+  1 | value      | Mon Jul 04 00:00:00 2022
+  2 | test       | Sun Jul 03 00:00:00 2022
+  3 | test       | Tue Jul 05 00:00:00 2022
+(3 rows)
+
+\c - - - :master_port
+TRUNCATE pg16.copy_default;
+\c - - - :worker_2_port
+COPY pg16.copy_default FROM stdin WITH (format csv, default '\D');
+SELECT * FROM pg16.copy_default ORDER BY id;
+ id | text_value |         ts_value
+---------------------------------------------------------------------
+  1 | value      | Mon Jul 04 00:00:00 2022
+  2 | test       | Sun Jul 03 00:00:00 2022
+  3 | test       | Tue Jul 05 00:00:00 2022
+(3 rows)
+
+\c - - - :master_port
+SET search_path TO pg16;
+SET citus.shard_count TO 1;
+SET citus.shard_replication_factor TO 1;
 -- DEFAULT cannot be used in COPY TO
 COPY (select 1 as test) TO stdout WITH (default '\D');
 ERROR:  COPY DEFAULT only available using COPY FROM

diff --git a/src/test/regress/sql/pg16.sql b/src/test/regress/sql/pg16.sql
@@ -192,6 +192,30 @@ COPY copy_default FROM stdin WITH (format csv, default '\D');
 SELECT * FROM copy_default ORDER BY id;
 TRUNCATE copy_default;
 
+\c - - - :worker_1_port
+COPY pg16.copy_default FROM stdin WITH (format csv, default '\D');
+1,value,2022-07-04
+2,\D,2022-07-03
+3,\D,\D
+\.
+SELECT * FROM pg16.copy_default ORDER BY id;
+
+\c - - - :master_port
+TRUNCATE pg16.copy_default;
+
+\c - - - :worker_2_port
+COPY pg16.copy_default FROM stdin WITH (format csv, default '\D');
+1,value,2022-07-04
+2,\D,2022-07-03
+3,\D,\D
+\.
+SELECT * FROM pg16.copy_default ORDER BY id;
+
+\c - - - :master_port
+SET search_path TO pg16;
+SET citus.shard_count TO 1;
+SET citus.shard_replication_factor TO 1;
+
 -- DEFAULT cannot be used in COPY TO
 COPY (select 1 as test) TO stdout WITH (default '\D');