File size: 3,467 Bytes
eb3f11e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 | import { vi } from "vitest";
import type { PluginRuntime } from "../../plugins/runtime/types.js";
type BoundTaskFlowRuntime = ReturnType<PluginRuntime["tasks"]["managedFlows"]["bindSession"]>;
type BoundAsyncManagedTaskFlows = ReturnType<
PluginRuntime["tasks"]["async"]["managedFlows"]["bindSession"]
>;
function createTaskFlowSessionMock(): BoundTaskFlowRuntime {
return {
sessionKey: "agent:main:main",
createManaged: vi.fn<BoundTaskFlowRuntime["createManaged"]>(),
tryCreateManaged: vi.fn<BoundTaskFlowRuntime["tryCreateManaged"]>(),
get: vi.fn<BoundTaskFlowRuntime["get"]>(),
list: vi.fn<BoundTaskFlowRuntime["list"]>(() => []),
findLatest: vi.fn<BoundTaskFlowRuntime["findLatest"]>(),
resolve: vi.fn<BoundTaskFlowRuntime["resolve"]>(),
getTaskSummary: vi.fn<BoundTaskFlowRuntime["getTaskSummary"]>(),
setWaiting: vi.fn<BoundTaskFlowRuntime["setWaiting"]>(),
resume: vi.fn<BoundTaskFlowRuntime["resume"]>(),
finish: vi.fn<BoundTaskFlowRuntime["finish"]>(),
fail: vi.fn<BoundTaskFlowRuntime["fail"]>(),
requestCancel: vi.fn<BoundTaskFlowRuntime["requestCancel"]>(),
cancel: vi.fn<BoundTaskFlowRuntime["cancel"]>(),
runTask: vi.fn<BoundTaskFlowRuntime["runTask"]>(),
};
}
function createAsyncReadSession(params: { sessionKey?: string }) {
return {
sessionKey: params.sessionKey ?? "agent:main:main",
get: vi.fn(async () => undefined),
list: vi.fn(async () => []),
findLatest: vi.fn(async () => undefined),
resolve: vi.fn(async () => undefined),
};
}
function createAsyncFlowReadSession(params: { sessionKey?: string }) {
return { ...createAsyncReadSession(params), getTaskSummary: vi.fn(async () => undefined) };
}
function createAsyncManagedFlowSession(params: {
sessionKey?: string;
}): BoundAsyncManagedTaskFlows {
const missing = { applied: false, code: "not_found" } as const;
return {
...createAsyncFlowReadSession(params),
createManaged: vi.fn<BoundAsyncManagedTaskFlows["createManaged"]>(),
tryCreateManaged: vi.fn(async () => null),
setWaiting: vi.fn(async () => missing),
resume: vi.fn(async () => missing),
finish: vi.fn(async () => missing),
fail: vi.fn(async () => missing),
requestCancel: vi.fn(async () => missing),
runTask: vi.fn(async () => ({
created: false as const,
found: false,
reason: "Flow not found.",
})),
};
}
function readBinding<Bound>(factory: (params: { sessionKey?: string }) => Bound) {
return { bindSession: vi.fn(factory), fromToolContext: vi.fn(factory) };
}
export function createPluginTasksRuntimeMock(): PluginRuntime["tasks"] {
return {
async: {
runs: readBinding(createAsyncReadSession),
flows: readBinding(createAsyncFlowReadSession),
managedFlows: readBinding(createAsyncManagedFlowSession),
},
runs: {
bindSession: vi.fn<PluginRuntime["tasks"]["runs"]["bindSession"]>(),
fromToolContext: vi.fn<PluginRuntime["tasks"]["runs"]["fromToolContext"]>(),
},
flows: {
bindSession: vi.fn<PluginRuntime["tasks"]["flows"]["bindSession"]>(),
fromToolContext: vi.fn<PluginRuntime["tasks"]["flows"]["fromToolContext"]>(),
},
managedFlows: {
bindSession:
vi.fn<PluginRuntime["tasks"]["managedFlows"]["bindSession"]>(createTaskFlowSessionMock),
fromToolContext:
vi.fn<PluginRuntime["tasks"]["managedFlows"]["fromToolContext"]>(createTaskFlowSessionMock),
},
};
}
|