Skip to content

Commit 9300d48

Browse files
steipetesteipete
committed
refactor: trim telegram test harness exports
1 parent 3961f52 commit 9300d48

5 files changed

Lines changed: 20 additions & 29 deletions

‎extensions/telegram/src/bot-message-context.route-test-support.ts‎

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -9,33 +9,34 @@ type AsyncUnknownMock = Mock<(...args: unknown[]) => Promise<unknown>>;
99
type BuildTelegramMessageContextForTest =
1010
typeof import("./bot-message-context.test-harness.js").buildTelegramMessageContextForTest;
1111
type BuildTelegramMessageContextForTestParams = Parameters<BuildTelegramMessageContextForTest>[0];
12-
type TelegramTestSessionRuntime = NonNullable<
13-
import("./bot-message-context.types.js").BuildTelegramMessageContextParams["sessionRuntime"]
14-
>;
12+
type BuildTelegramMessageContextParams =
13+
import("./bot-message-context.types.js").BuildTelegramMessageContextParams;
1514

1615
const hoisted = vi.hoisted((): { recordInboundSessionMock: AsyncUnknownMock } => ({
1716
recordInboundSessionMock: vi.fn().mockResolvedValue(undefined),
1817
}));
1918

2019
export const recordInboundSessionMock: AsyncUnknownMock = hoisted.recordInboundSessionMock;
2120
const finalizeInboundContextForTest = finalizeTelegramInboundContextForTest as NonNullable<
22-
TelegramTestSessionRuntime["finalizeInboundContext"]
21+
NonNullable<BuildTelegramMessageContextParams["sessionRuntime"]>["finalizeInboundContext"]
2322
>;
2423
const recordInboundSessionForTest: NonNullable<
25-
TelegramTestSessionRuntime["recordInboundSession"]
24+
NonNullable<BuildTelegramMessageContextParams["sessionRuntime"]>["recordInboundSession"]
2625
> = async (params) => {
2726
await recordInboundSessionMock(params);
2827
};
2928

30-
export const telegramRouteTestSessionRuntime = {
29+
export const telegramRouteTestSessionRuntime: NonNullable<
30+
BuildTelegramMessageContextParams["sessionRuntime"]
31+
> = {
3132
finalizeInboundContext: finalizeInboundContextForTest,
3233
readSessionUpdatedAt: () => undefined,
3334
recordInboundSession: recordInboundSessionForTest,
3435
resolveInboundLastRouteSessionKey: ({ route, sessionKey }) =>
3536
route.lastRoutePolicy === "main" ? route.mainSessionKey : sessionKey,
3637
resolvePinnedMainDmOwnerFromAllowlist: () => null,
3738
resolveStorePath: () => "/tmp/openclaw/session-store.json",
38-
} satisfies TelegramTestSessionRuntime;
39+
};
3940

4041
export async function loadTelegramMessageContextRouteHarness() {
4142
const { buildTelegramMessageContextForTest } =

‎extensions/telegram/src/bot-native-commands.test-helpers.ts‎

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@ import type { TelegramAccountConfig } from "openclaw/plugin-sdk/config-types";
44
import type { MockFn } from "openclaw/plugin-sdk/plugin-test-runtime";
55
import type { RuntimeEnv } from "openclaw/plugin-sdk/runtime-env";
66
import { vi } from "vitest";
7-
import { createNativeCommandTestParams } from "./bot-native-commands.fixture-test-support.js";
87
import type { RegisterTelegramNativeCommandsParams } from "./bot-native-commands.js";
98
import { registerTelegramNativeCommands } from "./bot-native-commands.js";
109

@@ -42,10 +41,6 @@ const pluginCommandMocks = vi.hoisted(() => ({
4241
matchPluginCommand: vi.fn<MatchPluginCommandFn>(() => null),
4342
executePluginCommand: vi.fn<ExecutePluginCommandFn>(async () => ({ text: "ok" })),
4443
}));
45-
export const getPluginCommandSpecs = pluginCommandMocks.getPluginCommandSpecs;
46-
export const matchPluginCommand = pluginCommandMocks.matchPluginCommand;
47-
export const executePluginCommand = pluginCommandMocks.executePluginCommand;
48-
4944
vi.mock("openclaw/plugin-sdk/plugin-runtime", () => ({
5045
getPluginCommandSpecs: pluginCommandMocks.getPluginCommandSpecs,
5146
matchPluginCommand: pluginCommandMocks.matchPluginCommand,
@@ -74,13 +69,9 @@ const replyPipelineMocks = vi.hoisted(() => {
7469
getAgentScopedMediaLocalRoots: vi.fn<GetAgentScopedMediaLocalRootsFn>(() => []),
7570
};
7671
});
77-
export const dispatchReplyWithBufferedBlockDispatcher =
78-
replyPipelineMocks.dispatchReplyWithBufferedBlockDispatcher;
79-
8072
const deliveryMocks = vi.hoisted(() => ({
8173
deliverReplies: vi.fn(async () => {}),
8274
}));
83-
export const deliverReplies = deliveryMocks.deliverReplies;
8475

8576
vi.mock("./bot-native-commands.runtime.js", () => ({
8677
getPluginCommandSpecs: pluginCommandMocks.getPluginCommandSpecs,
@@ -120,7 +111,6 @@ vi.mock("openclaw/plugin-sdk/conversation-runtime", () => ({
120111
}));
121112
vi.mock("./bot/delivery.js", () => ({ deliverReplies: deliveryMocks.deliverReplies }));
122113
vi.mock("./bot/delivery.replies.js", () => ({ deliverReplies: deliveryMocks.deliverReplies }));
123-
export { createNativeCommandTestParams };
124114

125115
export function createNativeCommandsHarness(params?: {
126116
cfg?: OpenClawConfig;

‎extensions/telegram/src/bot.media.e2e-harness.ts‎

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@ type DispatchReplyWithBufferedBlockDispatcherFn =
1313
type DispatchReplyHarnessParams = Parameters<DispatchReplyWithBufferedBlockDispatcherFn>[0];
1414
type FetchRemoteMediaFn = typeof import("openclaw/plugin-sdk/media-runtime").fetchRemoteMedia;
1515

16-
export const useSpy: Mock = vi.fn();
17-
export const middlewareUseSpy: Mock = vi.fn();
16+
const useSpy: Mock = vi.fn();
17+
const middlewareUseSpy: Mock = vi.fn();
1818
export const onSpy: Mock = vi.fn();
19-
export const stopSpy: Mock = vi.fn();
19+
const stopSpy: Mock = vi.fn();
2020
export const sendChatActionSpy: Mock = vi.fn();
2121

2222
function defaultUndiciFetch(input: RequestInfo | URL, init?: RequestInit) {
@@ -25,7 +25,7 @@ function defaultUndiciFetch(input: RequestInfo | URL, init?: RequestInit) {
2525

2626
export const undiciFetchSpy: Mock = vi.fn(defaultUndiciFetch);
2727

28-
export function resetUndiciFetchMock() {
28+
function resetUndiciFetchMock() {
2929
undiciFetchSpy.mockReset();
3030
undiciFetchSpy.mockImplementation(defaultUndiciFetch);
3131
}
@@ -84,7 +84,7 @@ export function setNextSavedMediaPath(params: {
8484
);
8585
}
8686

87-
export function resetSaveMediaBufferMock() {
87+
function resetSaveMediaBufferMock() {
8888
saveMediaBufferSpy.mockReset();
8989
saveMediaBufferSpy.mockImplementation(defaultSaveMediaBuffer);
9090
}

‎extensions/telegram/src/draft-stream.test-helpers.ts‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { vi } from "vitest";
22

3-
export type TestDraftStream = {
3+
type TestDraftStream = {
44
update: ReturnType<typeof vi.fn<(text: string) => void>>;
55
flush: ReturnType<typeof vi.fn<() => Promise<void>>>;
66
messageId: ReturnType<typeof vi.fn<() => number | undefined>>;

‎extensions/telegram/src/target-writeback.test-shared.ts‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,15 @@ import { beforeAll, beforeEach, describe, expect, it, vi, type Mock } from "vite
44
type UnknownMock = Mock<(...args: unknown[]) => unknown>;
55
type AsyncUnknownMock = Mock<(...args: unknown[]) => Promise<unknown>>;
66

7-
export const readConfigFileSnapshotForWrite: AsyncUnknownMock = vi.fn();
8-
export const writeConfigFile: AsyncUnknownMock = vi.fn();
9-
export const replaceConfigFile: AsyncUnknownMock = vi.fn(async (params: unknown) => {
7+
const readConfigFileSnapshotForWrite: AsyncUnknownMock = vi.fn();
8+
const writeConfigFile: AsyncUnknownMock = vi.fn();
9+
const replaceConfigFile: AsyncUnknownMock = vi.fn(async (params: unknown) => {
1010
const record = params as { nextConfig?: unknown; writeOptions?: unknown };
1111
await writeConfigFile(record.nextConfig, record.writeOptions);
1212
});
13-
export const loadCronStore: AsyncUnknownMock = vi.fn();
14-
export const resolveCronStorePath: UnknownMock = vi.fn();
15-
export const saveCronStore: AsyncUnknownMock = vi.fn();
13+
const loadCronStore: AsyncUnknownMock = vi.fn();
14+
const resolveCronStorePath: UnknownMock = vi.fn();
15+
const saveCronStore: AsyncUnknownMock = vi.fn();
1616

1717
vi.mock("openclaw/plugin-sdk/config-mutation", async () => {
1818
const actual = await vi.importActual<typeof import("openclaw/plugin-sdk/config-mutation")>(

0 commit comments

Comments
 (0)