From 870d95f35eff8396bd71d9016d538e9eda627577 Mon Sep 17 00:00:00 2001
From: "Josep Lluis Giralt D'Lacoste ( Pep )"
 <6720169+gilacost@users.noreply.github.com>
Date: Sat, 7 Mar 2026 21:35:18 +0700
Subject: [PATCH] fix(test): add missing getRuntimeMetrics mock to MockSession
 (#92)

The production code calls connection.session.getRuntimeMetrics() when
closing the WebSocket server, but MockSession in the relay-reconnect
tests didn't implement this method, causing all close() calls to throw.
---
 .../src/server/websocket-server.relay-reconnect.test.ts  | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/packages/server/src/server/websocket-server.relay-reconnect.test.ts b/packages/server/src/server/websocket-server.relay-reconnect.test.ts
index 786765cc0..2d1799b75 100644
--- a/packages/server/src/server/websocket-server.relay-reconnect.test.ts
+++ b/packages/server/src/server/websocket-server.relay-reconnect.test.ts
@@ -37,6 +37,15 @@ const sessionMock = vi.hoisted(() => {
     handleMessage = vi.fn(async () => {});
     handleBinaryFrame = vi.fn((_frame: unknown) => {});
     getClientActivity = vi.fn(() => null);
+    getRuntimeMetrics = vi.fn(() => ({
+      checkoutDiffTargetCount: 0,
+      checkoutDiffSubscriptionCount: 0,
+      checkoutDiffWatcherCount: 0,
+      checkoutDiffFallbackRefreshTargetCount: 0,
+      terminalDirectorySubscriptionCount: 0,
+      terminalSubscriptionCount: 0,
+      terminalStreamCount: 0,
+    }));
     readonly args: Record<string, unknown>;
 
     constructor(args: Record<string, unknown>) {