File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -540,6 +540,31 @@ def _create_test_client(
540540 return TestClient (app )
541541
542542
543+ def test_web_reload_agents_uses_module_api_server_import (
544+ mock_session_service ,
545+ mock_artifact_service ,
546+ mock_memory_service ,
547+ mock_agent_loader ,
548+ mock_eval_sets_manager ,
549+ mock_eval_set_results_manager ,
550+ ):
551+ """Regression test for web reload mode with DevServer available."""
552+ with patch ("google.adk.cli.api_server.Observer" ) as mock_observer_class :
553+ _create_test_client (
554+ mock_session_service ,
555+ mock_artifact_service ,
556+ mock_memory_service ,
557+ mock_agent_loader ,
558+ mock_eval_sets_manager ,
559+ mock_eval_set_results_manager ,
560+ reload_agents = True ,
561+ )
562+
563+ observer = mock_observer_class .return_value
564+ observer .schedule .assert_called_once ()
565+ observer .start .assert_called_once ()
566+
567+
543568def test_agent_with_bigquery_analytics_plugin (
544569 tmp_path ,
545570 mock_session_service ,
You can’t perform that action at this time.
0 commit comments