From 32003c4aa1250fd3be140beded4d0c1b119f8a01 Mon Sep 17 00:00:00 2001
From: Murat Tuncer <mtuncer@citusdata.com>
Date: Mon, 26 Sep 2016 11:53:30 +0300
Subject: [PATCH] Add tests with spaces in table names

---
 src/test/regress/expected/multi_truncate.out | 59 ++++++++++++++++++++
 src/test/regress/sql/multi_truncate.sql      | 34 +++++++++++
 2 files changed, 93 insertions(+)

diff --git a/src/test/regress/expected/multi_truncate.out b/src/test/regress/expected/multi_truncate.out
index 5adde4f58..6088073ec 100644
--- a/src/test/regress/expected/multi_truncate.out
+++ b/src/test/regress/expected/multi_truncate.out
@@ -232,3 +232,62 @@ ERROR:  master_modify_multiple_shards cannot run inside a transaction block
 CONTEXT:  SQL statement "SELECT master_modify_multiple_shards(commandText)"
 PL/pgSQL function citus_truncate_trigger() line 17 at PERFORM
 DROP TABLE test_truncate_hash;
+-- test with table with spaces in it
+CREATE TABLE "a b hash" (a int, b int);
+SELECT master_create_distributed_table('"a b hash"', 'a', 'hash');
+ master_create_distributed_table 
+---------------------------------
+ 
+(1 row)
+
+SELECT master_create_worker_shards('"a b hash"', 4, 1);
+ master_create_worker_shards 
+-----------------------------
+ 
+(1 row)
+
+INSERT INTO "a b hash" values (1, 0);
+SELECT * from "a b hash";
+ a | b 
+---+---
+ 1 | 0
+(1 row)
+
+TRUNCATE TABLE "a b hash";
+SELECT * from "a b hash";
+ a | b 
+---+---
+(0 rows)
+
+DROP TABLE "a b hash";
+-- now with append
+CREATE TABLE "a b append" (a int, b int);
+SELECT master_create_distributed_table('"a b append"', 'a', 'append');
+ master_create_distributed_table 
+---------------------------------
+ 
+(1 row)
+
+SELECT master_create_empty_shard('"a b append"') AS new_shard_id \gset
+UPDATE pg_dist_shard SET shardminvalue = 1, shardmaxvalue = 500
+WHERE shardid = :new_shard_id;
+SELECT master_create_empty_shard('"a b append"') AS new_shard_id \gset
+UPDATE pg_dist_shard SET shardminvalue = 501, shardmaxvalue = 1000
+WHERE shardid = :new_shard_id;
+INSERT INTO "a b append" values (1, 1);
+INSERT INTO "a b append" values (600, 600);
+SELECT * FROM "a b append" ORDER BY a;
+  a  |  b  
+-----+-----
+   1 |   1
+ 600 | 600
+(2 rows)
+
+TRUNCATE TABLE "a b append";
+-- verify all shards are dropped
+SELECT shardid FROM pg_dist_shard where logicalrelid = '"a b append"'::regclass;
+ shardid 
+---------
+(0 rows)
+
+DROP TABLE "a b append";
diff --git a/src/test/regress/sql/multi_truncate.sql b/src/test/regress/sql/multi_truncate.sql
index 532af901a..4abc9f27e 100644
--- a/src/test/regress/sql/multi_truncate.sql
+++ b/src/test/regress/sql/multi_truncate.sql
@@ -140,3 +140,37 @@ SELECT shardid FROM pg_dist_shard where logicalrelid = 'test_truncate_hash'::reg
 BEGIN; TRUNCATE TABLE test_truncate_hash; COMMIT;
 
 DROP TABLE test_truncate_hash;
+
+-- test with table with spaces in it
+CREATE TABLE "a b hash" (a int, b int);
+SELECT master_create_distributed_table('"a b hash"', 'a', 'hash');
+SELECT master_create_worker_shards('"a b hash"', 4, 1);
+INSERT INTO "a b hash" values (1, 0);
+SELECT * from "a b hash";
+TRUNCATE TABLE "a b hash";
+SELECT * from "a b hash";
+
+DROP TABLE "a b hash";
+
+-- now with append
+CREATE TABLE "a b append" (a int, b int);
+SELECT master_create_distributed_table('"a b append"', 'a', 'append');
+SELECT master_create_empty_shard('"a b append"') AS new_shard_id \gset
+UPDATE pg_dist_shard SET shardminvalue = 1, shardmaxvalue = 500
+WHERE shardid = :new_shard_id;
+
+SELECT master_create_empty_shard('"a b append"') AS new_shard_id \gset
+UPDATE pg_dist_shard SET shardminvalue = 501, shardmaxvalue = 1000
+WHERE shardid = :new_shard_id;
+
+INSERT INTO "a b append" values (1, 1);
+INSERT INTO "a b append" values (600, 600);
+
+SELECT * FROM "a b append" ORDER BY a;
+
+TRUNCATE TABLE "a b append";
+
+-- verify all shards are dropped
+SELECT shardid FROM pg_dist_shard where logicalrelid = '"a b append"'::regclass;
+
+DROP TABLE "a b append";
\ No newline at end of file