Format RAISE statement

Author: nene

src/syntax/procedural_language.ts

@@ -193,8 +193,14 @@ export const proceduralLanguageMap: Partial<CstToDocMap<AllProceduralNodes>> = {
     }
   },
   // RAISE
-  raise_stmt: (print) => group(print.spaced(["raiseKw", "using"])),
-  raise_using_clause: (print) => group(print.spaced(["usingKw", "options"])),
+  raise_stmt: (print, node) =>
+    group([
+      print.spaced(["raiseKw", "level", "error"]),
+      node.using ? indent([line, print("using")]) : [],
+    ]),
+  raise_level: (print) => print("levelKw"),
+  raise_using_clause: (print) =>
+    group([print("usingKw"), indent([line, print("options")])]),
   raise_option_element: (print) => [print("nameKw"), " = ", print("value")],
   // ASSERT
   assert_stmt: (print) =>

test/proc/raise.test.ts

@@ -1,5 +1,5 @@
 import dedent from "dedent-js";
-import { testBigquery } from "../test_utils";
+import { testBigquery, testPlpgsql } from "../test_utils";
 
 describe("raise", () => {
   it(`formats RAISE statement`, async () => {
@@ -13,4 +13,33 @@ describe("raise", () => {
       RAISE USING MESSAGE = 'Serious error!'
     `);
   });
+
+  it(`formats RAISE with condition name`, async () => {
+    await testPlpgsql(dedent`
+      RAISE division_by_zero
+    `);
+  });
+
+  it(`formats RAISE with SQLSTATE`, async () => {
+    await testPlpgsql(dedent`
+      RAISE SQLSTATE 'P0001'
+    `);
+  });
+
+  it(`formats RAISE with level and USING clause`, async () => {
+    await testPlpgsql(dedent`
+      RAISE WARNING index_out_of_range
+        USING MESSAGE = 'Index is out of range', HINT = 'Whatever'
+    `);
+  });
+
+  it(`formats RAISE with long USING clause`, async () => {
+    await testPlpgsql(dedent`
+      RAISE index_out_of_range
+        USING
+          MESSAGE = 'Index is out of range',
+          HINT = 'Please check the index and try again',
+          DETAIL = 'The index you provided is 10, but the maximum allowed is 5'
+    `);
+  });
 });