add regression tests for copied foreign keys

funbringer · funbringer · commit a9101d1cda25 · 2016-09-06T19:03:56.000+03:00
diff --git a/expected/pg_pathman.out b/expected/pg_pathman.out
@@ -1707,3 +1707,62 @@ EXPLAIN (COSTS OFF) SELECT * FROM messages;
    ->  Seq Scan on messages_2
 (3 rows)
 
+/* Check primary keys generation */
+CREATE TABLE test_ref(comment TEXT UNIQUE);
+INSERT INTO test_ref VALUES('test');
+CREATE TABLE test_fkey(
+	id			INT NOT NULL,
+	comment		TEXT,
+	FOREIGN KEY (comment) REFERENCES test_ref(comment));
+INSERT INTO test_fkey SELECT generate_series(1, 1000), 'test';
+SELECT create_range_partitions('test_fkey', 'id', 1, 100);
+NOTICE:  sequence "test_fkey_seq" does not exist, skipping
+ create_range_partitions 
+-------------------------
+                      10
+(1 row)
+
+INSERT INTO test_fkey VALUES(1, 'wrong');
+ERROR:  insert or update on table "test_fkey_1" violates foreign key constraint "test_fkey_1_comment_fkey"
+SELECT drop_partitions('test_fkey');
+NOTICE:  function public.test_fkey_upd_trig_func() does not exist, skipping
+NOTICE:  100 rows copied from test_fkey_10
+NOTICE:  100 rows copied from test_fkey_9
+NOTICE:  100 rows copied from test_fkey_8
+NOTICE:  100 rows copied from test_fkey_7
+NOTICE:  100 rows copied from test_fkey_6
+NOTICE:  100 rows copied from test_fkey_5
+NOTICE:  100 rows copied from test_fkey_4
+NOTICE:  100 rows copied from test_fkey_3
+NOTICE:  100 rows copied from test_fkey_2
+NOTICE:  100 rows copied from test_fkey_1
+ drop_partitions 
+-----------------
+              10
+(1 row)
+
+SELECT create_hash_partitions('test_fkey', 'id', 10);
+ create_hash_partitions 
+------------------------
+                     10
+(1 row)
+
+INSERT INTO test_fkey VALUES(1, 'wrong');
+ERROR:  insert or update on table "test_fkey_0" violates foreign key constraint "test_fkey_0_comment_fkey"
+SELECT drop_partitions('test_fkey');
+NOTICE:  function public.test_fkey_upd_trig_func() does not exist, skipping
+NOTICE:  94 rows copied from test_fkey_9
+NOTICE:  108 rows copied from test_fkey_8
+NOTICE:  118 rows copied from test_fkey_7
+NOTICE:  95 rows copied from test_fkey_6
+NOTICE:  90 rows copied from test_fkey_5
+NOTICE:  101 rows copied from test_fkey_4
+NOTICE:  116 rows copied from test_fkey_3
+NOTICE:  90 rows copied from test_fkey_2
+NOTICE:  90 rows copied from test_fkey_1
+NOTICE:  98 rows copied from test_fkey_0
+ drop_partitions 
+-----------------
+              10
+(1 row)
+
diff --git a/sql/pg_pathman.sql b/sql/pg_pathman.sql
@@ -644,3 +644,23 @@ SELECT create_range_partitions('messages', 'id', 1, 100, 2);
 ALTER TABLE replies DROP CONSTRAINT replies_message_id_fkey;
 SELECT create_range_partitions('messages', 'id', 1, 100, 2);
 EXPLAIN (COSTS OFF) SELECT * FROM messages;
+
+/* Check primary keys generation */
+CREATE TABLE test_ref(comment TEXT UNIQUE);
+INSERT INTO test_ref VALUES('test');
+
+CREATE TABLE test_fkey(
+	id			INT NOT NULL,
+	comment		TEXT,
+	FOREIGN KEY (comment) REFERENCES test_ref(comment));
+
+INSERT INTO test_fkey SELECT generate_series(1, 1000), 'test';
+
+SELECT create_range_partitions('test_fkey', 'id', 1, 100);
+INSERT INTO test_fkey VALUES(1, 'wrong');
+INSERT INTO test_fkey VALUES(1, 'test');
+SELECT drop_partitions('test_fkey');
+
+SELECT create_hash_partitions('test_fkey', 'id', 10);
+INSERT INTO test_fkey VALUES(1, 'test');
+SELECT drop_partitions('test_fkey');
