From 6bad5a66231848da2f65ab91d0e156cdbbcca164 Mon Sep 17 00:00:00 2001
From: Adam Guo <adamguo@amazon.com>
Date: Fri, 1 Dec 2023 10:25:27 -0500
Subject: [PATCH] Allow superuser to ALTER OWNER of datatype functions (#251)

---
 src/tleextension.c                | 4 +++-
 test/t/002_pg_tle_dump_restore.pl | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/tleextension.c b/src/tleextension.c
index 6bc5cc5..7dbafcc 100644
--- a/src/tleextension.c
+++ b/src/tleextension.c
@@ -4315,7 +4315,9 @@ _PU_HOOK
 			{
 				AlterOwnerStmt *stmt = (AlterOwnerStmt *) pu_parsetree;
 
-				if (!IsBinaryUpgrade && stmt->objectType == OBJECT_FUNCTION)
+				if (!IsBinaryUpgrade &&
+					!superuser() &&
+					stmt->objectType == OBJECT_FUNCTION)
 				{
 					ObjectAddress address;
 					Relation	relation;
diff --git a/test/t/002_pg_tle_dump_restore.pl b/test/t/002_pg_tle_dump_restore.pl
index 863542d..ff8dd87 100644
--- a/test/t/002_pg_tle_dump_restore.pl
+++ b/test/t/002_pg_tle_dump_restore.pl
@@ -298,7 +298,7 @@
 
 # Restore freshly created db with psql -d newdb -f olddb.sql
 $node->command_ok(
-    [ 'psql',  '-d', $restored_db, '-f', $dumpfilename ],
+    [ 'psql',  '-d', $restored_db, '-f', $dumpfilename, '-v', 'ON_ERROR_STOP=1' ],
     'restore new db from sql dump'
 );