SET search_path TO truncate_tests_schema;
-- Test truncate rollback on a basic table
SELECT COUNT(*) FROM basic_table;
 count
---------------------------------------------------------------------
    10
(1 row)

BEGIN;
TRUNCATE basic_table;
SELECT COUNT(*) FROM basic_table;
 count
---------------------------------------------------------------------
     0
(1 row)

ROLLBACK;
SELECT COUNT(*) FROM basic_table;
 count
---------------------------------------------------------------------
    10
(1 row)

-- Test truncate on a basic table
SELECT COUNT(*) FROM basic_table;
 count
---------------------------------------------------------------------
    10
(1 row)

TRUNCATE basic_table;
SELECT COUNT(*) FROM basic_table;
 count
---------------------------------------------------------------------
     0
(1 row)

-- Test trucate rollback on partitioned table
SELECT COUNT(*) FROM partitioned_table_0;
 count
---------------------------------------------------------------------
     5
(1 row)

BEGIN;
TRUNCATE partitioned_table;
SELECT COUNT(*) FROM partitioned_table_0;
 count
---------------------------------------------------------------------
     0
(1 row)

ROLLBACK;
SELECT COUNT(*) FROM partitioned_table_0;
 count
---------------------------------------------------------------------
     5
(1 row)

-- Test truncate a partioned table
SELECT COUNT(*) FROM partitioned_table;
 count
---------------------------------------------------------------------
    10
(1 row)

SELECT COUNT(*) FROM partitioned_table_1;
 count
---------------------------------------------------------------------
     5
(1 row)

TRUNCATE partitioned_table;
SELECT COUNT(*) FROM partitioned_table;
 count
---------------------------------------------------------------------
     0
(1 row)

SELECT COUNT(*) FROM partitioned_table_1;
 count
---------------------------------------------------------------------
     0
(1 row)