From b2a45790989a0a1c72fe1f8b72aff7f3f499b459 Mon Sep 17 00:00:00 2001
From: Grandis SYF <gra@dr.com>
Date: Mon, 20 Apr 2026 14:09:01 +0700
Subject: [PATCH] fix(proxy): update tests for tool_result+text and disconnect
 handler changes

---
 tests/unit/proxy/messages-route.test.ts                | 10 ++++------
 .../request-transformer-regressions.test.ts            |  4 ++--
 2 files changed, 6 insertions(+), 8 deletions(-)

diff --git a/tests/unit/proxy/messages-route.test.ts b/tests/unit/proxy/messages-route.test.ts
index 3ac40437..f118a794 100644
--- a/tests/unit/proxy/messages-route.test.ts
+++ b/tests/unit/proxy/messages-route.test.ts
@@ -31,17 +31,15 @@ describe('attachDisconnectAbortHandlers', () => {
     );
 
     expect(req.listenerCount('aborted')).toBe(1);
-    expect(req.listenerCount('close')).toBe(1);
+    expect(req.listenerCount('close')).toBe(0);
     expect(req.socket.listenerCount('close')).toBe(1);
-    expect(res.listenerCount('close')).toBe(1);
+    expect(res.listenerCount('close')).toBe(0);
     expect(res.socket.listenerCount('close')).toBe(1);
 
     cleanup();
 
     expect(req.listenerCount('aborted')).toBe(0);
-    expect(req.listenerCount('close')).toBe(0);
     expect(req.socket.listenerCount('close')).toBe(0);
-    expect(res.listenerCount('close')).toBe(0);
     expect(res.socket.listenerCount('close')).toBe(0);
   });
 
@@ -60,9 +58,9 @@ describe('attachDisconnectAbortHandlers', () => {
       }
     );
 
-    req.emit('close');
+    req.emit('aborted');
     req.socket.emit('close');
-    res.emit('close');
+    res.socket.emit('close');
 
     expect(controller.signal.aborted).toBe(true);
     expect(disconnectCount).toBe(1);
diff --git a/tests/unit/proxy/transformers/request-transformer-regressions.test.ts b/tests/unit/proxy/transformers/request-transformer-regressions.test.ts
index e9a05c18..cc182d11 100644
--- a/tests/unit/proxy/transformers/request-transformer-regressions.test.ts
+++ b/tests/unit/proxy/transformers/request-transformer-regressions.test.ts
@@ -120,7 +120,7 @@ describe('ProxyRequestTransformer regressions', () => {
           },
         ],
       })
-    ).toThrow('tool_result blocks must come before other user content');
+    ).not.toThrow();
 
     expect(() =>
       new ProxyRequestTransformer().transform({
@@ -135,7 +135,7 @@ describe('ProxyRequestTransformer regressions', () => {
           },
         ],
       })
-    ).toThrow('must start with tool_result blocks for pending tool_use ids');
+    ).toThrow('must include tool_result blocks for pending tool_use ids');
   });
 
   it('rejects tool_result content that cannot be represented as OpenAI tool text', () => {