f64d25bd3e
* fix(telegram): scope DM topic thread keys by chat id * test(telegram): update dm topic session-key expectation * fix(telegram): parse scoped dm thread ids in outbound recovery * chore(telegram): format accounts config merge block * test(nodes): simplify mocked exports for ts tuple spreads
89 lines
2.6 KiB
TypeScript
89 lines
2.6 KiB
TypeScript
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
const gatewayMocks = vi.hoisted(() => ({
|
|
callGatewayTool: vi.fn(),
|
|
readGatewayCallOptions: vi.fn(() => ({})),
|
|
}));
|
|
|
|
const nodeUtilsMocks = vi.hoisted(() => ({
|
|
resolveNodeId: vi.fn(async () => "node-1"),
|
|
listNodes: vi.fn(async () => []),
|
|
resolveNodeIdFromList: vi.fn(() => "node-1"),
|
|
}));
|
|
|
|
const screenMocks = vi.hoisted(() => ({
|
|
parseScreenRecordPayload: vi.fn(() => ({
|
|
base64: "ZmFrZQ==",
|
|
format: "mp4",
|
|
durationMs: 300_000,
|
|
fps: 10,
|
|
screenIndex: 0,
|
|
hasAudio: true,
|
|
})),
|
|
screenRecordTempPath: vi.fn(() => "/tmp/screen-record.mp4"),
|
|
writeScreenRecordToFile: vi.fn(async () => ({ path: "/tmp/screen-record.mp4" })),
|
|
}));
|
|
|
|
vi.mock("./gateway.js", () => ({
|
|
callGatewayTool: gatewayMocks.callGatewayTool,
|
|
readGatewayCallOptions: gatewayMocks.readGatewayCallOptions,
|
|
}));
|
|
|
|
vi.mock("./nodes-utils.js", () => ({
|
|
resolveNodeId: nodeUtilsMocks.resolveNodeId,
|
|
listNodes: nodeUtilsMocks.listNodes,
|
|
resolveNodeIdFromList: nodeUtilsMocks.resolveNodeIdFromList,
|
|
}));
|
|
|
|
vi.mock("../../cli/nodes-screen.js", () => ({
|
|
parseScreenRecordPayload: screenMocks.parseScreenRecordPayload,
|
|
screenRecordTempPath: screenMocks.screenRecordTempPath,
|
|
writeScreenRecordToFile: screenMocks.writeScreenRecordToFile,
|
|
}));
|
|
|
|
import { createNodesTool } from "./nodes-tool.js";
|
|
|
|
describe("createNodesTool screen_record duration guardrails", () => {
|
|
beforeEach(() => {
|
|
gatewayMocks.callGatewayTool.mockReset();
|
|
gatewayMocks.readGatewayCallOptions.mockReset();
|
|
gatewayMocks.readGatewayCallOptions.mockReturnValue({});
|
|
nodeUtilsMocks.resolveNodeId.mockClear();
|
|
screenMocks.parseScreenRecordPayload.mockClear();
|
|
screenMocks.writeScreenRecordToFile.mockClear();
|
|
});
|
|
|
|
it("caps durationMs schema at 300000", () => {
|
|
const tool = createNodesTool();
|
|
const schema = tool.parameters as {
|
|
properties?: {
|
|
durationMs?: {
|
|
maximum?: number;
|
|
};
|
|
};
|
|
};
|
|
expect(schema.properties?.durationMs?.maximum).toBe(300_000);
|
|
});
|
|
|
|
it("clamps screen_record durationMs argument to 300000 before gateway invoke", async () => {
|
|
gatewayMocks.callGatewayTool.mockResolvedValue({ payload: { ok: true } });
|
|
const tool = createNodesTool();
|
|
|
|
await tool.execute("call-1", {
|
|
action: "screen_record",
|
|
node: "macbook",
|
|
durationMs: 900_000,
|
|
});
|
|
|
|
expect(gatewayMocks.callGatewayTool).toHaveBeenCalledWith(
|
|
"node.invoke",
|
|
{},
|
|
expect.objectContaining({
|
|
params: expect.objectContaining({
|
|
durationMs: 300_000,
|
|
}),
|
|
}),
|
|
);
|
|
});
|
|
});
|