Ensure sqlTypeCase effects ARRAY[] expression in BigQuery, but not in PostgreSQL

Author: nene

src/syntax/expr.ts

@@ -176,6 +176,7 @@ export const exprMap: CstToDocMap<AllExprNodes> = {
   interval_expr: (print) => print.spaced(["intervalKw", "expr", "unit"]),
   interval_unit_range: (print) => print.spaced(["fromUnit", "toKw", "toUnit"]),
   interval_unit: (print) => print("unitKw"),
+  array_literal_expr: (print) => print(["arrayKw", "expr"]),
   array_expr: (print) =>
     group(["[", indent([softline, print("expr")]), softline, "]"]),
   struct_expr: (print) =>

test/expr/literal.test.ts

@@ -39,7 +39,7 @@ describe("literal", () => {
   });
 
   describe("array literals", () => {
-    it(`formats array literals`, async () => {
+    it(`formats BigQuery array literals`, async () => {
       await testBigquery(
         dedent`
           SELECT
@@ -51,7 +51,7 @@ describe("literal", () => {
       );
     });
 
-    it(`formats long array literal to multiple lines`, async () => {
+    it(`formats long BigQuery array literal to multiple lines`, async () => {
       await testBigquery(
         dedent`
           SELECT
@@ -64,6 +64,21 @@ describe("literal", () => {
         `,
       );
     });
+
+    it(`formats PostgreSQL array literals`, async () => {
+      await testPostgresql(
+        dedent`
+          SELECT
+            ARRAY[1, 2, 3],
+            ARRAY[
+              'a somewhat large array',
+              'containing some strings',
+              'which themselves',
+              'are somewhat long.'
+            ]
+        `,
+      );
+    });
   });
 
   describe("struct literals", () => {

test/options/typeCase.test.ts

@@ -72,6 +72,39 @@ describe("sqlTypeCase option", () => {
     `);
   });
 
+  it(`does not apply to ARRAY[] literals in PostgreSQL`, async () => {
+    expect(
+      await pretty(`SELECT ARRAY[1, 2, 3]`, {
+        sqlTypeCase: "lower",
+        dialect: "postgresql",
+      }),
+    ).toBe(dedent`
+      SELECT ARRAY[1, 2, 3]
+    `);
+  });
+
+  it(`applies to ARRAY[] literals in BigQuery`, async () => {
+    expect(
+      await pretty(`SELECT ARRAY[1], ARRAY<INT64>[2]`, {
+        sqlTypeCase: "lower",
+        dialect: "bigquery",
+      }),
+    ).toBe(dedent`
+      SELECT array[1], array<int64>[2]
+    `);
+  });
+
+  it(`applies to STRUCT() literals in BigQuery`, async () => {
+    expect(
+      await pretty(`SELECT STRUCT(2), STRUCT<INT>(2)`, {
+        sqlTypeCase: "lower",
+        dialect: "bigquery",
+      }),
+    ).toBe(dedent`
+      SELECT struct(2), struct<int>(2)
+    `);
+  });
+
   it(`does not apply to SETOF data types`, async () => {
     expect(
       await pretty(`CREATE TABLE t (x SETOF INT)`, {