From e1b6c9b29b52cbb48b83c54bd45d61a0c04febb5 Mon Sep 17 00:00:00 2001
From: Vincent Koc <vincentkoc@ieee.org>
Date: Wed, 1 Apr 2026 22:28:18 +0900
Subject: [PATCH] perf(test): trim more matrix and telegram reload churn

---
 extensions/matrix/src/matrix/send/client.test.ts | 8 +++++---
 extensions/telegram/src/allowed-updates.test.ts  | 5 ++---
 extensions/telegram/src/audit.test.ts            | 8 +++++---
 3 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/extensions/matrix/src/matrix/send/client.test.ts b/extensions/matrix/src/matrix/send/client.test.ts
index 514e42a3065..2e7555256fa 100644
--- a/extensions/matrix/src/matrix/send/client.test.ts
+++ b/extensions/matrix/src/matrix/send/client.test.ts
@@ -1,4 +1,4 @@
-import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
+import { afterEach, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
 import {
   createMockMatrixClient,
   expectExplicitMatrixClientConfig,
@@ -38,10 +38,12 @@ let withResolvedMatrixControlClient: typeof import("./client.js").withResolvedMa
 let withResolvedMatrixSendClient: typeof import("./client.js").withResolvedMatrixSendClient;
 
 describe("matrix send client helpers", () => {
-  beforeEach(async () => {
-    vi.resetModules();
+  beforeAll(async () => {
     ({ withResolvedMatrixControlClient, withResolvedMatrixSendClient } =
       await import("./client.js"));
+  });
+
+  beforeEach(() => {
     primeMatrixClientResolverMocks({
       resolved: {},
     });
diff --git a/extensions/telegram/src/allowed-updates.test.ts b/extensions/telegram/src/allowed-updates.test.ts
index 1e6534c1a53..35ad07718af 100644
--- a/extensions/telegram/src/allowed-updates.test.ts
+++ b/extensions/telegram/src/allowed-updates.test.ts
@@ -1,10 +1,9 @@
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { beforeAll, describe, expect, it, vi } from "vitest";
 let API_CONSTANTS: typeof import("grammy").API_CONSTANTS;
 let DEFAULT_TELEGRAM_UPDATE_TYPES: typeof import("./allowed-updates.js").DEFAULT_TELEGRAM_UPDATE_TYPES;
 let resolveTelegramAllowedUpdates: typeof import("./allowed-updates.js").resolveTelegramAllowedUpdates;
 
-beforeEach(async () => {
-  vi.resetModules();
+beforeAll(async () => {
   ({ API_CONSTANTS } = await import("grammy"));
   ({ DEFAULT_TELEGRAM_UPDATE_TYPES, resolveTelegramAllowedUpdates } =
     await import("./allowed-updates.js"));
diff --git a/extensions/telegram/src/audit.test.ts b/extensions/telegram/src/audit.test.ts
index d2e38db4864..37f9a12ea90 100644
--- a/extensions/telegram/src/audit.test.ts
+++ b/extensions/telegram/src/audit.test.ts
@@ -1,4 +1,4 @@
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
 
 let collectTelegramUnmentionedGroupIds: typeof import("./audit.js").collectTelegramUnmentionedGroupIds;
 let auditTelegramGroupMembership: typeof import("./audit.js").auditTelegramGroupMembership;
@@ -33,14 +33,16 @@ async function auditSingleGroup() {
 }
 
 describe("telegram audit", () => {
-  beforeEach(async () => {
-    vi.resetModules();
+  beforeAll(async () => {
     vi.doMock("./fetch.js", () => ({
       resolveTelegramApiBase: resolveTelegramApiBaseMock,
       resolveTelegramFetch: resolveTelegramFetchMock,
     }));
     ({ collectTelegramUnmentionedGroupIds, auditTelegramGroupMembership } =
       await import("./audit.js"));
+  });
+
+  beforeEach(() => {
     fetchWithTimeoutMock.mockReset();
     resolveTelegramFetchMock.mockClear();
     resolveTelegramApiBaseMock.mockClear();