Allow superuser to ALTER OWNER of datatype functions (#251)

aws · Dec 1, 2023 · 6bad5a6 · 6bad5a6
1 parent 93ef5bf
commit 6bad5a6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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'
 );