From c558caf4bb16f20139321daa12fab1811c02b746 Mon Sep 17 00:00:00 2001
From: gurkanindibay <gindibay@microsoft.com>
Date: Fri, 22 Mar 2024 14:48:06 +0300
Subject: [PATCH] Fixes some test errors

---
 src/test/regress/expected/failure_setup.out   | 24 +++++++++++++++++++
 .../expected/single_node_enterprise.out       |  8 +++----
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/src/test/regress/expected/failure_setup.out b/src/test/regress/expected/failure_setup.out
index 4cbb4b0a4..7a26f8097 100644
--- a/src/test/regress/expected/failure_setup.out
+++ b/src/test/regress/expected/failure_setup.out
@@ -1,3 +1,27 @@
+ALTER SYSTEM SET citus.enable_ddl_propagation = 'true';
+SELECT pg_reload_conf();
+ pg_reload_conf
+---------------------------------------------------------------------
+ t
+(1 row)
+
+\c - - - :worker_1_port
+ALTER SYSTEM SET citus.enable_ddl_propagation = 'true';
+SELECT pg_reload_conf();
+ pg_reload_conf
+---------------------------------------------------------------------
+ t
+(1 row)
+
+\c - - - :worker_2_port
+ALTER SYSTEM SET citus.enable_ddl_propagation = 'true';
+SELECT pg_reload_conf();
+ pg_reload_conf
+---------------------------------------------------------------------
+ t
+(1 row)
+
+\c - - - :master_port
 SELECT citus.mitmproxy('conn.allow()');
  mitmproxy
 ---------------------------------------------------------------------
diff --git a/src/test/regress/expected/single_node_enterprise.out b/src/test/regress/expected/single_node_enterprise.out
index 9e3ffad76..b93bf752f 100644
--- a/src/test/regress/expected/single_node_enterprise.out
+++ b/src/test/regress/expected/single_node_enterprise.out
@@ -1,4 +1,6 @@
-
+-- we already have lots of tests targeting
+-- single node citus clusters in sql/single_node.sql
+-- in this file, we are testing enterprise features
 ALTER SYSTEM SET citus.enable_ddl_propagation = 'true';
 SELECT pg_reload_conf();
  pg_reload_conf
@@ -23,9 +25,7 @@ SELECT pg_reload_conf();
 (1 row)
 
 \c - - - :master_port
--- we already have lots of tests targeting
--- single node citus clusters in sql/single_node.sql
--- in this file, we are testing enterprise features
+
 CREATE SCHEMA single_node_ent;
 SET search_path TO single_node_ent;
 SET citus.shard_count TO 4;