Skip to content

Commit ba623a0

Browse files
author
Your Name
committed
Mock turn count for yield tests
1 parent 2526278 commit ba623a0

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

tests/subagents/test_finished.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ async def test_sub_agent_with_summary_updates_info(self):
3434
mock_coder.uuid = "sub-uuid"
3535
mock_coder.parent_uuid = "parent-uuid"
3636
mock_coder.files_edited_by_tools = set()
37+
mock_coder.turn_count = 0
3738

3839
mock_info = MagicMock()
3940
mock_info.coder.uuid = "sub-uuid"
@@ -90,6 +91,7 @@ async def test_unknown_parent_uuid_caught_gracefully(self):
9091
mock_coder.uuid = "sub-uuid"
9192
mock_coder.parent_uuid = "nonexistent-parent"
9293
mock_coder.files_edited_by_tools = set()
94+
mock_coder.turn_count = 0
9395

9496
with (
9597
patch.object(AgentService, "_instances", {}),
@@ -106,6 +108,7 @@ async def test_returns_summary_in_response(self):
106108
mock_coder.uuid = "test-uuid"
107109
mock_coder.parent_uuid = ""
108110
mock_coder.files_edited_by_tools = set()
111+
mock_coder.turn_count = 0
109112

110113
result = await Tool.execute(mock_coder, summary="completed successfully")
111114
assert "Summary: completed successfully" in str(result)

0 commit comments

Comments
 (0)