diff --git a/tests/test_orchestrator_pm_history.py b/tests/test_orchestrator_pm_history.py index fec73cc0..cda9ce1b 100644 --- a/tests/test_orchestrator_pm_history.py +++ b/tests/test_orchestrator_pm_history.py @@ -59,13 +59,13 @@ class TestOrchestratorPMHistory(unittest.TestCase): self.assertIn("No overview available", summary) @patch('src.orchestrator_pm.summarize.build_summary_markdown') - @patch('src.ai_client.send_result') - def test_generate_tracks_with_history(self, mock_send_result: MagicMock, mock_summarize: MagicMock) -> None: + @patch('src.ai_client.send') + def test_generate_tracks_with_history(self, mock_send: MagicMock, mock_summarize: MagicMock) -> None: mock_summarize.return_value = "REPO_MAP" - mock_send_result.return_value = Result(data="[]") + mock_send.return_value = Result(data="[]") history_summary = "PAST_HISTORY_SUMMARY" orchestrator_pm.generate_tracks("req", {}, [], history_summary=history_summary) - args, kwargs = mock_send_result.call_args + args, kwargs = mock_send.call_args self.assertIn(history_summary, kwargs['user_message']) self.assertIn("### TRACK HISTORY:", kwargs['user_message'])