Support ON CONFLICT clause in Postgres and SQLite

Author: nene

src/languages/postgresql/postgresql.formatter.ts

@@ -36,6 +36,8 @@ const onelineClauses = expandPhrases([
   // - update:
   'UPDATE [ONLY]',
   'WHERE CURRENT OF',
+  // - insert:
+  'ON CONFLICT',
   // - delete:
   'DELETE FROM [ONLY]',
   // - drop table:

src/languages/sqlite/sqlite.formatter.ts

@@ -32,6 +32,8 @@ const reservedClauses = expandPhrases([
 const onelineClauses = expandPhrases([
   // - update:
   'UPDATE [OR ABORT | OR FAIL | OR IGNORE | OR REPLACE | OR ROLLBACK]',
+  // - insert:
+  'ON CONFLICT',
   // - delete:
   'DELETE FROM',
   // - drop table:

test/behavesLikeMariaDbFormatter.ts

@@ -116,14 +116,4 @@ export default function behavesLikeMariaDbFormatter(format: FormatFn) {
         (2, 'Dog');
     `);
   });
-
-  // Regression test for issue #535
-  it('gracefully handles ON CONFLICT .. DO syntax', () => {
-    expect(format(`INSERT INTO tbl VALUES (1,'Blah') ON CONFLICT DO NOTHING;`)).toBe(dedent`
-      INSERT INTO
-        tbl
-      VALUES
-        (1, 'Blah') ON CONFLICT DO NOTHING;
-    `);
-  });
 }

test/features/onConflict.ts

@@ -0,0 +1,16 @@
+import dedent from 'dedent-js';
+
+import { FormatFn } from '../../src/sqlFormatter.js';
+
+export default function supportsOnConflict(format: FormatFn) {
+  // Regression test for issue #535
+  it('supports INSERT .. ON CONFLICT syntax', () => {
+    expect(format(`INSERT INTO tbl VALUES (1,'Blah') ON CONFLICT DO NOTHING;`)).toBe(dedent`
+      INSERT INTO
+        tbl
+      VALUES
+        (1, 'Blah')
+      ON CONFLICT DO NOTHING;
+    `);
+  });
+}

test/postgresql.test.ts

@@ -25,6 +25,7 @@ import supportsInsertInto from './features/insertInto.js';
 import supportsUpdate from './features/update.js';
 import supportsTruncateTable from './features/truncateTable.js';
 import supportsCreateView from './features/createView.js';
+import supportsOnConflict from './features/onConflict.js';
 
 describe('PostgreSqlFormatter', () => {
   const language = 'postgresql';
@@ -45,6 +46,7 @@ describe('PostgreSqlFormatter', () => {
   });
   supportsDeleteFrom(format);
   supportsInsertInto(format);
+  supportsOnConflict(format);
   supportsUpdate(format, { whereCurrentOf: true });
   supportsTruncateTable(format, { withoutTable: true });
   supportsStrings(format, ["''-qq", "U&''", "X''", "B''"]);

test/sqlite.test.ts

@@ -21,6 +21,7 @@ import supportsLimiting from './features/limiting.js';
 import supportsInsertInto from './features/insertInto.js';
 import supportsUpdate from './features/update.js';
 import supportsCreateView from './features/createView.js';
+import supportsOnConflict from './features/onConflict.js';
 
 describe('SqliteFormatter', () => {
   const language = 'sqlite';
@@ -40,6 +41,7 @@ describe('SqliteFormatter', () => {
   });
   supportsDeleteFrom(format);
   supportsInsertInto(format);
+  supportsOnConflict(format);
   supportsUpdate(format);
   supportsStrings(format, ["''-qq", "X''"]);
   supportsIdentifiers(format, [`""-qq`, '``', '[]']);