This commit is contained in:
2026-03-05 17:13:59 -05:00
parent d4923c5198
commit fd36aad539
11 changed files with 48 additions and 48 deletions

View File

@@ -7,8 +7,8 @@ from src import mma_prompts
class TestOrchestratorPM(unittest.TestCase):
@patch('summarize.build_summary_markdown')
@patch('ai_client.send')
@patch('src.summarize.build_summary_markdown')
@patch('src.ai_client.send')
def test_generate_tracks_success(self, mock_send: Any, mock_summarize: Any) -> None:
# Setup mocks
mock_summarize.return_value = "REPO_MAP_CONTENT"
@@ -43,8 +43,8 @@ class TestOrchestratorPM(unittest.TestCase):
# Verify result
self.assertEqual(result[0]['id'], mock_response_data[0]['id'])
@patch('summarize.build_summary_markdown')
@patch('ai_client.send')
@patch('src.summarize.build_summary_markdown')
@patch('src.ai_client.send')
def test_generate_tracks_markdown_wrapped(self, mock_send: Any, mock_summarize: Any) -> None:
mock_summarize.return_value = "REPO_MAP"
mock_response_data = [{"id": "track_1"}]
@@ -58,8 +58,8 @@ class TestOrchestratorPM(unittest.TestCase):
result = orchestrator_pm.generate_tracks("req", {}, [])
self.assertEqual(result, expected_result)
@patch('summarize.build_summary_markdown')
@patch('ai_client.send')
@patch('src.summarize.build_summary_markdown')
@patch('src.ai_client.send')
def test_generate_tracks_malformed_json(self, mock_send: Any, mock_summarize: Any) -> None:
mock_summarize.return_value = "REPO_MAP"
mock_send.return_value = "NOT A JSON"