From 8de802eec5abec74232fee460a5fa15e10e69a96 Mon Sep 17 00:00:00 2001
From: Marco Slot <marco.slot@gmail.com>
Date: Wed, 10 Nov 2021 10:50:48 +0100
Subject: [PATCH] Enable local_shared_pool_size 5 in arbitrary configs test

---
 src/test/regress/citus_tests/config.py        |  3 +--
 .../expected/prepared_statements_1.out        | 22 +++++++++++++++++++
 .../regress/sql/prepared_statements_1.sql     | 15 +++++++++++++
 3 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/src/test/regress/citus_tests/config.py b/src/test/regress/citus_tests/config.py
index 52b0714c8..beccf8b26 100644
--- a/src/test/regress/citus_tests/config.py
+++ b/src/test/regress/citus_tests/config.py
@@ -227,8 +227,7 @@ class CitusSmallSharedPoolSizeConfig(CitusMXBaseClusterConfig):
     def __init__(self, arguments):
         super().__init__(arguments)
         self.new_settings = {
-            # can be uncommented after https://github.com/citusdata/citus/issues/5342
-            # "citus.local_shared_pool_size": 5,
+             "citus.local_shared_pool_size": 5,
             "citus.max_shared_pool_size": 5,
         }
 
diff --git a/src/test/regress/expected/prepared_statements_1.out b/src/test/regress/expected/prepared_statements_1.out
index e71be3f9d..418453705 100644
--- a/src/test/regress/expected/prepared_statements_1.out
+++ b/src/test/regress/expected/prepared_statements_1.out
@@ -24,3 +24,25 @@ BEGIN;
 (1 row)
 
 ROLLBACK;
+PREPARE xact_repartitioned_prepared AS
+	SELECT count(*) FROM repartition_prepared_test t1 JOIN repartition_prepared_test t2 USING (b);
+BEGIN;
+	-- Prepared re-partition join in a transaction block after a write
+	INSERT INTO repartition_prepared_test VALUES (1,2);
+	EXECUTE xact_repartitioned_prepared;
+ count
+---------------------------------------------------------------------
+   226
+(1 row)
+
+ROLLBACK;
+BEGIN;
+	-- Prepared re-partition join in a transaction block before a write
+	EXECUTE xact_repartitioned_prepared;
+ count
+---------------------------------------------------------------------
+   209
+(1 row)
+
+	INSERT INTO repartition_prepared_test VALUES (1,2);
+ROLLBACK;
diff --git a/src/test/regress/sql/prepared_statements_1.sql b/src/test/regress/sql/prepared_statements_1.sql
index ec3f521ad..aee511696 100644
--- a/src/test/regress/sql/prepared_statements_1.sql
+++ b/src/test/regress/sql/prepared_statements_1.sql
@@ -17,3 +17,18 @@ BEGIN;
 	CREATE TEMP TABLE repartition_prepared_tmp AS EXECUTE repartition_prepared(1);
 	SELECT count(*) from repartition_prepared_tmp;
 ROLLBACK;
+
+PREPARE xact_repartitioned_prepared AS
+	SELECT count(*) FROM repartition_prepared_test t1 JOIN repartition_prepared_test t2 USING (b);
+
+BEGIN;
+	-- Prepared re-partition join in a transaction block after a write
+	INSERT INTO repartition_prepared_test VALUES (1,2);
+	EXECUTE xact_repartitioned_prepared;
+ROLLBACK;
+
+BEGIN;
+	-- Prepared re-partition join in a transaction block before a write
+	EXECUTE xact_repartitioned_prepared;
+	INSERT INTO repartition_prepared_test VALUES (1,2);
+ROLLBACK;