external
/
citus
mirror of https://github.com/citusdata/citus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add test for citus table size func in transaction with modification 

Add test for citus_relation_size
pull/4356/head
Ahmet Gedemenli 2020-11-30 11:41:24 +03:00
parent feecb7b423
commit 67761897ab
2 changed files with 122 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

63
src/test/regress/expected/single_node.out
View File

@ -237,6 +237,69 @@ drop cascades to table "Quoed.Schema".simple_table_name_90630518
drop cascades to table "Quoed.Schema".simple_table_name_90630519
drop cascades to table "Quoed.Schema".simple_table_name_90630520
drop cascades to table "Quoed.Schema".simple_table_name_90630521
-- test citus size functions in transaction with modification
CREATE TABLE test_citus_size_func (a int);
SELECT create_distributed_table('test_citus_size_func', 'a');
create_distributed_table
---------------------------------------------------------------------
(1 row)
INSERT INTO test_citus_size_func VALUES(1), (2);
BEGIN;
-- DDL with citus_table_size
ALTER TABLE test_citus_size_func ADD COLUMN newcol INT;
SELECT citus_table_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- DDL with citus_relation_size
ALTER TABLE test_citus_size_func ADD COLUMN newcol INT;
SELECT citus_relation_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- DDL with citus_total_relation_size
ALTER TABLE test_citus_size_func ADD COLUMN newcol INT;
SELECT citus_total_relation_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- single shard insert with citus_table_size
INSERT INTO test_citus_size_func VALUES (3);
SELECT citus_table_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- multi shard modification with citus_table_size
INSERT INTO test_citus_size_func SELECT * FROM test_citus_size_func;
SELECT citus_table_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- single shard insert with citus_relation_size
INSERT INTO test_citus_size_func VALUES (3);
SELECT citus_relation_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- multi shard modification with citus_relation_size
INSERT INTO test_citus_size_func SELECT * FROM test_citus_size_func;
SELECT citus_relation_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- single shard insert with citus_total_relation_size
INSERT INTO test_citus_size_func VALUES (3);
SELECT citus_total_relation_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
BEGIN;
-- multi shard modification with citus_total_relation_size
INSERT INTO test_citus_size_func SELECT * FROM test_citus_size_func;
SELECT citus_total_relation_size('test_citus_size_func');
ERROR: citus size functions cannot be called in transaction blocks which contain multi-shard data modifications
ROLLBACK;
-- we should be able to limit intermediate results
BEGIN;
SET LOCAL citus.max_intermediate_result_size TO 0;

59
src/test/regress/sql/single_node.sql
View File

@ -121,6 +121,65 @@ ALTER TABLE simple_table_name RENAME CONSTRAINT "looo oooo ooooo ooooooooooooooo
SET search_path TO single_node;
DROP SCHEMA "Quoed.Schema" CASCADE;
-- test citus size functions in transaction with modification
CREATE TABLE test_citus_size_func (a int);
SELECT create_distributed_table('test_citus_size_func', 'a');
INSERT INTO test_citus_size_func VALUES(1), (2);
BEGIN;
-- DDL with citus_table_size
ALTER TABLE test_citus_size_func ADD COLUMN newcol INT;
SELECT citus_table_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- DDL with citus_relation_size
ALTER TABLE test_citus_size_func ADD COLUMN newcol INT;
SELECT citus_relation_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- DDL with citus_total_relation_size
ALTER TABLE test_citus_size_func ADD COLUMN newcol INT;
SELECT citus_total_relation_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- single shard insert with citus_table_size
INSERT INTO test_citus_size_func VALUES (3);
SELECT citus_table_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- multi shard modification with citus_table_size
INSERT INTO test_citus_size_func SELECT * FROM test_citus_size_func;
SELECT citus_table_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- single shard insert with citus_relation_size
INSERT INTO test_citus_size_func VALUES (3);
SELECT citus_relation_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- multi shard modification with citus_relation_size
INSERT INTO test_citus_size_func SELECT * FROM test_citus_size_func;
SELECT citus_relation_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- single shard insert with citus_total_relation_size
INSERT INTO test_citus_size_func VALUES (3);
SELECT citus_total_relation_size('test_citus_size_func');
ROLLBACK;
BEGIN;
-- multi shard modification with citus_total_relation_size
INSERT INTO test_citus_size_func SELECT * FROM test_citus_size_func;
SELECT citus_total_relation_size('test_citus_size_func');
ROLLBACK;
-- we should be able to limit intermediate results
BEGIN;
SET LOCAL citus.max_intermediate_result_size TO 0;