Function simulateCallToolAsTask

server/tests/unit/training-agent.test.ts:116–131

* Simulate a task-augmented CallTool request — includes the `task` field in params.

(
  server: ReturnType<typeof createTrainingAgentServer>,
  toolName: string,
  args: Record<string, unknown>,
  taskParams: { ttl?: number } = {},
)

 * Simulate a task-augmented CallTool request — includes the `task` field in params.
 */
async function simulateCallToolAsTask(
  server: ReturnType<typeof createTrainingAgentServer>,
  toolName: string,
  args: Record<string, unknown>,
  taskParams: { ttl?: number } = {},
): Promise<Record<string, unknown>> {
  const requestHandlers = (server as any)._requestHandlers as Map<string, Function>;
  const handler = requestHandlers.get('tools/call');
  if (!handler) {
    throw new Error('CallTool handler not found');
  }
  return handler(
    { method: 'tools/call', params: { name: toolName, arguments: withIdempotencyKey(toolName, args), task: taskParams } },
    {},
  );
}

/**
 * Simulate a tasks/get request.