Enable local_shared_pool_size 5 in arbitrary configs test

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,
         }
 

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;

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;