Test DuckDB TRUNCATE statement

Author: nene

test/duckdb.test.ts

@@ -27,6 +27,7 @@ import supportsOnConflict from './features/onConflict.js';
 import supportsIsDistinctFrom from './features/isDistinctFrom.js';
 import supportsArrayLiterals from './features/arrayLiterals.js';
 import supportsDataTypeCase from './options/dataTypeCase.js';
+import supportsTruncateTable from './features/truncateTable.js';
 
 describe('DuckDBFormatter', () => {
   const language = 'duckdb';
@@ -50,8 +51,7 @@ describe('DuckDBFormatter', () => {
   supportsInsertInto(format);
   supportsOnConflict(format);
   supportsUpdate(format);
-  // TODO: only without-TABLE is supported
-  // supportsTruncateTable(format, { withoutTable: true });
+  supportsTruncateTable(format, { withTable: false, withoutTable: true });
   supportsStrings(format, ["''-qq", "X''", "B''"]);
   supportsIdentifiers(format, [`""-qq`]);
   supportsBetween(format);

test/features/truncateTable.ts

@@ -3,19 +3,22 @@ import dedent from 'dedent-js';
 import { FormatFn } from '../../src/sqlFormatter.js';
 
 interface TruncateTableConfig {
+  withTable?: boolean;
   withoutTable?: boolean;
 }
 
 export default function supportsTruncateTable(
   format: FormatFn,
-  { withoutTable }: TruncateTableConfig = {}
+  { withTable = true, withoutTable }: TruncateTableConfig = {}
 ) {
-  it('formats TRUNCATE TABLE statement', () => {
-    const result = format('TRUNCATE TABLE Customers;');
-    expect(result).toBe(dedent`
-      TRUNCATE TABLE Customers;
-    `);
-  });
+  if (withTable) {
+    it('formats TRUNCATE TABLE statement', () => {
+      const result = format('TRUNCATE TABLE Customers;');
+      expect(result).toBe(dedent`
+        TRUNCATE TABLE Customers;
+      `);
+    });
+  }
 
   if (withoutTable) {
     it('formats TRUNCATE statement (without TABLE)', () => {