Support multi-line SQL statements (#41)

ronin-co · Feb 5, 2025 · 261e125 · 261e125
1 parent cc91bd9
commit 261e125
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 1 deletion.
diff --git a/src/queries/statements.ts b/src/queries/statements.ts
@@ -30,7 +30,9 @@ export const getSyntaxProxySQL = (options: {
     });
 
     const statement: Statement = {
-      statement: text,
+      // Collapse whitespace and newlines into single spaces, then trim leading or
+      // trailing spaces.
+      statement: text.replace(/\s+/g, ' ').trim(),
       params,
     };
 

diff --git a/tests/queries/statements.test.ts b/tests/queries/statements.test.ts
@@ -43,3 +43,27 @@ test('using raw SQL in batch', async () => {
 
   expect(statementHandlerSpy).not.toHaveBeenCalled();
 });
+
+test('using raw SQL with multiple lines', async () => {
+  let statement: Statement | undefined;
+
+  const sqlProxy = getSyntaxProxySQL({
+    callback: (value) => {
+      statement = value;
+    },
+  });
+
+  const accountHandle = 'elaine';
+
+  sqlProxy`
+    UPDATE accounts
+    SET "points" = 11
+    WHERE "handle" = ${accountHandle}
+    RETURNING *
+  `;
+
+  expect(statement).toMatchObject({
+    statement: 'UPDATE accounts SET "points" = 11 WHERE "handle" = $1 RETURNING *',
+    params: ['elaine'],
+  });
+});