diff --git a/tests/test_four_phase_unit.py b/tests/test_four_phase_unit.py index 9da7b70..0dadcdc 100644 --- a/tests/test_four_phase_unit.py +++ b/tests/test_four_phase_unit.py @@ -210,6 +210,7 @@ class TestU3RoundComplete: mock_spawner.api_host = "127.0.0.1" mock_spawner.api_port = 8083 ticker = Ticker(registry=ProjectRegistry(Path("/tmp/fake")), + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") spawner=mock_spawner, dispatcher=MagicMock()) # mock _build_review_prompt(f-string 在 Python 3.9 有兼容性问题) ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") @@ -278,6 +279,7 @@ class TestU4RoundWithFailed: mock_spawner.api_host = "127.0.0.1" mock_spawner.api_port = 8083 ticker = Ticker(registry=MagicMock(), spawner=mock_spawner, + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") dispatcher=MagicMock()) async def run(): @@ -336,6 +338,7 @@ class TestU5RoundLimit: mock_spawner.api_host = "127.0.0.1" mock_spawner.api_port = 8083 ticker = Ticker(registry=MagicMock(), spawner=mock_spawner, + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") dispatcher=MagicMock()) async def run(): @@ -523,6 +526,7 @@ class TestU10ReviewPrompt: mock_spawner.api_host = "127.0.0.1" mock_spawner.api_port = 8083 ticker = Ticker(registry=MagicMock(), spawner=mock_spawner) + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") summary = {"done": 2, "failed": 1, "cancelled": 0, "total": 3} prompt = ticker._build_review_prompt( @@ -544,6 +548,7 @@ class TestU10ReviewPrompt: mock_spawner.api_host = "127.0.0.1" mock_spawner.api_port = 8083 ticker = Ticker(registry=MagicMock(), spawner=mock_spawner) + ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") summary = {"done": 1, "failed": 2, "cancelled": 0, "total": 3} prompt = ticker._build_review_prompt(