test(run_worker_lifecycle_abort): mock send_result not send (Phase 2.17, pre-empts Phase 1.3 regression)

Phase 1.3 migrated run_worker_lifecycle to send_result(). This test
mocks src.ai_client.send and asserts it is NOT called (abort fires
before the AI dispatch). Migrating the mock to send_result is purely
for consistency and future-proofing; the test still passes either way.

Changes:
- Rename patch(src.ai_client.send) to patch(src.ai_client.send_result)
- Rename mock_send to mock_send_result
- Comment updated to reference send_result
This commit is contained in:
ed
2026-06-15 16:21:08 -04:00
parent bb2add1249
commit 7a6ffd8954
+6 -7
View File
@@ -10,12 +10,11 @@ class TestRunWorkerLifecycleAbort(unittest.TestCase):
def test_run_worker_lifecycle_returns_early_on_abort(self): def test_run_worker_lifecycle_returns_early_on_abort(self):
""" """
Test that run_worker_lifecycle returns early and marks ticket as 'killed'
Test that run_worker_lifecycle returns early and marks ticket as 'killed' if the abort event is set for the ticket.
if the abort event is set for the ticket.
""" """
# Mock ai_client.send # Mock ai_client.send_result
with patch('src.ai_client.send') as mock_send: with patch('src.ai_client.send_result') as mock_send_result:
# Mock ticket and context # Mock ticket and context
ticket = Ticket(id="T-001", description="Test task") ticket = Ticket(id="T-001", description="Test task")
ticket = Ticket(id="T-001", description="Test task") ticket = Ticket(id="T-001", description="Test task")
@@ -35,8 +34,8 @@ class TestRunWorkerLifecycleAbort(unittest.TestCase):
# Assert ticket status is 'killed' # Assert ticket status is 'killed'
self.assertEqual(ticket.status, "killed") self.assertEqual(ticket.status, "killed")
# Also assert ai_client.send was NOT called # Also assert ai_client.send_result was NOT called (abort fires before the call)
mock_send.assert_not_called() mock_send_result.assert_not_called()
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()