checkpoint: mma_orchestrator track

tests/test_orchestrator_pm.py

@@ -0,0 +1,81 @@
+import unittest
+from unittest.mock import patch, MagicMock
+import json
+import orchestrator_pm
+import mma_prompts
+
+class TestOrchestratorPM(unittest.TestCase):
+
+    @patch('summarize.build_summary_markdown')
+    @patch('ai_client.send')
+    def test_generate_tracks_success(self, mock_send, mock_summarize):
+        # Setup mocks
+        mock_summarize.return_value = "REPO_MAP_CONTENT"
+        
+        mock_response_data = [
+            {
+                "id": "track_1",
+                "type": "Track",
+                "module": "test_module",
+                "persona": "Tech Lead",
+                "severity": "Medium",
+                "goal": "Test goal",
+                "acceptance_criteria": ["criteria 1"]
+            }
+        ]
+        mock_send.return_value = json.dumps(mock_response_data)
+        
+        user_request = "Implement unit tests"
+        project_config = {"files": {"paths": ["src"]}}
+        file_items = [{"path": "src/main.py", "content": "print('hello')"}]
+        
+        # Execute
+        result = orchestrator_pm.generate_tracks(user_request, project_config, file_items)
+        
+        # Verify summarize call
+        mock_summarize.assert_called_once_with(file_items)
+        
+        # Verify ai_client.send call
+        expected_system_prompt = mma_prompts.PROMPTS['tier1_epic_init']
+        mock_send.assert_called_once()
+        args, kwargs = mock_send.call_args
+        self.assertEqual(kwargs['md_content'], "")
+        self.assertEqual(kwargs['system_prompt'], expected_system_prompt)
+        self.assertIn(user_request, kwargs['user_message'])
+        self.assertIn("REPO_MAP_CONTENT", kwargs['user_message'])
+        self.assertEqual(kwargs['model_name'], "gemini-1.5-pro")
+        
+        # Verify result
+        self.assertEqual(result, mock_response_data)
+
+    @patch('summarize.build_summary_markdown')
+    @patch('ai_client.send')
+    def test_generate_tracks_markdown_wrapped(self, mock_send, mock_summarize):
+        mock_summarize.return_value = "REPO_MAP"
+        
+        mock_response_data = [{"id": "track_1"}]
+        # Wrapped in ```json ... ```
+        mock_send.return_value = f"Here is the plan:\n```json\n{json.dumps(mock_response_data)}\n```\nHope this helps."
+        
+        result = orchestrator_pm.generate_tracks("req", {}, [])
+        self.assertEqual(result, mock_response_data)
+        
+        # Wrapped in ``` ... ```
+        mock_send.return_value = f"```\n{json.dumps(mock_response_data)}\n```"
+        result = orchestrator_pm.generate_tracks("req", {}, [])
+        self.assertEqual(result, mock_response_data)
+
+    @patch('summarize.build_summary_markdown')
+    @patch('ai_client.send')
+    def test_generate_tracks_malformed_json(self, mock_send, mock_summarize):
+        mock_summarize.return_value = "REPO_MAP"
+        mock_send.return_value = "NOT A JSON"
+        
+        # Should return empty list and print error (we can mock print if we want to be thorough)
+        with patch('builtins.print') as mock_print:
+            result = orchestrator_pm.generate_tracks("req", {}, [])
+            self.assertEqual(result, [])
+            mock_print.assert_any_call("Error parsing Tier 1 response: Expecting value: line 1 column 1 (char 0)")
+
+if __name__ == '__main__':
+    unittest.main()