mirror of
https://github.com/openclaw/openclaw.git
synced 2026-03-12 07:20:45 +00:00
test(gateway): widen before tool hook mock typing (#43476)
* test(gateway): widen before tool hook mock typing * chore: update pnpm.lock
This commit is contained in:
2
pnpm-lock.yaml
generated
2
pnpm-lock.yaml
generated
@@ -116,7 +116,7 @@ importers:
|
|||||||
specifier: ^5.2.1
|
specifier: ^5.2.1
|
||||||
version: 5.2.1
|
version: 5.2.1
|
||||||
file-type:
|
file-type:
|
||||||
specifier: ^21.3.1
|
specifier: 21.3.1
|
||||||
version: 21.3.1
|
version: 21.3.1
|
||||||
grammy:
|
grammy:
|
||||||
specifier: ^1.41.1
|
specifier: ^1.41.1
|
||||||
|
|||||||
@@ -1,14 +1,21 @@
|
|||||||
import { createServer, type IncomingMessage, type ServerResponse } from "node:http";
|
import { createServer, type IncomingMessage, type ServerResponse } from "node:http";
|
||||||
import type { AddressInfo } from "node:net";
|
import type { AddressInfo } from "node:net";
|
||||||
import { afterAll, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
|
import { afterAll, beforeAll, beforeEach, describe, expect, it, vi } from "vitest";
|
||||||
|
import type { runBeforeToolCallHook as runBeforeToolCallHookType } from "../agents/pi-tools.before-tool-call.js";
|
||||||
|
|
||||||
|
type RunBeforeToolCallHook = typeof runBeforeToolCallHookType;
|
||||||
|
type RunBeforeToolCallHookArgs = Parameters<RunBeforeToolCallHook>[0];
|
||||||
|
type RunBeforeToolCallHookResult = Awaited<ReturnType<RunBeforeToolCallHook>>;
|
||||||
|
|
||||||
const TEST_GATEWAY_TOKEN = "test-gateway-token-1234567890";
|
const TEST_GATEWAY_TOKEN = "test-gateway-token-1234567890";
|
||||||
const hookMocks = vi.hoisted(() => ({
|
const hookMocks = vi.hoisted(() => ({
|
||||||
resolveToolLoopDetectionConfig: vi.fn(() => ({ warnAt: 3 })),
|
resolveToolLoopDetectionConfig: vi.fn(() => ({ warnAt: 3 })),
|
||||||
runBeforeToolCallHook: vi.fn(async ({ params }: { params: unknown }) => ({
|
runBeforeToolCallHook: vi.fn(
|
||||||
blocked: false as const,
|
async (args: RunBeforeToolCallHookArgs): Promise<RunBeforeToolCallHookResult> => ({
|
||||||
params,
|
blocked: false,
|
||||||
})),
|
params: args.params,
|
||||||
|
}),
|
||||||
|
),
|
||||||
}));
|
}));
|
||||||
|
|
||||||
let cfg: Record<string, unknown> = {};
|
let cfg: Record<string, unknown> = {};
|
||||||
@@ -224,10 +231,12 @@ beforeEach(() => {
|
|||||||
hookMocks.resolveToolLoopDetectionConfig.mockClear();
|
hookMocks.resolveToolLoopDetectionConfig.mockClear();
|
||||||
hookMocks.resolveToolLoopDetectionConfig.mockImplementation(() => ({ warnAt: 3 }));
|
hookMocks.resolveToolLoopDetectionConfig.mockImplementation(() => ({ warnAt: 3 }));
|
||||||
hookMocks.runBeforeToolCallHook.mockClear();
|
hookMocks.runBeforeToolCallHook.mockClear();
|
||||||
hookMocks.runBeforeToolCallHook.mockImplementation(async ({ params }: { params: unknown }) => ({
|
hookMocks.runBeforeToolCallHook.mockImplementation(
|
||||||
|
async (args: RunBeforeToolCallHookArgs): Promise<RunBeforeToolCallHookResult> => ({
|
||||||
blocked: false,
|
blocked: false,
|
||||||
params,
|
params: args.params,
|
||||||
}));
|
}),
|
||||||
|
);
|
||||||
});
|
});
|
||||||
|
|
||||||
const resolveGatewayToken = (): string => TEST_GATEWAY_TOKEN;
|
const resolveGatewayToken = (): string => TEST_GATEWAY_TOKEN;
|
||||||
|
|||||||
Reference in New Issue
Block a user