diff --git a/src/server/streamableHttp.test.ts b/src/server/streamableHttp.test.ts index dda69331..949cafc7 100644 --- a/src/server/streamableHttp.test.ts +++ b/src/server/streamableHttp.test.ts @@ -72,8 +72,11 @@ async function createTestServer(config: TestServerConfig = {}): Promise<{ * Helper to stop test server */ async function stopTestServer({ server, transport }: { server: Server; transport: StreamableHTTPServerTransport }): Promise { + // First close the transport to ensure all SSE streams are closed await transport.close(); - await new Promise((resolve) => server.close(() => resolve())); + + // Close the server without waiting indefinitely + server.close(); } /** @@ -649,8 +652,8 @@ describe("StreamableHTTPServerTransport", () => { expect(deleteResponse.status).toBe(200); - // Clean up - await new Promise((resolve) => tempServer.close(() => resolve())); + // Clean up - don't wait indefinitely for server close + tempServer.close(); }); it("should reject DELETE requests with invalid session ID", async () => {