From 87bfb6154442d11d25f39f0661484c989407dad3 Mon Sep 17 00:00:00 2001
From: Justin Spahr-Summers <justin@anthropic.com>
Date: Thu, 23 Jan 2025 11:53:41 +0000
Subject: [PATCH] Try a different way of waiting for close

---
 src/client/sse.test.ts | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/client/sse.test.ts b/src/client/sse.test.ts
index c1777ada..f59c45fe 100644
--- a/src/client/sse.test.ts
+++ b/src/client/sse.test.ts
@@ -1,7 +1,7 @@
-import { SSEClientTransport } from "./sse.js";
-import { createServer, type Server, type IncomingMessage } from "http";
-import { JSONRPCMessage } from "../types.js";
+import { createServer, type IncomingMessage, type Server } from "http";
 import { AddressInfo } from "net";
+import { JSONRPCMessage } from "../types.js";
+import { SSEClientTransport } from "./sse.js";
 
 describe("SSEClientTransport", () => {
   let server: Server;
@@ -56,10 +56,9 @@ describe("SSEClientTransport", () => {
     });
   });
 
-  afterEach((done) => {
-    transport?.close().then(() => {
-      server.close(done);
-    });
+  afterEach(async () => {
+    await transport.close();
+    await server.close();
   });
 
   describe("connection handling", () => {