Added WebSocket shim for http tests

neondatabase · Dec 18, 2024 · bcf0060 · bcf0060
1 parent afbca19
commit bcf0060
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/http.test.ts b/tests/http.test.ts
@@ -1,4 +1,4 @@
-import { expect, test, vi } from 'vitest';
+import { expect, test, vi, beforeAll } from 'vitest';
 import { neon, neonConfig, Pool } from '../dist/npm';
 import { sampleQueries } from './sampleQueries';
 
@@ -7,6 +7,13 @@ const sql = neon(DB_URL);
 const sqlFull = neon(DB_URL, { fullResults: true });
 const pool = new Pool({ connectionString: DB_URL });
 
+beforeAll(async () => {
+  if (typeof WebSocket !== 'function') {
+    const { WebSocket } = await import('ws');
+    neonConfig.webSocketConstructor = WebSocket;
+  }
+});
+
 test(
   'http query results match WebSocket query results',
   { timeout: 30000 },