Transaction safe Truncate

Rod Taylor

2 files changed, 22 insertions, 0 deletions

diff --git a/src/test/regress/expected/truncate.out b/src/test/regress/expected/truncate.out
index 8751b1fdb7d..2aa2f3810e3 100644
--- a/src/test/regress/expected/truncate.out
+++ b/src/test/regress/expected/truncate.out
@@ -10,7 +10,21 @@ SELECT * FROM truncate_a;
     2
 (2 rows)
 
+-- Roll truncate back
+BEGIN;
 TRUNCATE truncate_a;
+ROLLBACK;
+SELECT * FROM truncate_a;
+ col1 
+------
+    1
+    2
+(2 rows)
+
+-- Commit the truncate this time
+BEGIN;
+TRUNCATE truncate_a;
+COMMIT;
 SELECT * FROM truncate_a;
  col1 
 ------
diff --git a/src/test/regress/sql/truncate.sql b/src/test/regress/sql/truncate.sql
index 5333113a9e7..79e229feac6 100644
--- a/src/test/regress/sql/truncate.sql
+++ b/src/test/regress/sql/truncate.sql
@@ -3,7 +3,15 @@ CREATE TABLE truncate_a (col1 integer primary key);
 INSERT INTO truncate_a VALUES (1);
 INSERT INTO truncate_a VALUES (2);
 SELECT * FROM truncate_a;
+-- Roll truncate back
+BEGIN;
 TRUNCATE truncate_a;
+ROLLBACK;
+SELECT * FROM truncate_a;
+-- Commit the truncate this time
+BEGIN;
+TRUNCATE truncate_a;
+COMMIT;
 SELECT * FROM truncate_a;
 
 -- Test foreign constraint check