Format DROP SUBSCRIPTION

Author: nene

src/syntax/subscription.ts

@@ -1,8 +1,8 @@
 import { AllSubscriptionNodes } from "sql-parser-cst";
-import { group, indent, line, join } from "../print_utils";
+import { group, line, join } from "../print_utils";
 import { CstToDocMap } from "../CstToDocMap";
 
-export const subscriptionMap: Partial<CstToDocMap<AllSubscriptionNodes>> = {
+export const subscriptionMap: CstToDocMap<AllSubscriptionNodes> = {
   create_subscription_stmt: (print, node) =>
     group(
       join(line, [
@@ -12,4 +12,9 @@ export const subscriptionMap: Partial<CstToDocMap<AllSubscriptionNodes>> = {
         ...(node.with ? [print.spaced(["with"])] : []),
       ]),
     ),
+
+  drop_subscription_stmt: (print) =>
+    group(
+      print.spaced(["dropSubscriptionKw", "ifExistsKw", "name", "behaviorKw"]),
+    ),
 };

test/postgresql/subscription.test.ts

@@ -34,4 +34,12 @@ describe("subscriptions", () => {
       `);
     });
   });
+
+  describe("DROP SUBSCRIPTION", () => {
+    it(`formats DROP SUBSCRIPTION`, async () => {
+      await testPostgresql(dedent`
+        DROP SUBSCRIPTION IF EXISTS my_sub CASCADE
+      `);
+    });
+  });
 });