diff --git a/tests/test_conductor_tech_lead.py b/tests/test_conductor_tech_lead.py index c8155ef6..944154e1 100644 --- a/tests/test_conductor_tech_lead.py +++ b/tests/test_conductor_tech_lead.py @@ -6,23 +6,23 @@ import pytest class TestConductorTechLead(unittest.TestCase): def test_generate_tickets_retry_failure(self) -> None: - with patch('src.ai_client.send_result') as mock_send_result: - mock_send_result.return_value = Result(data="invalid json") + with patch('src.ai_client.send') as mock_send: + mock_send.return_value = Result(data="invalid json") # conductor_tech_lead.generate_tickets now raises RuntimeError on error after 3 attempts with pytest.raises(RuntimeError): conductor_tech_lead.generate_tickets("brief", "skeletons") - assert mock_send_result.call_count == 3 + assert mock_send.call_count == 3 def test_generate_tickets_retry_success(self) -> None: - with patch('src.ai_client.send_result') as mock_send_result: - mock_send_result.side_effect = [Result(data="invalid json"), Result(data='[{"Task": "Test"}]')] + with patch('src.ai_client.send') as mock_send: + mock_send.side_effect = [Result(data="invalid json"), Result(data='[{"Task": "Test"}]')] tickets = conductor_tech_lead.generate_tickets("brief", "skeletons") assert tickets == [{"Task": "Test"}] - assert mock_send_result.call_count == 2 + assert mock_send.call_count == 2 def test_generate_tickets_success(self) -> None: - with patch('src.ai_client.send_result') as mock_send_result: - mock_send_result.return_value = Result(data='[{"id": "T1", "description": "desc", "depends_on": []}]') + with patch('src.ai_client.send') as mock_send: + mock_send.return_value = Result(data='[{"id": "T1", "description": "desc", "depends_on": []}]') tickets = conductor_tech_lead.generate_tickets("brief", "skeletons") self.assertEqual(len(tickets), 1) self.assertEqual(tickets[0]['id'], "T1")