Include WebSocket in sandbox, fixes cloudflare#30

ffflabs · Aug 22, 2021 · 633fedd · 633fedd
1 parent 1c0ed30
commit 633fedd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/modules/ws.ts b/src/modules/ws.ts
@@ -183,6 +183,7 @@ export class WebSocketsModule extends Module {
       MessageEvent,
       CloseEvent,
       ErrorEvent,
+      WebSocket,
       WebSocketPair,
     };
   }

diff --git a/test/modules/ws.spec.ts b/test/modules/ws.spec.ts
@@ -249,9 +249,10 @@ test("terminateWebSocket: throws if web socket already closed", async (t) => {
   });
 });
 
-test("buildSandbox: includes WebSocketPair", (t) => {
+test("buildSandbox: includes WebSocket and WebSocketPair", (t) => {
   const module = new WebSocketsModule(new NoOpLog());
   const sandbox = module.buildSandbox();
+  t.true(typeof sandbox.WebSocket === "function");
   t.true(typeof sandbox.WebSocketPair === "function");
 });
 test("buildSandbox: sends and responds to web socket messages", async (t) => {