From e4e915d2ecfb7593cff8fb0037625349f85d43ca Mon Sep 17 00:00:00 2001
From: Jarred Sumner <709451+Jarred-Sumner@users.noreply.github.com>
Date: Wed, 22 Jun 2022 21:11:28 -0700
Subject: cleanup websocket test

---
 integration/bunjs-only-snippets/websocket.test.js | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'integration/bunjs-only-snippets/websocket.test.js')

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;
   });
-- 
cgit v1.2.3