cleanup websocket test

1 files changed, 6 insertions, 3 deletions

diff --git a/integration/bunjs-only-snippets/websocket.test.js b/integration/bunjs-only-snippets/websocket.test.js
index 37c131591..a71f3dc7d 100644
--- a/integration/bunjs-only-snippets/websocket.test.js
+++ b/integration/bunjs-only-snippets/websocket.test.js
@@ -11,7 +11,11 @@ describe("WebSocket", () => {
       ws.onopen = resolve;
       ws.onerror = reject;
     });
+    var closed = new Promise((resolve, reject) => {
+      ws.onclose = resolve;
+    });
     ws.close();
+    await closed;
   });
 
   it("should send and receive messages", async () => {
@@ -20,8 +24,7 @@ describe("WebSocket", () => {
       ws.onopen = resolve;
       ws.onerror = reject;
       ws.onclose = () => {
-        expect(ws.bufferedAmount).toBe(0);
-        resolve();
+        reject("WebSocket closed");
       };
     });
     var promise = new Promise((resolve, reject) => {
@@ -31,8 +34,8 @@ describe("WebSocket", () => {
         resolve();
       };
       ws.onerror = reject;
+      ws.send("Hello World!");
     });
-    ws.send("Hello World");
 
     await promise;
   });