Set up test fixtures.
(self)
| 193 | """Tests for Runner._find_agent_to_run method.""" |
| 194 | |
| 195 | def setup_method(self): |
| 196 | """Set up test fixtures.""" |
| 197 | self.session_service = InMemorySessionService() |
| 198 | self.artifact_service = InMemoryArtifactService() |
| 199 | |
| 200 | # Create test agents |
| 201 | self.root_agent = MockLlmAgent("root_agent") |
| 202 | self.sub_agent1 = MockLlmAgent("sub_agent1", parent_agent=self.root_agent) |
| 203 | self.sub_agent2 = MockLlmAgent("sub_agent2", parent_agent=self.root_agent) |
| 204 | self.non_transferable_agent = MockLlmAgent( |
| 205 | "non_transferable", |
| 206 | disallow_transfer_to_parent=True, |
| 207 | parent_agent=self.root_agent, |
| 208 | ) |
| 209 | |
| 210 | self.root_agent.sub_agents = [ |
| 211 | self.sub_agent1, |
| 212 | self.sub_agent2, |
| 213 | self.non_transferable_agent, |
| 214 | ] |
| 215 | |
| 216 | self.runner = Runner( |
| 217 | app_name="test_app", |
| 218 | agent=self.root_agent, |
| 219 | session_service=self.session_service, |
| 220 | artifact_service=self.artifact_service, |
| 221 | ) |
| 222 | |
| 223 | def test_find_agent_to_run_with_function_response_scenario(self): |
| 224 | """Test finding agent when last event is function response.""" |
nothing calls this directly
no test coverage detected