Function test_single_agent_mode_detection

tests/unittests/cli/test_fast_api.py:3023–3084

Verify that pointing agents_dir to a single agent folder enables single agent mode.

(
    tmp_path,
    mock_session_service,
    mock_artifact_service,
    mock_memory_service,
    mock_eval_sets_manager,
    mock_eval_set_results_manager,
)

def test_single_agent_mode_detection(
    tmp_path,
    mock_session_service,
    mock_artifact_service,
    mock_memory_service,
    mock_eval_sets_manager,
    mock_eval_set_results_manager,
):
  """Verify that pointing agents_dir to a single agent folder enables single agent mode."""
  agent_folder = tmp_path / "my_only_agent"
  agent_folder.mkdir()
  (agent_folder / "agent.py").write_text("root_agent = None")

  with (
      patch.object(signal, "signal", autospec=True, return_value=None),
      patch.object(
          fast_api_module,
          "create_session_service_from_options",
          autospec=True,
          return_value=mock_session_service,
      ),
      patch.object(
          fast_api_module,
          "create_artifact_service_from_options",
          autospec=True,
          return_value=mock_artifact_service,
      ),
      patch.object(
          fast_api_module,
          "create_memory_service_from_options",
          autospec=True,
          return_value=mock_memory_service,
      ),
      patch.object(
          fast_api_module,
          "LocalEvalSetsManager",
          autospec=True,
          return_value=mock_eval_sets_manager,
      ),
      patch.object(
          fast_api_module,
          "LocalEvalSetResultsManager",
          autospec=True,
          return_value=mock_eval_set_results_manager,
      ),
  ):
    app = get_fast_api_app(
        agents_dir=str(agent_folder),
        web=True,
        session_service_uri="",
        artifact_service_uri="",
        memory_service_uri="",
        allow_origins=None,
        a2a=False,
        host="127.0.0.1",
        port=8000,
    )
    client = TestClient(app)