Private
Public Access
0
0

test(ai_client): rename send_result to send in test_conductor_tech_lead

11 references renamed (planned 8; the count grew with the @patch pattern + local var name).
Test file state: GREEN. 9 tests pass.
This commit is contained in:
2026-06-17 00:33:36 -04:00
parent 58fe3a9cb5
commit 423f9a95b0
+8 -8
View File
@@ -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")