From febbaddc401cf31410c5fe8b5fc6dfd7da042a2e Mon Sep 17 00:00:00 2001
From: gurkanindibay <gindibay@microsoft.com>
Date: Fri, 5 Jan 2024 13:49:04 +0300
Subject: [PATCH] Fixes test errors

---
 src/backend/distributed/commands/utility_hook.c  |  3 ++-
 src/test/regress/expected/multi_test_helpers.out | 14 +++++++++++++-
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/src/backend/distributed/commands/utility_hook.c b/src/backend/distributed/commands/utility_hook.c
index 154b9ca6c..2043982fe 100644
--- a/src/backend/distributed/commands/utility_hook.c
+++ b/src/backend/distributed/commands/utility_hook.c
@@ -1664,10 +1664,10 @@ RunPostprocessMainDBCommand(Node *parsetree)
 	}
 }
 
+
 /*
  * IsStatementSupportedIn2Pc returns true if the statement is supported in 2pc
  */
-
 static bool
 IsStatementSupportedIn2Pc(Node *parsetree)
 {
@@ -1685,6 +1685,7 @@ IsStatementSupportedIn2Pc(Node *parsetree)
 	return false;
 }
 
+
 /*
  * IsStatementMarkDistributedFor2PC returns true if the statement should be marked
  * as distributed in 2pc
diff --git a/src/test/regress/expected/multi_test_helpers.out b/src/test/regress/expected/multi_test_helpers.out
index 70a541d2a..5fc694d13 100644
--- a/src/test/regress/expected/multi_test_helpers.out
+++ b/src/test/regress/expected/multi_test_helpers.out
@@ -625,4 +625,16 @@ BEGIN
     ) q2
     JOIN pg_dist_node USING (nodeid);
 END;
-$func$ LANGUAGE plpgsql;
\ No newline at end of file
+$func$ LANGUAGE plpgsql;
+CREATE OR REPLACE FUNCTION check_database_privileges(role_name text, db_name text, permissions text[])
+RETURNS TABLE(permission text, result text)
+AS $func$
+DECLARE
+    permission text;
+BEGIN
+    FOREACH permission IN ARRAY permissions
+    LOOP
+        RETURN QUERY EXECUTE format($inner$SELECT '%s', result FROM run_command_on_all_nodes($$select has_database_privilege('%s','%s', '%s'); $$)$inner$, permission, role_name, db_name, permission);
+    END LOOP;
+END;
+$func$ LANGUAGE plpgsql;