Function _make_test_runner

tests/unittests/telemetry/test_telemetry_context.py:568–581

Builds an InMemoryRunner with a deterministic MockModel. Each invocation drives one ``generate_content`` call returning a short text response, which is enough to exercise the ``use_inference_span`` + ``trace_inference_result`` code path that consumes ``RunConfig.telemetry``.

(
    agent_name: str = 'telemetry_test_agent',
)

def _make_test_runner(
    agent_name: str = 'telemetry_test_agent',
) -> InMemoryRunner:
  """Builds an InMemoryRunner with a deterministic MockModel.

  Each invocation drives one ``generate_content`` call returning a short
  text response, which is enough to exercise the ``use_inference_span`` +
  ``trace_inference_result`` code path that consumes ``RunConfig.telemetry``.
  """
  mock_model = MockModel.create(
      responses=[Part.from_text(text='ok')],
  )
  agent = Agent(name=agent_name, model=mock_model)
  return InMemoryRunner(root_agent=agent)


@pytest.fixture