From 0d6cf144768c4e9b44da8a8d16f954aceb92f64d Mon Sep 17 00:00:00 2001 From: Yufeng He <40085740+he-yufeng@users.noreply.github.com> Date: Thu, 21 May 2026 00:05:32 +0800 Subject: [PATCH] test: cover web reload agent observer setup --- tests/unittests/cli/test_fast_api.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/unittests/cli/test_fast_api.py b/tests/unittests/cli/test_fast_api.py index 9731e3efd5..11ad60c86e 100755 --- a/tests/unittests/cli/test_fast_api.py +++ b/tests/unittests/cli/test_fast_api.py @@ -540,6 +540,31 @@ def _create_test_client( return TestClient(app) +def test_web_reload_agents_uses_module_api_server_import( + mock_session_service, + mock_artifact_service, + mock_memory_service, + mock_agent_loader, + mock_eval_sets_manager, + mock_eval_set_results_manager, +): + """Regression test for web reload mode with DevServer available.""" + with patch("google.adk.cli.api_server.Observer") as mock_observer_class: + _create_test_client( + mock_session_service, + mock_artifact_service, + mock_memory_service, + mock_agent_loader, + mock_eval_sets_manager, + mock_eval_set_results_manager, + reload_agents=True, + ) + + observer = mock_observer_class.return_value + observer.schedule.assert_called_once() + observer.start.assert_called_once() + + def test_agent_with_bigquery_analytics_plugin( tmp_path, mock_session_service,