fix: include changes to search_path manipulation

Author: soedirgo

ansible/files/postgresql_extension_custom_scripts/pgmq/after-create.sql

@@ -2,11 +2,11 @@ do $$
 declare
   extoid oid := (select oid from pg_extension where extname = 'pgmq');
   extversion text := (select extversion from pg_extension where extname = 'pgmq');
+  search_path text := (select current_setting('search_path'));
   r record;
   cls pg_class%rowtype;
 begin
-
-  set local search_path = '';
+  perform set_config('search_path', '', true);
 
 /*
     Override the pgmq.drop_queue to check if relevant tables are owned
@@ -180,4 +180,7 @@ end if;
 
     end if;
   end loop;
+
+  -- restore configs
+  perform set_config('search_path', search_path, true);
 end $$;

ansible/files/postgresql_extension_custom_scripts/supabase_vault/after-create.sql

@@ -1,8 +1,9 @@
 do $$
 declare
   extversion text := (select extversion from pg_extension where extname = 'supabase_vault');
+  search_path text := (select current_setting('search_path'));
 begin
-  set local search_path = '';
+  perform set_config('search_path', '', true);
 
   if extversion != '0.2.8' then
     grant usage on schema vault to postgres with grant option;
@@ -14,4 +15,7 @@ begin
     grant select, delete on vault.secrets, vault.decrypted_secrets to service_role;
     grant execute on function vault.create_secret, vault.update_secret, vault._crypto_aead_det_decrypt to service_role;
   end if;
+
+  -- restore configs
+  perform set_config('search_path', search_path, true);
 end $$;

nix/tests/expected/pgmq.out

@@ -200,3 +200,10 @@ order by
  pgmq        | validate_queue_name           | postgres
 (40 rows)
 
+-- assert search_path is preserved after after-create script is run
+show search_path;
+            search_path            
+-----------------------------------
+ "$user", public, auth, extensions
+(1 row)
+

nix/tests/expected/vault.out

@@ -97,3 +97,10 @@ ORDER BY
  vault  | secrets | secrets_pkey     | supabase_admin | Unique
 (2 rows)
 
+-- assert search_path is preserved after after-create script is run
+show search_path;
+            search_path            
+-----------------------------------
+ "$user", public, auth, extensions
+(1 row)
+

nix/tests/sql/pgmq.sql

@@ -101,3 +101,6 @@ where
   n.nspname = 'pgmq'
 order by
   p.proname;
+
+-- assert search_path is preserved after after-create script is run
+show search_path;

nix/tests/sql/vault.sql

@@ -51,3 +51,6 @@ WHERE
 ORDER BY
     t.relname,
     i.relname;
+
+-- assert search_path is preserved after after-create script is run
+show search_path;