diff --git a/clients/web/src/App.test.tsx b/clients/web/src/App.test.tsx index c66cb03b..75faad77 100644 --- a/clients/web/src/App.test.tsx +++ b/clients/web/src/App.test.tsx @@ -41,6 +41,10 @@ vi.mock("@inspector/core/mcp/index.js", () => { callTool = vi .fn() .mockResolvedValue({ success: true, result: { acts: [] } }); + callToolStream = vi + .fn() + .mockResolvedValue({ success: true, result: { acts: [] } }); + cancelRequestorTask = vi.fn().mockResolvedValue(undefined); getPrompt = vi.fn().mockResolvedValue({ result: { messages: [] } }); readResource = vi .fn() @@ -166,7 +170,11 @@ vi.mock("@inspector/core/react/useManagedResourceTemplates.js", () => ({ useManagedResourceTemplates: vi.fn(() => ({ resourceTemplates: [] })), })); vi.mock("@inspector/core/react/useManagedRequestorTasks.js", () => ({ - useManagedRequestorTasks: vi.fn(() => ({ tasks: [], refresh: vi.fn() })), + useManagedRequestorTasks: vi.fn(() => ({ + tasks: [], + refresh: vi.fn().mockResolvedValue([]), + clearCompleted: vi.fn(), + })), })); vi.mock("@inspector/core/react/useResourceSubscriptions.js", () => ({ useResourceSubscriptions: vi.fn(() => ({ subscriptions: [] })), @@ -223,15 +231,26 @@ vi.mock("./components/views/InspectorView/InspectorView", () => ({ filterText: string; visibleLevels: Record; }) => void; - onCallTool: (name: string, args: Record) => void; + progressByTaskId?: Record; + onCallTool: ( + name: string, + args: Record, + runAsTask?: boolean, + ) => void; onGetPrompt: (name: string, args: Record) => void; onReadResource: (uri: string) => void; onSetLogLevel: (level: string) => void; + onCancelTask: (taskId: string) => void; + onClearCompletedTasks: () => void; + onRefreshTasks: () => void; }) => (
{props.toolCallState?.status ?? "none"} + + {Object.keys(props.progressByTaskId ?? {}).join(",") || "none"} + {props.toolsUi?.selectedToolName ?? "none"} @@ -307,6 +326,14 @@ vi.mock("./components/views/InspectorView/InspectorView", () => ({ set-log-filter + + + +