Helper to create a TestClient with the given get_fast_api_app overrides.
(
mock_session_service,
mock_artifact_service,
mock_memory_service,
mock_agent_loader,
mock_eval_sets_manager,
mock_eval_set_results_manager,
**app_kwargs,
)
| 478 | |
| 479 | |
| 480 | def _create_test_client( |
| 481 | mock_session_service, |
| 482 | mock_artifact_service, |
| 483 | mock_memory_service, |
| 484 | mock_agent_loader, |
| 485 | mock_eval_sets_manager, |
| 486 | mock_eval_set_results_manager, |
| 487 | **app_kwargs, |
| 488 | ): |
| 489 | """Helper to create a TestClient with the given get_fast_api_app overrides.""" |
| 490 | defaults = dict( |
| 491 | agents_dir=".", |
| 492 | web=True, |
| 493 | session_service_uri="", |
| 494 | artifact_service_uri="", |
| 495 | memory_service_uri="", |
| 496 | allow_origins=["*"], |
| 497 | a2a=False, |
| 498 | host="127.0.0.1", |
| 499 | port=8000, |
| 500 | ) |
| 501 | defaults.update(app_kwargs) |
| 502 | with ( |
| 503 | patch.object(signal, "signal", autospec=True, return_value=None), |
| 504 | patch.object( |
| 505 | fast_api_module, |
| 506 | "create_session_service_from_options", |
| 507 | autospec=True, |
| 508 | return_value=mock_session_service, |
| 509 | ), |
| 510 | patch.object( |
| 511 | fast_api_module, |
| 512 | "create_artifact_service_from_options", |
| 513 | autospec=True, |
| 514 | return_value=mock_artifact_service, |
| 515 | ), |
| 516 | patch.object( |
| 517 | fast_api_module, |
| 518 | "create_memory_service_from_options", |
| 519 | autospec=True, |
| 520 | return_value=mock_memory_service, |
| 521 | ), |
| 522 | patch.object( |
| 523 | fast_api_module, |
| 524 | "AgentLoader", |
| 525 | autospec=True, |
| 526 | return_value=mock_agent_loader, |
| 527 | ), |
| 528 | patch.object( |
| 529 | fast_api_module, |
| 530 | "LocalEvalSetsManager", |
| 531 | autospec=True, |
| 532 | return_value=mock_eval_sets_manager, |
| 533 | ), |
| 534 | patch.object( |
| 535 | fast_api_module, |
| 536 | "LocalEvalSetResultsManager", |
| 537 | autospec=True, |
no test coverage detected