Add nixos test for pg_partman extension

Run upgrade and pg_regress tests for pg_partman extension in NixOS tests.

Author: jfroche

nix/ext/pg_partman.nix

@@ -97,8 +97,9 @@ pkgs.buildEnv {
     inherit versions numberOfVersions switch-ext-version;
     pname = "${pname}-all";
     hasBackgroundWorker = true;
+    defaultSchema = "partman";
     defaultSettings = {
-      shared_preload_libraries = [ "pg_partman_bgw" ];
+      shared_preload_libraries = [ "pg_partman" ];
     };
     version =
       "multi-" + lib.concatStringsSep "-" (map (v: lib.replaceStrings [ "." ] [ "-" ] v) versions);

nix/ext/tests/default.nix

@@ -146,12 +146,14 @@ let
             "17": [${lib.concatStringsSep ", " (map (s: ''"${s}"'') (versions "17"))}],
           }
           extension_name = "${pname}"
+          support_upgrade = True
           pg17_configuration = "${pg17-configuration}"
           ext_has_background_worker = ${
             if (installedExtension "15") ? hasBackgroundWorker then "True" else "False"
           }
           sql_test_directory = Path("${../../tests}")
           pg_regress_test_name = "${(installedExtension "15").pgRegressTestName or pname}"
+          ext_schema = "${(installedExtension "15").defaultSchema or "public"}"
 
           ${builtins.readFile ./lib.py}
 
@@ -160,7 +162,8 @@ let
           server.wait_for_unit("multi-user.target")
           server.wait_for_unit("postgresql.service")
 
-          test = PostgresExtensionTest(server, extension_name, versions, sql_test_directory)
+          test = PostgresExtensionTest(server, extension_name, versions, sql_test_directory, support_upgrade, ext_schema)
+          test.create_schema()
 
           with subtest("Check upgrade path with postgresql 15"):
             test.check_upgrade_path("15")
@@ -224,6 +227,7 @@ builtins.listToAttrs (
       "pg_stat_monitor"
       "pg_tle"
       "pgaudit"
+      "pg_partman"
       "vector"
       "wal2json"
       "wrappers"

nix/ext/tests/lib.py

@@ -20,6 +20,7 @@ def __init__(
         versions: Versions,
         sql_test_dir: Path,
         support_upgrade: bool = True,
+        schema: str = "public",
     ):
         """Initialize the PostgreSQL extension test framework.
 
@@ -35,6 +36,10 @@ def __init__(
         self.versions = versions
         self.support_upgrade = support_upgrade
         self.sql_test_dir = sql_test_dir
+        self.schema = schema
+
+    def create_schema(self):
+        self.run_sql(f"CREATE SCHEMA IF NOT EXISTS {self.schema};")
 
     def run_sql(self, query: str) -> str:
         return self.vm.succeed(
@@ -50,8 +55,12 @@ def drop_extension(self):
         self.run_sql(f"DROP EXTENSION IF EXISTS {self.extension_name};")
 
     def install_extension(self, version: str):
+        if self.schema != "public":
+            ext_schema = f"SCHEMA {self.schema} "
+        else:
+            ext_schema = ""
         self.run_sql(
-            f"""CREATE EXTENSION {self.extension_name} WITH VERSION '{version}' CASCADE;"""
+            f"""CREATE EXTENSION {self.extension_name} WITH {ext_schema}VERSION '{version}' CASCADE;"""
         )
         # Verify version was installed correctly
         self.assert_version_matches(version)

nix/tests/expected/pg_partman.out

@@ -18,3 +18,7 @@ SELECT partman.create_parent(
  t
 (1 row)
 
+DROP SCHEMA partman_test CASCADE;
+NOTICE:  drop cascades to 2 other objects
+DETAIL:  drop cascades to table partman_test.time_taptest_table
+drop cascades to table partman_test.time_taptest_table_template

nix/tests/sql/pg_partman.sql

@@ -18,3 +18,5 @@ SELECT partman.create_parent(
     , p_interval := '1 day'
     , p_template_table := 'partman_test.time_taptest_table_template'
 );
+
+DROP SCHEMA partman_test CASCADE;