diff --git a/tests/test_four_phase_unit.py b/tests/test_four_phase_unit.py index d7df249..a7ec99f 100644 --- a/tests/test_four_phase_unit.py +++ b/tests/test_four_phase_unit.py @@ -210,9 +210,9 @@ 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") async def run(): result = await ticker._check_round_complete(db_path, pid) @@ -242,7 +242,6 @@ class TestU3RoundComplete: ticker = Ticker(registry=MagicMock(), spawner=MagicMock(), dispatcher=MagicMock()) - ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") async def run(): return await ticker._check_round_complete(db_path, pid) @@ -279,7 +278,6 @@ 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(): @@ -313,7 +311,6 @@ class TestU5RoundLimit: ticker = Ticker(registry=MagicMock(), spawner=MagicMock(), dispatcher=MagicMock()) - ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") async def run(): return await ticker._check_round_complete(db_path, pid) @@ -339,7 +336,6 @@ 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(): @@ -384,7 +380,6 @@ class TestU6MentionGrouping: ticker = Ticker(registry=MagicMock(), spawner=mock_spawner, dispatcher=MagicMock()) - ticker._build_review_prompt = MagicMock(return_value="Review prompt mock") async def run(): return await ticker._process_mentions(db_path, pid) @@ -533,7 +528,6 @@ 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( @@ -557,7 +551,6 @@ 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(