feat: 斗鱼外部单协作台账与店铺实时同步

2026-10-20 20:11:35 +08:00
parent c778ae9a09
commit 71574908ff
18 changed files with 6187 additions and 4 deletions
@@ -0,0 +1,395 @@
+"""斗鱼外部单协作：账号导入、状态机、鱼翅/积分与兑换记录查询。"""
+
+import pytest
+
+from tests.support.db import get_test_engine, make_test_session
+from web.backend.models import Account, User
+from web.backend.schemas import DouyuCollabAccountLookupOut, DouyuCollabOrderOut
+from web.backend.services import douyu_collab_service as collab
+
+
+class FakeClient:
+    def __init__(self, *, points=5000, fish=1000, records=None):
+        self.points = points
+        self.fish = fish
+        self.records = records if records is not None else []
+
+    def elite_user_info(self, **_):
+        return {
+            "manual_type": 1,
+            "manual_score": self.points,
+            "expire_time": 0,
+            "user_info": {"nickname": "斗鱼昵称"},
+        }
+
+    def bind_info(self, **_):
+        return {
+            "role_name": "角色A",
+            "area_name": "微信",
+            "plat_name": "苹果(iOS)",
+            "nickname": "微信昵称",
+        }
+
+    def gold_account(self):
+        return {"gold": self.fish}
+
+    def list_goods(self, **_):
+        return {
+            "goods": [
+                {
+                    "commodityId": "gift-a",
+                    "commodityName": "礼包A",
+                    "score": 100,
+                    "status": 0,
+                    "openStatus": 1,
+                    "mark": "",
+                    "count": 10,
+                }
+            ]
+        }
+
+    def exchange_records(self, **_):
+        return {"records": self.records}
+
+    def exchange_orders(self, **_):
+        return {"orders": []}
+
+
+class FakeNoBindClient(FakeClient):
+    def bind_info(self, **_):
+        raise RuntimeError("未绑定角色")
+
+
+class TestDouyuCollabOrders:
+    def setup_method(self):
+        self.engine = get_test_engine()
+        self.session = make_test_session(self.engine)
+        self.admin = User(
+            username="douyu-collab-admin",
+            password_hash="hash",
+            role="super_admin",
+            is_active=True,
+        )
+        self.session.add(self.admin)
+        self.session.commit()
+        self.account = Account(
+            username="douyu-account-a",
+            password="password",
+            email="a@example.com",
+            email_password="mail-password",
+            cookie="acf_uid=100; acf_nickname=test",
+        )
+        self.session.add(self.account)
+        self.session.commit()
+
+    def teardown_method(self):
+        self.session.close()
+
+    def test_lookup_account_resolves_internal_ck_and_role(self):
+        result = collab.lookup_account(
+            self.session, "douyu-account-a", client=FakeClient()
+        )
+        assert result["found"] is True
+        assert result["ok"] is True
+        assert result["account_id"] == self.account.id
+        assert result["bind_game_role"] == "角色A"
+        assert result["manual_score"] == 5000
+        assert result["fish_fin_balance"] == 1000
+        assert result["goods"][0]["commodity_id"] == "gift-a"
+        DouyuCollabAccountLookupOut.model_validate(result)
+
+    def test_create_strict_order_uses_internal_account_and_advances(self):
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-001",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-1.png"],
+            client=FakeClient(),
+        )
+        assert order.account_id == self.account.id
+        assert order.bind_game_role == "角色A"
+        assert order.points_balance == 5000
+        assert order.fish_fin_balance == 1000
+        assert order.status == collab.STATUS_TO_RECHARGE
+        DouyuCollabOrderOut.model_validate(collab.serialize_one(self.session, order))
+
+    def test_same_account_allows_multiple_orders(self):
+        collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-002",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-1.png"],
+            client=FakeClient(),
+        )
+        second = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-003",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-2.png"],
+            client=FakeClient(),
+        )
+        assert second.account_id == self.account.id
+        assert collab.group_points_total(self.session, second) == 200
+
+    def test_status_flow_and_exchange_record_marks_done(self):
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-004",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-3.png"],
+            client=FakeClient(),
+        )
+        collab.set_status(
+            self.session,
+            self.admin,
+            order,
+            status=collab.STATUS_RECHARGED,
+            reason="鱼翅已充值",
+        )
+        collab.set_status(
+            self.session,
+            self.admin,
+            order,
+            status=collab.STATUS_SCRIPTED,
+            reason="脚本已上",
+        )
+        result = collab.query_activity(
+            self.session,
+            self.admin,
+            order,
+            client=FakeClient(records=[{"commodityName": "礼包A"}]),
+        )
+        assert result["verdict"] == collab.VERDICT_SUCCESS
+        assert result["order"].status == collab.STATUS_DONE
+        assert result["order"].done_at is not None
+
+    def test_points_shortage_enters_points_error(self):
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-005",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-4.png"],
+            client=FakeClient(),
+        )
+        collab.set_status(
+            self.session,
+            self.admin,
+            order,
+            status=collab.STATUS_RECHARGED,
+            reason="鱼翅已充值",
+        )
+        order.points_threshold = 10
+        self.session.commit()
+        result = collab.query_activity(
+            self.session,
+            self.admin,
+            order,
+            client=FakeClient(points=50, records=[]),
+        )
+        assert result["points_reason"] == "shortage"
+        assert result["order"].status == collab.STATUS_POINTS_ERROR
+
+    def test_draft_can_be_created_before_account_resolution(self):
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-006",
+            shop="",
+            account_username="not-exists-yet",
+            draft=True,
+        )
+        assert order.status == collab.STATUS_INCOMPLETE
+        assert order.account_id is None
+        assert order.account_username == "not-exists-yet"
+
+    def test_draft_existing_username_links_account_and_can_be_completed(self):
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-009",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            draft=True,
+        )
+        assert order.account_id == self.account.id
+        collab.update_order(
+            self.session,
+            self.admin,
+            order,
+            {
+                "account_username": "douyu-account-a",
+                "gift_name": "礼包A",
+                "commodity_id": "gift-a",
+                "required_points": 100,
+                "home_images": ["collab/home-7.png"],
+            },
+            client=FakeClient(),
+        )
+        self.session.refresh(order)
+        assert order.bind_game_role == "角色A"
+        assert order.status == collab.STATUS_TO_RECHARGE
+        assert collab.serialize_one(self.session, order)["missing_fields"] == []
+
+    def test_missing_bind_role_keeps_incomplete_and_explains_why(self):
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-011",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            draft=True,
+        )
+        collab.update_order(
+            self.session,
+            self.admin,
+            order,
+            {
+                "account_username": "douyu-account-a",
+                "gift_name": "礼包A",
+                "commodity_id": "gift-a",
+                "required_points": 100,
+                "home_images": ["collab/home-missing-role.png"],
+            },
+            client=FakeNoBindClient(),
+        )
+        self.session.refresh(order)
+        assert order.status == collab.STATUS_INCOMPLETE
+        payload = collab.serialize_one(self.session, order)
+        assert "绑定角色（先校验账号）" in payload["missing_fields"]
+        DouyuCollabOrderOut.model_validate(payload)
+
+    def test_done_order_is_not_counted_again_in_group_points(self):
+        first = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-007",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-5.png"],
+            client=FakeClient(),
+        )
+        second = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-008",
+            shop="店铺A",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-6.png"],
+            client=FakeClient(),
+        )
+        assert collab.group_points_total(self.session, second) == 200
+        collab.set_status(
+            self.session,
+            self.admin,
+            first,
+            status=collab.STATUS_DONE,
+            reason="已兑换",
+        )
+        assert collab.group_points_total(self.session, second) == 100
+
+    def test_goods_normalization_keeps_zero_values(self):
+        goods = collab.serialize_live_goods(
+            [
+                {
+                    "commodityId": "gift-23",
+                    "commodityName": "23返场礼包",
+                    "score": 8000,
+                    "status": 0,
+                    "openStatus": 0,
+                    "startTime": "1792749600",
+                    "endTime": "1793375999",
+                    "mark": "限时",
+                    "count": 2000,
+                }
+            ]
+        )
+        assert goods[0]["status"] == "0"
+        assert goods[0]["open_status"] == 0
+        assert goods[0]["start_time"] == 1792749600
+        assert goods[0]["is_limited"] is True
+
+    def test_shop_rename_cascades_and_delete_rejects_in_use(self):
+        shop = collab.create_shop(self.session, self.admin, name="店铺B", remark="测试")
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-009",
+            shop="店铺B",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-7.png"],
+            client=FakeClient(),
+        )
+        collab.update_shop(
+            self.session,
+            self.admin,
+            shop,
+            {"name": "店铺B-改名", "is_active": False},
+        )
+        self.session.refresh(order)
+        assert order.shop == "店铺B-改名"
+        assert shop.is_active is False
+
+        with pytest.raises(collab.CollabError):
+            collab.delete_shop(self.session, self.admin, shop)
+
+        collab.delete_order(self.session, self.admin, order)
+        assert collab.delete_shop(self.session, self.admin, shop)["success"] is True
+
+    def test_ck_status_filter_and_summary_count_invalid(self):
+        order = collab.create_order(
+            self.session,
+            self.admin,
+            order_no="DY-010",
+            shop="店铺C",
+            account_username="douyu-account-a",
+            gift_name="礼包A",
+            required_points=100,
+            home_images=["collab/home-8.png"],
+            client=FakeClient(),
+        )
+        self.account.ck_check_status = collab.CK_INVALID
+        self.session.commit()
+
+        invalid = collab.list_orders(self.session, ck_status=collab.CK_INVALID)
+        assert [item["id"] for item in invalid["items"]] == [order.id]
+        assert collab.list_orders(self.session, ck_status=collab.CK_VALID)["total"] == 0
+        assert collab.summary(self.session)["ck_invalid"] == 1
+
+    def test_douyu_collab_routes_registered(self):
+        from web.backend.main import app
+
+        paths = app.openapi()["paths"]
+        assert "/api/douyu/external-orders" in paths
+        assert "/api/douyu/external-orders/{order_id}/status" in paths
+        assert "/api/douyu/external-orders/status/batch" in paths
+        assert "/api/douyu/external-orders/verify/batch" in paths
+        assert "/api/douyu/external-orders/accounts/lookup" in paths
+        assert "/api/douyu/collab-shops/{shop_id}" in paths
+        assert "put" in paths["/api/douyu/collab-shops/{shop_id}"]
+        assert "delete" in paths["/api/douyu/collab-shops/{shop_id}"]
@@ -26,7 +26,7 @@ TESTS_DIR = Path(__file__).resolve().parent
 PROJECT_ROOT = TESTS_DIR.parent
 VERSIONS_DIR = PROJECT_ROOT / "web" / "backend" / "migrations" / "versions"

-HEAD_REVISION = "20261019_0061"
+HEAD_REVISION = "20261020_0062"

 # 迁移新增、但不声明在模型里的复合查询索引。
 EXTRA_INDEXES = (