diff --git a/packages/sdk-java/qwencode/src/test/java/com/alibaba/qwen/code/daemon/DaemonSessionClientTest.java b/packages/sdk-java/qwencode/src/test/java/com/alibaba/qwen/code/daemon/DaemonSessionClientTest.java index 2073583aab..d0fda9cfb5 100644 --- a/packages/sdk-java/qwencode/src/test/java/com/alibaba/qwen/code/daemon/DaemonSessionClientTest.java +++ b/packages/sdk-java/qwencode/src/test/java/com/alibaba/qwen/code/daemon/DaemonSessionClientTest.java @@ -261,11 +261,14 @@ class DaemonSessionClientTest { CountDownLatch cancelReceived = new CountDownLatch(1); server.createContext("/session/session-1/cancel", exchange -> { cancelReceived.countDown(); + exchange.getResponseHeaders().set("Connection", "close"); exchange.sendResponseHeaders(204, -1); exchange.close(); }); - server.createContext("/session/session-1/detach", noContent()); - server.createContext("/session/session-2/detach", noContent()); + server.createContext("/session/session-1/detach", + noContentAndCloseConnection()); + server.createContext("/session/session-2/detach", + noContentAndCloseConnection()); try (DaemonClient daemon = newClient(); DaemonSessionClient first = daemon.createSession()) { @@ -2949,6 +2952,14 @@ class DaemonSessionClientTest { }; } + private static HttpHandler noContentAndCloseConnection() { + return exchange -> { + exchange.getResponseHeaders().set("Connection", "close"); + exchange.sendResponseHeaders(204, -1); + exchange.close(); + }; + } + private static void sendJson(HttpExchange exchange, int status, String body) throws IOException { byte[] bytes = body.getBytes(StandardCharsets.UTF_8);