From 39af21ecbd06cb84ec76894a1ce33eb7eb1df498 Mon Sep 17 00:00:00 2001
From: Jarred Sumner <jarred@jarredsumner.com>
Date: Wed, 15 May 2024 22:27:04 -0700
Subject: [PATCH] Update websocket.test.js

---
 test/js/web/websocket/websocket.test.js | 23 +++++++++--------------
 1 file changed, 9 insertions(+), 14 deletions(-)

diff --git a/test/js/web/websocket/websocket.test.js b/test/js/web/websocket/websocket.test.js
index 4fb6db8960..741e84368d 100644
--- a/test/js/web/websocket/websocket.test.js
+++ b/test/js/web/websocket/websocket.test.js
@@ -58,7 +58,7 @@ describe("WebSocket", () => {
     Bun.gc(true);
   });
   it("should handle shutdown properly", async () => {
-    const server = Bun.serve({
+    using server = Bun.serve({
       port: 0,
       tls: COMMON_CERT,
       fetch(req, server) {
@@ -75,19 +75,14 @@ describe("WebSocket", () => {
         open(ws) {},
       },
     });
-    try {
-      for (let i = 0; i < 10_000; i++) {
-        const ws = new WebSocket(server.url.href, { tls: { rejectUnauthorized: false } });
-        await new Promise((resolve, reject) => {
-          ws.onopen = resolve;
-          ws.onerror = reject;
-        });
-        ws.send("message");
-      }
-      Bun.gc(true);
-    } finally {
-      server.stop(true);
-      Bun.gc(true);
+
+    for (let i = 0; i < 10_000; i++) {
+      const ws = new WebSocket(server.url.href, { tls: { rejectUnauthorized: false } });
+      await new Promise((resolve, reject) => {
+        ws.onopen = resolve;
+        ws.onerror = reject;
+      });
+      ws.send("message");
     }
   }, 60_000);