feat: add two-human AI autochess mode

2026-10-15 16:08:59 +08:00
parent e713ff369a
commit 9fff57134f
18 changed files with 228 additions and 41 deletions
@@ -0,0 +1,135 @@
+"""Behavioral coverage for the two-human plus two-AI test mode."""
+
+from __future__ import annotations
+
+import json
+import tempfile
+import unittest
+from pathlib import Path
+
+from test_jwebot_edge import MODULE, config
+from jwebot_official_extensions import cs_autochess as AUTOCHESS
+
+
+class Release065TwoHumanAiTests(unittest.TestCase):
+    def setUp(self):
+        self.temp = tempfile.TemporaryDirectory()
+        self.engine = MODULE.EdgeApplication(config(Path(self.temp.name)))
+        self.counter = 0
+
+    def tearDown(self):
+        self.engine.close()
+        self.temp.cleanup()
+
+    def command(self, question, *, sender="Alice", event_id=None):
+        self.counter += 1
+        return self.engine.local.handle(
+            {
+                "question": question,
+                "sender": sender,
+                "group_fingerprint": "group-a",
+                "event_id": event_id or f"two-ai-{self.counter}",
+            },
+            {},
+        )
+
+    def state(self):
+        row = self.engine.ledger.connection.execute(
+            """
+            SELECT state_json FROM autochess_runs
+            WHERE group_fingerprint = 'group-a'
+            ORDER BY season_number DESC
+            LIMIT 1
+            """
+        ).fetchone()
+        self.assertIsNotNone(row)
+        return json.loads(row[0])
+
+    def test_two_humans_add_two_ai_and_use_four_seed_bracket(self):
+        self.command("/自走棋开始")
+        self.command("/自走棋报名", sender="Bob")
+        added = self.command("/自走棋加AI战队")
+        self.assertIn("AI 战队 A、AI 战队 B", added)
+        self.assertIn("第 3、4 种子", added)
+        self.assertIn("2 人+2 AI", added)
+        self.assertIn(
+            "已经锁定参赛名单",
+            self.command("/自走棋报名", sender="Carol"),
+        )
+
+        launch = self.command("/自走棋开赛")
+        state = self.state()
+        self.assertIn("2 人+2 AI 测试赛", launch)
+        self.assertEqual(state["ai_test_variant"], "two_plus_two")
+        self.assertEqual(
+            state["seed_order"][2:],
+            [
+                AUTOCHESS.AI_PERSON_KEY,
+                AUTOCHESS.AI_SECOND_PERSON_KEY,
+            ],
+        )
+        self.assertEqual(
+            state["preparation_order"],
+            state["seed_order"][:2],
+        )
+        self.assertTrue(
+            state["players"][AUTOCHESS.AI_PERSON_KEY]["is_ai"]
+        )
+        self.assertTrue(
+            state["players"][AUTOCHESS.AI_SECOND_PERSON_KEY]["is_ai"]
+        )
+        semifinal = state["round_plan"][-2]
+        self.assertEqual(
+            semifinal["pairs"],
+            [
+                [state["seed_order"][0], state["seed_order"][3]],
+                [state["seed_order"][1], state["seed_order"][2]],
+            ],
+        )
+
+        self.assertIn("到 bob 做准备了", self.command("/结束回合"))
+        self.assertIn(
+            "全员准备完毕",
+            self.command("/结束回合", sender="Bob"),
+        )
+        settled = self.command("/自走棋结算")
+        self.assertIn("AI 战队 A 自动运营", settled)
+        self.assertIn("AI 战队 B 自动运营", settled)
+        self.assertEqual(
+            self.engine.ledger.connection.execute(
+                """
+                SELECT COUNT(*) FROM autochess_story_logs
+                WHERE story_type = 'ai_operation'
+                """
+            ).fetchone()[0],
+            2,
+        )
+
+    def test_add_two_ai_is_idempotent_and_cannot_be_repeated(self):
+        self.command("/自走棋开始")
+        self.command("/自走棋报名", sender="Bob")
+        first = self.command(
+            "/自走棋加AI战队",
+            event_id="same-add-two-ai",
+        )
+        replay = self.command(
+            "/自走棋加AI战队",
+            event_id="same-add-two-ai",
+        )
+        self.assertEqual(first, replay)
+        self.assertIn(
+            "已经加入",
+            self.command("/自走棋加AI战队"),
+        )
+        state = self.state()
+        self.assertEqual(
+            sum(
+                bool(player.get("is_ai"))
+                for player in state["players"].values()
+            ),
+            2,
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()