From 7c4bffdecd3ebb45a752f024e7b17901b45fde3f Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Tue, 31 Mar 2026 23:29:24 +0900
Subject: [PATCH] fix(ci): rebalance telegram dm thread tests

---
 .../src/bot-message-context.dm-threads.test.ts    | 15 ++++++++++++++-
 test/fixtures/test-timings.channels.json          |  4 ++--
 2 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/extensions/telegram/src/bot-message-context.dm-threads.test.ts b/extensions/telegram/src/bot-message-context.dm-threads.test.ts
index 78e4d067cb4..68855885bf2 100644
--- a/extensions/telegram/src/bot-message-context.dm-threads.test.ts
+++ b/extensions/telegram/src/bot-message-context.dm-threads.test.ts
@@ -1,4 +1,16 @@
-import { afterEach, beforeEach, describe, expect, it } from "vitest";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+const { recordInboundSessionMock } = vi.hoisted(() => ({
+  recordInboundSessionMock: vi.fn().mockResolvedValue(undefined),
+}));
+
+vi.mock("openclaw/plugin-sdk/conversation-runtime", async (importOriginal) => {
+  const actual = await importOriginal<typeof import("openclaw/plugin-sdk/conversation-runtime")>();
+  return {
+    ...actual,
+    recordInboundSession: (...args: unknown[]) => recordInboundSessionMock(...args),
+  };
+});
+
 const { buildTelegramMessageContextForTest } =
   await import("./bot-message-context.test-harness.js");
 const { clearRuntimeConfigSnapshot, setRuntimeConfigSnapshot } =
@@ -10,6 +22,7 @@ beforeEach(() => {
 
 afterEach(() => {
   clearRuntimeConfigSnapshot();
+  recordInboundSessionMock.mockClear();
 });
 
 describe("buildTelegramMessageContext dm thread sessions", () => {
diff --git a/test/fixtures/test-timings.channels.json b/test/fixtures/test-timings.channels.json
index 646b91d62fd..244b3f5bda7 100644
--- a/test/fixtures/test-timings.channels.json
+++ b/test/fixtures/test-timings.channels.json
@@ -121,7 +121,7 @@
       "durationMs": 1600
     },
     "extensions/telegram/src/bot-message-context.dm-threads.test.ts": {
-      "durationMs": 1600
+      "durationMs": 60770
     },
     "extensions/whatsapp/src/inbound/access-control.test.ts": {
       "durationMs": 1600
@@ -241,7 +241,7 @@
       "durationMs": 610
     },
     "extensions/telegram/src/bot-message-context.dm-topic-threadid.test.ts": {
-      "durationMs": 581
+      "durationMs": 73540
     },
     "extensions/slack/src/monitor.threading.missing-thread-ts.test.ts": {
       "durationMs": 570