mirror of https://github.com/citusdata/citus.git
add tests
Onur Tirtir
parent
407a0cdadc
commit
c34b6a5601
|
|
@ -278,6 +278,107 @@ SELECT
|
||||||
t
|
t
|
||||||
(1 row)
|
(1 row)
|
||||||
|
|
||||||
|
create table test_a_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_a_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_a_tbl_1', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select create_distributed_table('test_a_tbl_2', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select count(*) = 2 as colocationids_different from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'single_node'
|
||||||
|
and pg_class.relname in ('test_a_tbl_1', 'test_a_tbl_2')
|
||||||
|
) q;
|
||||||
|
colocationids_different
|
||||||
|
-------------------------
|
||||||
|
t
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
DROP TABLE test_a_tbl_1, test_a_tbl_2;
|
||||||
|
create table test_d_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_d_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_d_tbl_1', 'text_col', shard_count=>4);
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select create_distributed_table('test_d_tbl_2', 'text_col', shard_count=>6);
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select alter_distributed_table('test_d_tbl_2', shard_count=>4);
|
||||||
|
NOTICE: creating a new table for single_node.test_d_tbl_2
|
||||||
|
NOTICE: moving the data of single_node.test_d_tbl_2
|
||||||
|
NOTICE: dropping the old single_node.test_d_tbl_2
|
||||||
|
NOTICE: renaming the new table to single_node.test_d_tbl_2
|
||||||
|
alter_distributed_table
|
||||||
|
-------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select count(*) = 2 as colocationids_different from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'single_node'
|
||||||
|
and pg_class.relname in ('test_d_tbl_1', 'test_d_tbl_2')
|
||||||
|
) q;
|
||||||
|
colocationids_different
|
||||||
|
-------------------------
|
||||||
|
t
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
DROP TABLE test_d_tbl_1, test_d_tbl_2;
|
||||||
|
create table test_b_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_b_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_b_tbl_1', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select create_distributed_table('test_b_tbl_2', 'text_col', colocate_with => 'test_b_tbl_1');
|
||||||
|
ERROR: cannot colocate tables test_b_tbl_1 and test_b_tbl_2
|
||||||
|
DETAIL: Distribution column collations don't match for test_b_tbl_1 and test_b_tbl_2.
|
||||||
|
DROP TABLE test_b_tbl_1, test_b_tbl_2;
|
||||||
|
create table test_c_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_c_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_c_tbl_1', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select create_distributed_table('test_c_tbl_2', 'text_col', colocate_with => 'none');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select alter_distributed_table('test_c_tbl_2', colocate_with=>'test_c_tbl_1');
|
||||||
|
ERROR: cannot colocate with test_c_tbl_1 because collation of its distribution column is different than test_c_tbl_2
|
||||||
|
DROP TABLE test_c_tbl_1, test_c_tbl_2;
|
||||||
SET client_min_messages TO WARNING;
|
SET client_min_messages TO WARNING;
|
||||||
DROP TABLE failover_to_local, single_node_nullkey_c1, single_node_nullkey_c2, ref_table_conversion_test, single_shard_conversion_test_1, single_shard_conversion_test_2;
|
DROP TABLE failover_to_local, single_node_nullkey_c1, single_node_nullkey_c2, ref_table_conversion_test, single_shard_conversion_test_1, single_shard_conversion_test_2;
|
||||||
DROP SCHEMA tenant_1 CASCADE;
|
DROP SCHEMA tenant_1 CASCADE;
|
||||||
|
|
|
||||||
|
|
@ -95,6 +95,125 @@ SELECT count(DISTINCT colocationid) FROM pg_dist_partition WHERE logicalrelid IN
|
||||||
2
|
2
|
||||||
(1 row)
|
(1 row)
|
||||||
|
|
||||||
|
\c - - - :master_port
|
||||||
|
SET search_path TO "Update Colocation";
|
||||||
|
create table test_a_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_a_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_a_tbl_1', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select create_distributed_table('test_a_tbl_2', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select result as colocationids_different from run_command_on_all_nodes($$
|
||||||
|
select count(*) = 2 from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'Update Colocation'
|
||||||
|
and pg_class.relname in ('test_a_tbl_1', 'test_a_tbl_2')
|
||||||
|
) q;
|
||||||
|
$$);
|
||||||
|
colocationids_different
|
||||||
|
-------------------------
|
||||||
|
t
|
||||||
|
t
|
||||||
|
t
|
||||||
|
(3 rows)
|
||||||
|
|
||||||
|
DROP TABLE test_a_tbl_1, test_a_tbl_2;
|
||||||
|
create table test_d_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_d_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_d_tbl_1', 'text_col', shard_count=>4);
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select create_distributed_table('test_d_tbl_2', 'text_col', shard_count=>6);
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select alter_distributed_table('test_d_tbl_2', shard_count=>4);
|
||||||
|
NOTICE: creating a new table for "Update Colocation".test_d_tbl_2
|
||||||
|
NOTICE: moving the data of "Update Colocation".test_d_tbl_2
|
||||||
|
NOTICE: dropping the old "Update Colocation".test_d_tbl_2
|
||||||
|
NOTICE: renaming the new table to "Update Colocation".test_d_tbl_2
|
||||||
|
alter_distributed_table
|
||||||
|
-------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select result as colocationids_different from run_command_on_all_nodes($$
|
||||||
|
select count(*) = 2 from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'Update Colocation'
|
||||||
|
and pg_class.relname in ('test_d_tbl_1', 'test_d_tbl_2')
|
||||||
|
) q;
|
||||||
|
$$);
|
||||||
|
colocationids_different
|
||||||
|
-------------------------
|
||||||
|
t
|
||||||
|
t
|
||||||
|
t
|
||||||
|
(3 rows)
|
||||||
|
|
||||||
|
DROP TABLE test_d_tbl_1, test_d_tbl_2;
|
||||||
|
create table test_b_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_b_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_b_tbl_1', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select create_distributed_table('test_b_tbl_2', 'text_col', colocate_with => 'test_b_tbl_1');
|
||||||
|
ERROR: cannot colocate tables test_b_tbl_1 and test_b_tbl_2
|
||||||
|
DETAIL: Distribution column collations don't match for test_b_tbl_1 and test_b_tbl_2.
|
||||||
|
DROP TABLE test_b_tbl_1, test_b_tbl_2;
|
||||||
|
create table test_c_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_c_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
select create_distributed_table('test_c_tbl_1', 'text_col');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
select create_distributed_table('test_c_tbl_2', 'text_col', colocate_with => 'none');
|
||||||
|
create_distributed_table
|
||||||
|
--------------------------
|
||||||
|
|
||||||
|
(1 row)
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select alter_distributed_table('test_c_tbl_2', colocate_with=>'test_c_tbl_1');
|
||||||
|
ERROR: cannot colocate with test_c_tbl_1 because collation of its distribution column is different than test_c_tbl_2
|
||||||
|
DROP TABLE test_c_tbl_1, test_c_tbl_2;
|
||||||
\c - postgres - :master_port
|
\c - postgres - :master_port
|
||||||
SET client_min_messages TO ERROR;
|
SET client_min_messages TO ERROR;
|
||||||
DROP SCHEMA "Update Colocation" cascade;
|
DROP SCHEMA "Update Colocation" cascade;
|
||||||
|
SET citus.enable_ddl_propagation TO OFF;
|
||||||
|
DROP ROLE mx_update_colocation;
|
||||||
|
\c - postgres - :worker_1_port
|
||||||
|
SET citus.enable_ddl_propagation TO OFF;
|
||||||
|
DROP ROLE mx_update_colocation;
|
||||||
|
\c - postgres - :worker_2_port
|
||||||
|
SET citus.enable_ddl_propagation TO OFF;
|
||||||
|
DROP ROLE mx_update_colocation;
|
||||||
|
|
|
||||||
|
|
@ -168,6 +168,67 @@ SELECT
|
||||||
WHERE logicalrelid = 'single_node.single_shard_conversion_test_2'::regclass
|
WHERE logicalrelid = 'single_node.single_shard_conversion_test_2'::regclass
|
||||||
);
|
);
|
||||||
|
|
||||||
|
create table test_a_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_a_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_a_tbl_1', 'text_col');
|
||||||
|
select create_distributed_table('test_a_tbl_2', 'text_col');
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select count(*) = 2 as colocationids_different from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'single_node'
|
||||||
|
and pg_class.relname in ('test_a_tbl_1', 'test_a_tbl_2')
|
||||||
|
) q;
|
||||||
|
|
||||||
|
DROP TABLE test_a_tbl_1, test_a_tbl_2;
|
||||||
|
|
||||||
|
create table test_d_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_d_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_d_tbl_1', 'text_col', shard_count=>4);
|
||||||
|
select create_distributed_table('test_d_tbl_2', 'text_col', shard_count=>6);
|
||||||
|
select alter_distributed_table('test_d_tbl_2', shard_count=>4);
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select count(*) = 2 as colocationids_different from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'single_node'
|
||||||
|
and pg_class.relname in ('test_d_tbl_1', 'test_d_tbl_2')
|
||||||
|
) q;
|
||||||
|
|
||||||
|
|
||||||
|
DROP TABLE test_d_tbl_1, test_d_tbl_2;
|
||||||
|
|
||||||
|
create table test_b_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_b_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_b_tbl_1', 'text_col');
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select create_distributed_table('test_b_tbl_2', 'text_col', colocate_with => 'test_b_tbl_1');
|
||||||
|
|
||||||
|
DROP TABLE test_b_tbl_1, test_b_tbl_2;
|
||||||
|
|
||||||
|
create table test_c_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_c_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_c_tbl_1', 'text_col');
|
||||||
|
select create_distributed_table('test_c_tbl_2', 'text_col', colocate_with => 'none');
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select alter_distributed_table('test_c_tbl_2', colocate_with=>'test_c_tbl_1');
|
||||||
|
|
||||||
|
DROP TABLE test_c_tbl_1, test_c_tbl_2;
|
||||||
|
|
||||||
SET client_min_messages TO WARNING;
|
SET client_min_messages TO WARNING;
|
||||||
DROP TABLE failover_to_local, single_node_nullkey_c1, single_node_nullkey_c2, ref_table_conversion_test, single_shard_conversion_test_1, single_shard_conversion_test_2;
|
DROP TABLE failover_to_local, single_node_nullkey_c1, single_node_nullkey_c2, ref_table_conversion_test, single_shard_conversion_test_1, single_shard_conversion_test_2;
|
||||||
DROP SCHEMA tenant_1 CASCADE;
|
DROP SCHEMA tenant_1 CASCADE;
|
||||||
|
|
|
||||||
|
|
@ -53,6 +53,84 @@ SELECT count(DISTINCT colocationid) FROM pg_dist_partition WHERE logicalrelid IN
|
||||||
SET search_path TO "Update Colocation";
|
SET search_path TO "Update Colocation";
|
||||||
SELECT count(DISTINCT colocationid) FROM pg_dist_partition WHERE logicalrelid IN ('t1'::regclass, 't2'::regclass);
|
SELECT count(DISTINCT colocationid) FROM pg_dist_partition WHERE logicalrelid IN ('t1'::regclass, 't2'::regclass);
|
||||||
|
|
||||||
|
\c - - - :master_port
|
||||||
|
SET search_path TO "Update Colocation";
|
||||||
|
|
||||||
|
create table test_a_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_a_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_a_tbl_1', 'text_col');
|
||||||
|
select create_distributed_table('test_a_tbl_2', 'text_col');
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select result as colocationids_different from run_command_on_all_nodes($$
|
||||||
|
select count(*) = 2 from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'Update Colocation'
|
||||||
|
and pg_class.relname in ('test_a_tbl_1', 'test_a_tbl_2')
|
||||||
|
) q;
|
||||||
|
$$);
|
||||||
|
|
||||||
|
DROP TABLE test_a_tbl_1, test_a_tbl_2;
|
||||||
|
|
||||||
|
create table test_d_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_d_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_d_tbl_1', 'text_col', shard_count=>4);
|
||||||
|
select create_distributed_table('test_d_tbl_2', 'text_col', shard_count=>6);
|
||||||
|
select alter_distributed_table('test_d_tbl_2', shard_count=>4);
|
||||||
|
|
||||||
|
-- make sure we assing them to different colocation groups
|
||||||
|
select result as colocationids_different from run_command_on_all_nodes($$
|
||||||
|
select count(*) = 2 from (
|
||||||
|
select distinct(colocationid)
|
||||||
|
from pg_dist_partition
|
||||||
|
join pg_class on (logicalrelid = pg_class.oid)
|
||||||
|
join pg_namespace on (relnamespace = pg_namespace.oid)
|
||||||
|
join pg_dist_colocation using (colocationid)
|
||||||
|
where pg_namespace.nspname = 'Update Colocation'
|
||||||
|
and pg_class.relname in ('test_d_tbl_1', 'test_d_tbl_2')
|
||||||
|
) q;
|
||||||
|
$$);
|
||||||
|
|
||||||
|
DROP TABLE test_d_tbl_1, test_d_tbl_2;
|
||||||
|
|
||||||
|
create table test_b_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_b_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_b_tbl_1', 'text_col');
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select create_distributed_table('test_b_tbl_2', 'text_col', colocate_with => 'test_b_tbl_1');
|
||||||
|
|
||||||
|
DROP TABLE test_b_tbl_1, test_b_tbl_2;
|
||||||
|
|
||||||
|
create table test_c_tbl_1(text_col text collate "C" unique);
|
||||||
|
create table test_c_tbl_2(text_col text collate "en-x-icu" unique);
|
||||||
|
|
||||||
|
select create_distributed_table('test_c_tbl_1', 'text_col');
|
||||||
|
select create_distributed_table('test_c_tbl_2', 'text_col', colocate_with => 'none');
|
||||||
|
|
||||||
|
-- errors
|
||||||
|
select alter_distributed_table('test_c_tbl_2', colocate_with=>'test_c_tbl_1');
|
||||||
|
|
||||||
|
DROP TABLE test_c_tbl_1, test_c_tbl_2;
|
||||||
|
|
||||||
\c - postgres - :master_port
|
\c - postgres - :master_port
|
||||||
SET client_min_messages TO ERROR;
|
SET client_min_messages TO ERROR;
|
||||||
DROP SCHEMA "Update Colocation" cascade;
|
DROP SCHEMA "Update Colocation" cascade;
|
||||||
|
|
||||||
|
SET citus.enable_ddl_propagation TO OFF;
|
||||||
|
DROP ROLE mx_update_colocation;
|
||||||
|
|
||||||
|
\c - postgres - :worker_1_port
|
||||||
|
SET citus.enable_ddl_propagation TO OFF;
|
||||||
|
DROP ROLE mx_update_colocation;
|
||||||
|
|
||||||
|
\c - postgres - :worker_2_port
|
||||||
|
SET citus.enable_ddl_propagation TO OFF;
|
||||||
|
DROP ROLE mx_update_colocation;
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue