Add tests for casing ARRAY data types

Author: karlhorky

src/languages/db2/db2.keywords.ts

@@ -403,6 +403,7 @@ export const keywords: string[] = [
 
 export const dataTypes: string[] = [
   // https://www.ibm.com/docs/en/db2-for-zos/12?topic=columns-data-types
+  'ARRAY',
   'CCSID',
   'CHAR',
   'CHARACTER',

src/languages/db2i/db2i.keywords.ts

@@ -498,6 +498,7 @@ export const keywords: string[] = [
 
 export const dataTypes: string[] = [
   // https://www.ibm.com/docs/en/i/7.2?topic=iaodsd-odbc-data-types-how-they-correspond-db2-i-database-types
+  'ARRAY',
   'BIGINT',
   'BINARY',
   'BIT',

test/features/arrayLiterals.ts

@@ -31,6 +31,87 @@ export default function supportsArrayLiterals(format: FormatFn, cfg: ArrayLitera
           ];
       `);
     });
+
+    it('supports preserving ARRAY[] literals keywords casing', () => {
+      expect(
+        format(
+          `SELECT ArrAy[1, 2] FROM aRRAY['aaa', 'bbb', 'ccc', 'ddd', 'eee', 'fff', 'ggg', 'hhh', 'iii', 'jjj'];`,
+          {
+            dataTypeCase: 'preserve',
+          }
+        )
+      ).toBe(dedent`
+        SELECT
+          ArrAy[1, 2]
+        FROM
+          aRRAY[
+            'aaa',
+            'bbb',
+            'ccc',
+            'ddd',
+            'eee',
+            'fff',
+            'ggg',
+            'hhh',
+            'iii',
+            'jjj'
+          ];
+      `);
+    });
+
+    it('supports converting ARRAY[] literals keywords to uppercase', () => {
+      expect(
+        format(
+          `SELECT ArrAy[1, 2] FROM aRRAY['aaa', 'bbb', 'ccc', 'ddd', 'eee', 'fff', 'ggg', 'hhh', 'iii', 'jjj'];`,
+          {
+            dataTypeCase: 'upper',
+          }
+        )
+      ).toBe(dedent`
+        SELECT
+          ARRAY[1, 2]
+        FROM
+          ARRAY[
+            'aaa',
+            'bbb',
+            'ccc',
+            'ddd',
+            'eee',
+            'fff',
+            'ggg',
+            'hhh',
+            'iii',
+            'jjj'
+          ];
+      `);
+    });
+
+    it('supports converting ARRAY[] literals keywords to lowercase', () => {
+      expect(
+        format(
+          `SELECT ArrAy[1, 2] FROM aRRAY['aaa', 'bbb', 'ccc', 'ddd', 'eee', 'fff', 'ggg', 'hhh', 'iii', 'jjj'];`,
+          {
+            dataTypeCase: 'lower',
+          }
+        )
+      ).toBe(dedent`
+        SELECT
+          array[1, 2]
+        FROM
+          array[
+            'aaa',
+            'bbb',
+            'ccc',
+            'ddd',
+            'eee',
+            'fff',
+            'ggg',
+            'hhh',
+            'iii',
+            'jjj'
+          ];
+      `);
+    });
   }
 
   if (cfg.withoutArrayPrefix) {