Function _build_mcp_test_runner

tests/unittests/telemetry/test_functional.py:530–547

Builds a single-turn agent runner whose only tool source is ``toolset``. Single-turn (one ``Part.from_text`` response) so the assertion on ``list_tools_call_count`` is unambiguous: exactly one agent invocation is performed.

(toolset: McpToolset)

def _build_mcp_test_runner(toolset: McpToolset) -> TestInMemoryRunner:
  """Builds a single-turn agent runner whose only tool source is ``toolset``.

  Single-turn (one ``Part.from_text`` response) so the assertion on
  ``list_tools_call_count`` is unambiguous: exactly one agent invocation
  is performed.
  """
  mock_model = MockModel.create(
      responses=[Part.from_text(text="text response")]
  )
  test_agent = Agent(
      name="some_root_agent",
      description="A sample root agent.",
      instruction="you are helpful",
      model=mock_model,
      tools=[toolset],
  )
  return TestInMemoryRunner(node=test_agent)


@pytest.mark.asyncio