@supabase/pg-delta 1.0.0-alpha.4 → 1.0.0-alpha.6

package/src/core/plan/serialize.test.ts

@@ -0,0 +1,317 @@
+import { describe, expect, test } from "bun:test";
+import type { Change } from "../change.types.ts";
+import { getObjectName, getObjectSchema, getParentInfo } from "./serialize.ts";
+
+describe("getObjectName", () => {
+  const cases: [string, unknown, string][] = [
+    [
+      "aggregate",
+      { objectType: "aggregate", aggregate: { name: "my_agg" } },
+      "my_agg",
+    ],
+    [
+      "collation",
+      { objectType: "collation", collation: { name: "my_coll" } },
+      "my_coll",
+    ],
+    [
+      "composite_type",
+      { objectType: "composite_type", compositeType: { name: "my_comp" } },
+      "my_comp",
+    ],
+    [
+      "domain",
+      { objectType: "domain", domain: { name: "my_domain" } },
+      "my_domain",
+    ],
+    ["enum", { objectType: "enum", enum: { name: "my_enum" } }, "my_enum"],
+    [
+      "event_trigger",
+      { objectType: "event_trigger", eventTrigger: { name: "my_evt" } },
+      "my_evt",
+    ],
+    [
+      "extension",
+      { objectType: "extension", extension: { name: "my_ext" } },
+      "my_ext",
+    ],
+    [
+      "foreign_data_wrapper",
+      {
+        objectType: "foreign_data_wrapper",
+        foreignDataWrapper: { name: "my_fdw" },
+      },
+      "my_fdw",
+    ],
+    [
+      "foreign_table",
+      { objectType: "foreign_table", foreignTable: { name: "my_ft" } },
+      "my_ft",
+    ],
+    ["index", { objectType: "index", index: { name: "my_idx" } }, "my_idx"],
+    [
+      "language",
+      { objectType: "language", language: { name: "plpgsql" } },
+      "plpgsql",
+    ],
+    [
+      "materialized_view",
+      { objectType: "materialized_view", materializedView: { name: "my_mv" } },
+      "my_mv",
+    ],
+    [
+      "procedure",
+      { objectType: "procedure", procedure: { name: "my_proc" } },
+      "my_proc",
+    ],
+    [
+      "publication",
+      { objectType: "publication", publication: { name: "my_pub" } },
+      "my_pub",
+    ],
+    ["range", { objectType: "range", range: { name: "my_range" } }, "my_range"],
+    [
+      "rls_policy",
+      { objectType: "rls_policy", policy: { name: "my_policy" } },
+      "my_policy",
+    ],
+    ["role", { objectType: "role", role: { name: "my_role" } }, "my_role"],
+    ["rule", { objectType: "rule", rule: { name: "my_rule" } }, "my_rule"],
+    [
+      "schema",
+      { objectType: "schema", schema: { name: "my_schema" } },
+      "my_schema",
+    ],
+    [
+      "sequence",
+      { objectType: "sequence", sequence: { name: "my_seq" } },
+      "my_seq",
+    ],
+    [
+      "server",
+      { objectType: "server", server: { name: "my_server" } },
+      "my_server",
+    ],
+    [
+      "subscription",
+      { objectType: "subscription", subscription: { name: "my_sub" } },
+      "my_sub",
+    ],
+    ["table", { objectType: "table", table: { name: "my_table" } }, "my_table"],
+    [
+      "trigger",
+      { objectType: "trigger", trigger: { name: "my_trigger" } },
+      "my_trigger",
+    ],
+    [
+      "user_mapping",
+      {
+        objectType: "user_mapping",
+        userMapping: { user: "alice", server: "remote" },
+      },
+      "alice@remote",
+    ],
+    ["view", { objectType: "view", view: { name: "my_view" } }, "my_view"],
+  ];
+
+  for (const [label, stub, expected] of cases) {
+    test(label, () => {
+      expect(getObjectName(stub as unknown as Change)).toBe(expected);
+    });
+  }
+});
+
+describe("getObjectSchema", () => {
+  const withSchema: [string, unknown, string][] = [
+    [
+      "aggregate",
+      { objectType: "aggregate", aggregate: { schema: "public" } },
+      "public",
+    ],
+    [
+      "collation",
+      { objectType: "collation", collation: { schema: "pg_catalog" } },
+      "pg_catalog",
+    ],
+    [
+      "composite_type",
+      { objectType: "composite_type", compositeType: { schema: "public" } },
+      "public",
+    ],
+    [
+      "domain",
+      { objectType: "domain", domain: { schema: "public" } },
+      "public",
+    ],
+    ["enum", { objectType: "enum", enum: { schema: "public" } }, "public"],
+    [
+      "extension",
+      { objectType: "extension", extension: { schema: "public" } },
+      "public",
+    ],
+    [
+      "foreign_table",
+      { objectType: "foreign_table", foreignTable: { schema: "public" } },
+      "public",
+    ],
+    ["index", { objectType: "index", index: { schema: "public" } }, "public"],
+    [
+      "materialized_view",
+      {
+        objectType: "materialized_view",
+        materializedView: { schema: "public" },
+      },
+      "public",
+    ],
+    [
+      "procedure",
+      { objectType: "procedure", procedure: { schema: "public" } },
+      "public",
+    ],
+    ["range", { objectType: "range", range: { schema: "public" } }, "public"],
+    [
+      "rls_policy",
+      { objectType: "rls_policy", policy: { schema: "public" } },
+      "public",
+    ],
+    ["rule", { objectType: "rule", rule: { schema: "public" } }, "public"],
+    [
+      "schema",
+      { objectType: "schema", schema: { name: "my_schema" } },
+      "my_schema",
+    ],
+    [
+      "sequence",
+      { objectType: "sequence", sequence: { schema: "public" } },
+      "public",
+    ],
+    ["table", { objectType: "table", table: { schema: "public" } }, "public"],
+    [
+      "trigger",
+      { objectType: "trigger", trigger: { schema: "public" } },
+      "public",
+    ],
+    ["view", { objectType: "view", view: { schema: "public" } }, "public"],
+  ];
+
+  for (const [label, stub, expected] of withSchema) {
+    test(`${label} returns schema`, () => {
+      expect(getObjectSchema(stub as unknown as Change)).toBe(expected);
+    });
+  }
+
+  const withoutSchema: [string, unknown][] = [
+    ["event_trigger", { objectType: "event_trigger" }],
+    ["foreign_data_wrapper", { objectType: "foreign_data_wrapper" }],
+    ["language", { objectType: "language" }],
+    ["publication", { objectType: "publication" }],
+    ["role", { objectType: "role" }],
+    ["server", { objectType: "server" }],
+    ["subscription", { objectType: "subscription" }],
+    ["user_mapping", { objectType: "user_mapping" }],
+  ];
+
+  for (const [label, stub] of withoutSchema) {
+    test(`${label} returns null`, () => {
+      expect(getObjectSchema(stub as unknown as Change)).toBeNull();
+    });
+  }
+});
+
+describe("getParentInfo", () => {
+  test("index on table", () => {
+    const change = {
+      objectType: "index",
+      index: { table_name: "users", table_relkind: "r" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({ type: "table", name: "users" });
+  });
+
+  test("index on materialized view", () => {
+    const change = {
+      objectType: "index",
+      index: { table_name: "user_stats", table_relkind: "m" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({
+      type: "materialized_view",
+      name: "user_stats",
+    });
+  });
+
+  test("trigger on table", () => {
+    const change = {
+      objectType: "trigger",
+      trigger: { table_name: "orders", table_relkind: "r" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({ type: "table", name: "orders" });
+  });
+
+  test("trigger on view", () => {
+    const change = {
+      objectType: "trigger",
+      trigger: { table_name: "order_view", table_relkind: "v" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({ type: "view", name: "order_view" });
+  });
+
+  test("trigger on materialized view", () => {
+    const change = {
+      objectType: "trigger",
+      trigger: { table_name: "order_mv", table_relkind: "m" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({
+      type: "materialized_view",
+      name: "order_mv",
+    });
+  });
+
+  test("rule on table", () => {
+    const change = {
+      objectType: "rule",
+      rule: { table_name: "items", relation_kind: "r" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({ type: "table", name: "items" });
+  });
+
+  test("rule on view", () => {
+    const change = {
+      objectType: "rule",
+      rule: { table_name: "item_view", relation_kind: "v" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({ type: "view", name: "item_view" });
+  });
+
+  test("rule on materialized view", () => {
+    const change = {
+      objectType: "rule",
+      rule: { table_name: "item_mv", relation_kind: "m" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({
+      type: "materialized_view",
+      name: "item_mv",
+    });
+  });
+
+  test("rls_policy", () => {
+    const change = {
+      objectType: "rls_policy",
+      policy: { table_name: "secrets" },
+    } as unknown as Change;
+    expect(getParentInfo(change)).toEqual({ type: "table", name: "secrets" });
+  });
+
+  const nullCases = [
+    "table",
+    "view",
+    "schema",
+    "sequence",
+    "role",
+    "extension",
+  ] as const;
+  for (const objectType of nullCases) {
+    test(`${objectType} returns null`, () => {
+      const change = { objectType } as unknown as Change;
+      expect(getParentInfo(change)).toBeNull();
+    });
+  }
+});

package/src/core/plan/serialize.ts

@@ -158,10 +158,24 @@ export function getParentInfo(change: Change): ParentInfo | null {
         change.index.table_relkind === "m" ? "materialized_view" : "table";
       return { type: parentType, name: change.index.table_name };
     }
-    case "trigger":
-      return { type: "table", name: change.trigger.table_name };
-    case "rule":
-      return { type: "table", name: change.rule.table_name };
+    case "trigger": {
+      const parentType =
+        change.trigger.table_relkind === "v"
+          ? "view"
+          : change.trigger.table_relkind === "m"
+            ? "materialized_view"
+            : "table";
+      return { type: parentType, name: change.trigger.table_name };
+    }
+    case "rule": {
+      const parentType =
+        change.rule.relation_kind === "v"
+          ? "view"
+          : change.rule.relation_kind === "m"
+            ? "materialized_view"
+            : "table";
+      return { type: parentType, name: change.rule.table_name };
+    }
     case "rls_policy":
       return { type: "table", name: change.policy.table_name };
     case "aggregate":

package/src/core/plan/sql-format/fixtures.ts

@@ -213,7 +213,6 @@ import {
   AlterPublicationAddTables,
   AlterPublicationDropSchemas,
   AlterPublicationDropTables,
-  AlterPublicationSetForAllTables,
   AlterPublicationSetList,
   AlterPublicationSetOptions,
   AlterPublicationSetOwner,
@@ -1080,6 +1079,7 @@ const trigger = new Trigger({
   schema: "public",
   name: "trg_audit",
   table_name: "table_with_very_long_name_for_formatting_and_wrapping_test",
+  table_relkind: "r",
   function_schema: "public",
   function_name: "audit_trigger_fn",
   trigger_type: 7,
@@ -1252,6 +1252,7 @@ const role = new Role({
       objtype: "r",
       grantee: "app_reader",
       privileges: [{ privilege: "SELECT", grantable: false }],
+      is_implicit: false,
     },
   ],
 });
@@ -1908,10 +1909,6 @@ const changeCases: ChangeCase[] = [
       setPublishViaPartitionRoot: true,
     }),
   },
-  {
-    label: "publication.alter.set_all_tables",
-    change: new AlterPublicationSetForAllTables({ publication }),
-  },
   {
     label: "publication.alter.set_list",
     change: new AlterPublicationSetList({ publication }),

package/src/core/plan/sql-format/format-lowercase-coverage.test.ts

@@ -52,6 +52,58 @@ describe("lowercase coverage formatting", () => {
     );
   });
 
+  test("lowercases all ALTER DEFAULT PRIVILEGES object-type keywords", () => {
+    const statements = [
+      "ALTER DEFAULT PRIVILEGES FOR ROLE app_user IN SCHEMA public GRANT ALL ON TABLES TO app_reader;",
+      "ALTER DEFAULT PRIVILEGES FOR ROLE app_user GRANT ALL ON SEQUENCES TO app_reader;",
+      "ALTER DEFAULT PRIVILEGES FOR ROLE app_user GRANT ALL ON ROUTINES TO PUBLIC;",
+      "ALTER DEFAULT PRIVILEGES FOR ROLE app_user GRANT ALL ON TYPES TO PUBLIC;",
+      "ALTER DEFAULT PRIVILEGES FOR ROLE app_user IN SCHEMA api GRANT ALL ON SCHEMAS TO app_admin;",
+      "ALTER DEFAULT PRIVILEGES FOR ROLE app_user REVOKE ALL ON SEQUENCES FROM app_reader;",
+      "ALTER DEFAULT PRIVILEGES FOR ROLE app_user REVOKE ALL ON TYPES FROM PUBLIC;",
+    ];
+
+    const formatted = formatSqlStatements(statements, {
+      keywordCase: "lower",
+    });
+
+    const normalized = formatted.map((v) => v.replace(/\s+/g, " ").trim());
+    expect(normalized).toMatchInlineSnapshot(`
+      [
+        "alter default privileges for role app_user in schema public grant all on tables to app_reader",
+        "alter default privileges for role app_user grant all on sequences to app_reader",
+        "alter default privileges for role app_user grant all on routines to public",
+        "alter default privileges for role app_user grant all on types to public",
+        "alter default privileges for role app_user in schema api grant all on schemas to app_admin",
+        "alter default privileges for role app_user revoke all on sequences from app_reader",
+        "alter default privileges for role app_user revoke all on types from public",
+      ]
+    `);
+  });
+
+  test("lowercases PUBLIC in standalone GRANT/REVOKE statements", () => {
+    const statements = [
+      "GRANT ALL ON SCHEMA public TO PUBLIC;",
+      "GRANT EXECUTE ON FUNCTION public.my_fn() TO PUBLIC;",
+      "REVOKE ALL ON SCHEMA public FROM PUBLIC;",
+      "GRANT USAGE ON TYPE public.my_type TO PUBLIC;",
+    ];
+
+    const formatted = formatSqlStatements(statements, {
+      keywordCase: "lower",
+    });
+
+    const normalized = formatted.map((v) => v.replace(/\s+/g, " ").trim());
+    expect(normalized).toMatchInlineSnapshot(`
+      [
+        "grant all on schema public to public",
+        "grant execute on function public.my_fn() to public",
+        "revoke all on schema public from public",
+        "grant usage on type public.my_type to public",
+      ]
+    `);
+  });
+
   test("preserves full CHECK clause text while casing surrounding structure", () => {
     const [formatted] = formatSqlStatements(
       [

package/src/core/plan/sql-format/format-off.test.ts

@@ -314,9 +314,6 @@ describe("sql formatting snapshots", () => {
       -- publication.alter.set_options
       ALTER PUBLICATION pub_custom SET (publish = 'insert, update, delete, truncate', publish_via_partition_root = false);
 
-      -- publication.alter.set_all_tables
-      ALTER PUBLICATION pub_custom SET FOR ALL TABLES;
-
       -- publication.alter.set_list
       ALTER PUBLICATION pub_custom SET TABLE public.articles_with_a_very_long_name_very_very_long_name_that_will_go_above_the_wrapping_limit (id, title) WHERE (published = true), TABLE public.comments_a_little_smaller_name_than_the_previous_one, TABLES IN SCHEMA analytics;
 
@@ -403,25 +400,25 @@ describe("sql formatting snapshots", () => {
       DROP FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(IN "p_schema_name_for_analytics" text, IN "p_table_name_for_metrics" text, IN "p_limit_count_default" integer);
 
       -- function.alter.change_owner
-      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name OWNER TO new_admin;
+      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) OWNER TO new_admin;
 
       -- function.alter.set_security
-      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name SECURITY INVOKER;
+      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) SECURITY INVOKER;
 
       -- function.alter.set_config
-      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name SET work_mem TO '256MB';
+      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) SET work_mem TO '256MB';
 
       -- function.alter.set_volatility
-      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name IMMUTABLE;
+      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) IMMUTABLE;
 
       -- function.alter.set_strictness
-      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name CALLED ON NULL INPUT;
+      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) CALLED ON NULL INPUT;
 
       -- function.alter.set_leakproof
-      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name LEAKPROOF;
+      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) LEAKPROOF;
 
       -- function.alter.set_parallel
-      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name PARALLEL RESTRICTED;
+      ALTER FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) PARALLEL RESTRICTED;
 
       -- function.comment
       COMMENT ON FUNCTION public.calculate_metrics_for_analytics_dashboard_with_extended_name(text,text,integer) IS 'Calculate metrics for a given table';
@@ -475,13 +472,13 @@ describe("sql formatting snapshots", () => {
       DROP POLICY allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test;
 
       -- policy.alter.set_roles
-      ALTER POLICY public.allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test TO authenticated, anon;
+      ALTER POLICY allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test TO authenticated, anon;
 
       -- policy.alter.set_using
-      ALTER POLICY public.allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test USING (auth.uid() = user_id AND status = 'active');
+      ALTER POLICY allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test USING (auth.uid() = user_id AND status = 'active');
 
       -- policy.alter.set_with_check
-      ALTER POLICY public.allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test WITH CHECK (auth.uid() = user_id);
+      ALTER POLICY allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test WITH CHECK (auth.uid() = user_id);
 
       -- policy.comment
       COMMENT ON POLICY allow_select_own ON public.table_with_very_long_name_for_formatting_and_wrapping_test IS 'rls policy comment';
@@ -563,7 +560,7 @@ describe("sql formatting snapshots", () => {
       REVOKE GRANT OPTION FOR SELECT ON analytics.daily_stats FROM app_reader;
 
       -- aggregate.create
-      CREATE AGGREGATE public.array_cat_agg(anycompatiblearray) (SFUNC = array_cat, STYPE = anycompatiblearray, COMBINEFUNC = array_cat, INITCOND = '{}', PARALLEL SAFE, STRICT);
+      CREATE AGGREGATE public.array_cat_agg(anycompatiblearray) (SFUNC = array_cat, STYPE = anycompatiblearray, COMBINEFUNC = array_cat, INITCOND = '{}', PARALLEL = SAFE, STRICT);
 
       -- aggregate.drop
       DROP AGGREGATE public.array_cat_agg(anycompatiblearray);
@@ -758,13 +755,13 @@ describe("sql formatting snapshots", () => {
       COMMENT ON FOREIGN TABLE public.remote_users IS NULL;
 
       -- foreign_table.grant
-      GRANT SELECT ON FOREIGN TABLE public.remote_users TO app_reader;
+      GRANT SELECT ON TABLE public.remote_users TO app_reader;
 
       -- foreign_table.revoke
-      REVOKE SELECT ON FOREIGN TABLE public.remote_users FROM app_reader;
+      REVOKE SELECT ON TABLE public.remote_users FROM app_reader;
 
       -- foreign_table.revoke_grant_option
-      REVOKE GRANT OPTION FOR SELECT ON FOREIGN TABLE public.remote_users FROM app_reader;
+      REVOKE GRANT OPTION FOR SELECT ON TABLE public.remote_users FROM app_reader;
 
       -- server.create
       CREATE SERVER remote_server TYPE 'postgresql' VERSION '16.0' FOREIGN DATA WRAPPER postgres_fdw OPTIONS (host 'remote.host', port '5432', dbname 'remote_db');

package/src/core/plan/sql-format/format-pretty-lower-leading.test.ts

@@ -403,9 +403,6 @@ describe("sql formatting snapshots", () => {
       alter publication pub_custom
           set (publish = 'insert, update, delete, truncate', publish_via_partition_root = false);
 
-      -- publication.alter.set_all_tables
-      alter publication pub_custom set for all tables;
-
       -- publication.alter.set_list
       alter publication pub_custom
           set table
@@ -520,29 +517,39 @@ describe("sql formatting snapshots", () => {
           in "p_table_name_for_metrics" text, in "p_limit_count_default" integer);
 
       -- function.alter.change_owner
-      alter function public.calculate_metrics_for_analytics_dashboard_with_extended_name owner to
-          new_admin;
+      alter function
+          public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) owner
+          to new_admin;
 
       -- function.alter.set_security
-      alter function public.calculate_metrics_for_analytics_dashboard_with_extended_name security invoker;
+      alter function
+          public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer)
+          security invoker;
 
       -- function.alter.set_config
-      alter function public.calculate_metrics_for_analytics_dashboard_with_extended_name
+      alter function
+          public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer)
           set work_mem to '256MB';
 
       -- function.alter.set_volatility
-      alter function public.calculate_metrics_for_analytics_dashboard_with_extended_name immutable;
+      alter function
+          public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer)
+          immutable;
 
       -- function.alter.set_strictness
-      alter function public.calculate_metrics_for_analytics_dashboard_with_extended_name called
+      alter function
+          public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer) called
           on null input;
 
       -- function.alter.set_leakproof
-      alter function public.calculate_metrics_for_analytics_dashboard_with_extended_name leakproof;
+      alter function
+          public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer)
+          leakproof;
 
       -- function.alter.set_parallel
-      alter function public.calculate_metrics_for_analytics_dashboard_with_extended_name parallel
-          restricted;
+      alter function
+          public.calculate_metrics_for_analytics_dashboard_with_extended_name(text, text, integer)
+          parallel restricted;
 
       -- function.comment
       comment on function
@@ -622,18 +629,16 @@ describe("sql formatting snapshots", () => {
       drop policy allow_select_own on public.table_with_very_long_name_for_formatting_and_wrapping_test;
 
       -- policy.alter.set_roles
-      alter policy public.allow_select_own
+      alter policy allow_select_own
           on public.table_with_very_long_name_for_formatting_and_wrapping_test to authenticated, anon;
 
       -- policy.alter.set_using
-      alter policy public.allow_select_own
-          on public.table_with_very_long_name_for_formatting_and_wrapping_test
+      alter policy allow_select_own on public.table_with_very_long_name_for_formatting_and_wrapping_test
           using (auth.uid() = user_id AND status = 'active');
 
       -- policy.alter.set_with_check
-      alter policy public.allow_select_own
-          on public.table_with_very_long_name_for_formatting_and_wrapping_test with
-          check (auth.uid() = user_id);
+      alter policy allow_select_own on public.table_with_very_long_name_for_formatting_and_wrapping_test
+          with check (auth.uid() = user_id);
 
       -- policy.comment
       comment on policy allow_select_own
@@ -745,7 +750,7 @@ describe("sql formatting snapshots", () => {
           , stype       = anycompatiblearray
           , combinefunc = array_cat
           , initcond    = '{}'
-          , parallel safe
+          , parallel    = SAFE
           , strict
       );
 
@@ -990,13 +995,13 @@ describe("sql formatting snapshots", () => {
       comment on foreign table public.remote_users is null;
 
       -- foreign_table.grant
-      grant select on foreign table public.remote_users to app_reader;
+      grant select on table public.remote_users to app_reader;
 
       -- foreign_table.revoke
-      revoke select on foreign table public.remote_users from app_reader;
+      revoke select on table public.remote_users from app_reader;
 
       -- foreign_table.revoke_grant_option
-      revoke grant option for select on foreign table public.remote_users from app_reader;
+      revoke grant option for select on table public.remote_users from app_reader;
 
       -- server.create
       create server remote_server