Generalize E'' string tests

Author: nene

test/duckdb.test.ts

@@ -52,7 +52,7 @@ describe('DuckDBFormatter', () => {
   supportsOnConflict(format);
   supportsUpdate(format);
   supportsTruncateTable(format, { withTable: false, withoutTable: true });
-  supportsStrings(format, ["''-qq", "X''", "B''"]);
+  supportsStrings(format, ["''-qq", "X''", "B''", "E''"]);
   supportsIdentifiers(format, [`""-qq`]);
   supportsBetween(format);
   // Missing: '::' type cast (tested separately)
@@ -138,18 +138,6 @@ describe('DuckDBFormatter', () => {
   });
 
   // DuckDB-specific string types
-  it("supports E'' strings with C-style escapes", () => {
-    expect(format("E'blah blah'")).toBe("E'blah blah'");
-    expect(format("E'some \\' FROM escapes'")).toBe("E'some \\' FROM escapes'");
-    expect(format("SELECT E'blah' FROM foo")).toBe(dedent`
-      SELECT
-        E'blah'
-      FROM
-        foo
-    `);
-    expect(format("E'blah''blah'")).toBe("E'blah''blah'");
-  });
-
   it('supports basic dollar-quoted strings', () => {
     expect(format('$$foo JOIN bar$$')).toBe('$$foo JOIN bar$$');
     expect(format('$$foo $ JOIN bar$$')).toBe('$$foo $ JOIN bar$$');

test/features/strings.ts

@@ -17,7 +17,8 @@ type StringType =
   | "B''" // no escaping
   | 'B""' // no escaping
   | "R''" // no escaping
-  | 'R""'; // no escaping
+  | 'R""' // no escaping
+  | "E''"; // with backslash escaping
 
 export default function supportsStrings(format: FormatFn, stringTypes: StringType[]) {
   if (stringTypes.includes('""-qq') || stringTypes.includes('""-bs')) {
@@ -238,4 +239,22 @@ export default function supportsStrings(format: FormatFn, stringTypes: StringTyp
       expect(format(`r"a ha"r"hm mm"`)).toBe(`r"a ha" r"hm mm"`);
     });
   }
+
+  if (stringTypes.includes("E''")) {
+    it("supports E'' strings with C-style escapes", () => {
+      expect(format("E'blah blah'")).toBe("E'blah blah'");
+      expect(format("E'some \\' FROM escapes'")).toBe("E'some \\' FROM escapes'");
+      expect(format("SELECT E'blah' FROM foo")).toBe(dedent`
+        SELECT
+          E'blah'
+        FROM
+          foo
+      `);
+      expect(format("E'blah''blah'")).toBe("E'blah''blah'");
+    });
+
+    it(`detects consecutive E'' strings as separate ones`, () => {
+      expect(format(`e'a ha'e'hm mm'`)).toBe(`e'a ha' e'hm mm'`);
+    });
+  }
 }

test/postgresql.test.ts

@@ -55,7 +55,7 @@ describe('PostgreSqlFormatter', () => {
   supportsOnConflict(format);
   supportsUpdate(format, { whereCurrentOf: true });
   supportsTruncateTable(format, { withoutTable: true });
-  supportsStrings(format, ["''-qq", "U&''", "X''", "B''"]);
+  supportsStrings(format, ["''-qq", "U&''", "X''", "B''", "E''"]);
   supportsIdentifiers(format, [`""-qq`, 'U&""']);
   supportsBetween(format);
   supportsSchema(format);
@@ -173,18 +173,6 @@ describe('PostgreSqlFormatter', () => {
   });
 
   // Postgres-specific string types
-  it("supports E'' strings with C-style escapes", () => {
-    expect(format("E'blah blah'")).toBe("E'blah blah'");
-    expect(format("E'some \\' FROM escapes'")).toBe("E'some \\' FROM escapes'");
-    expect(format("SELECT E'blah' FROM foo")).toBe(dedent`
-      SELECT
-        E'blah'
-      FROM
-        foo
-    `);
-    expect(format("E'blah''blah'")).toBe("E'blah''blah'");
-  });
-
   it('supports dollar-quoted strings', () => {
     expect(format('$xxx$foo $$ LEFT JOIN $yyy$ bar$xxx$')).toBe(
       '$xxx$foo $$ LEFT JOIN $yyy$ bar$xxx$'