From 1512fab782806ee4f8d32f3a87e82cf10dea3293 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Fri, 3 Apr 2026 17:33:22 +0100
Subject: [PATCH] test: remove discord channel partial mocks

---
 extensions/discord/src/channel.test.ts | 25 +++++++++++++------------
 1 file changed, 13 insertions(+), 12 deletions(-)

diff --git a/extensions/discord/src/channel.test.ts b/extensions/discord/src/channel.test.ts
index c6fd99ddfac..469d9d1db8f 100644
--- a/extensions/discord/src/channel.test.ts
+++ b/extensions/discord/src/channel.test.ts
@@ -9,37 +9,33 @@ let setDiscordRuntime: typeof import("./runtime.js").setDiscordRuntime;
 const probeDiscordMock = vi.hoisted(() => vi.fn());
 const monitorDiscordProviderMock = vi.hoisted(() => vi.fn());
 const auditDiscordChannelPermissionsMock = vi.hoisted(() => vi.fn());
+const collectDiscordAuditChannelIdsMock = vi.hoisted(() =>
+  vi.fn(() => ({ channelIds: [], unresolvedChannels: 0 })),
+);
 const sleepWithAbortMock = vi.hoisted(() => vi.fn(async () => undefined));
 
-vi.mock("openclaw/plugin-sdk/runtime-env", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("openclaw/plugin-sdk/runtime-env")>();
+vi.mock("openclaw/plugin-sdk/runtime-env", () => {
   return {
-    ...actual,
     sleepWithAbort: sleepWithAbortMock,
   };
 });
 
-vi.mock("./probe.js", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("./probe.js")>();
+vi.mock("./probe.js", () => {
   return {
-    ...actual,
     probeDiscord: probeDiscordMock,
   };
 });
 
-vi.mock("./monitor/provider.runtime.js", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("./monitor/provider.runtime.js")>();
+vi.mock("./monitor/provider.runtime.js", () => {
   return {
-    ...actual,
     monitorDiscordProvider: monitorDiscordProviderMock,
   };
 });
 
-vi.mock("./audit.js", async (importOriginal) => {
-  const actual = await importOriginal<typeof import("./audit.js")>();
+vi.mock("./audit.js", () => {
   return {
-    ...actual,
     auditDiscordChannelPermissions: auditDiscordChannelPermissionsMock,
+    collectDiscordAuditChannelIds: collectDiscordAuditChannelIdsMock,
   };
 });
 
@@ -82,6 +78,11 @@ afterEach(() => {
   probeDiscordMock.mockReset();
   monitorDiscordProviderMock.mockReset();
   auditDiscordChannelPermissionsMock.mockReset();
+  collectDiscordAuditChannelIdsMock.mockReset();
+  collectDiscordAuditChannelIdsMock.mockReturnValue({
+    channelIds: [],
+    unresolvedChannels: 0,
+  });
   sleepWithAbortMock.mockReset();
   sleepWithAbortMock.mockResolvedValue(undefined);
 });