功能: 接入鱼翅供应商直充渠道

tests/test_douyu_gold_recharge_channel.py

@@ -0,0 +1,108 @@
+"""斗鱼鱼翅充值渠道分流测试。"""
+
+import os
+import unittest
+from unittest.mock import Mock, patch
+
+os.environ.setdefault("DATABASE_URL", "sqlite://")
+os.environ.setdefault("APP_ENCRYPTION_KEY", "MDEyMzQ1Njc4OWFiY2RlZjAxMjM0NTY3ODlhYmNkZWY=")
+
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+
+from web.backend.database import Base
+from web.backend.models import Account, DouyuTask, User
+from web.backend.services.douyu_runner import DouyuBatchRunner
+
+
+class DouyuGoldRechargeChannelTests(unittest.TestCase):
+    def setUp(self):
+        self.engine = create_engine("sqlite://")
+        Base.metadata.create_all(self.engine)
+        self.session = sessionmaker(bind=self.engine)()
+        user = User(username="operator", password_hash="hash", role="super_admin")
+        self.session.add(user)
+        self.session.commit()
+        self.account = Account(
+            username="douyu-user",
+            password="password",
+            email="mail@example.com",
+            email_password="mail-password",
+            uid="10001",
+        )
+        self.session.add(self.account)
+        self.session.commit()
+        self.task = DouyuTask(
+            batch_id="batch", account_id=self.account.id, task_type="create_gold_qr",
+            handbook_scope="elite", status="running", created_by=user.id,
+        )
+        self.session.add(self.task)
+        self.session.commit()
+        self.runner = DouyuBatchRunner(self.session, "batch", "create_gold_qr")
+
+    def tearDown(self):
+        self.session.close()
+        Base.metadata.drop_all(self.engine)
+        self.engine.dispose()
+
+    @patch("web.backend.services.douyu_runner.FishFinRechargeClient")
+    def test_supplier_channel_creates_order_using_uid_and_finishes_on_success(self, client_class):
+        supplier = Mock()
+        supplier.create_order.return_value = {
+            "code": 200,
+            "result": {"order_status": 2, "order_no": "supplier-001"},
+            "sign": "not-stored",
+        }
+        client_class.return_value = supplier
+        config = {
+            "gold_recharge_channel": "supplier_api",
+            "gold_api_product_id": "gold-product",
+            "gold_api_account_template_name": "斗鱼账号",
+        }
+
+        self.task.result = {"payload": {"amount": 10}}
+        self.session.commit()
+        self.runner._execute_create_gold_qr(
+            self.session, self.task, self.account, "acf_uid=10001", config,
+        )
+
+        supplier.create_order.assert_called_once_with(
+            charge_account="10001",
+            buy_num=10,
+            customer_price=unittest.mock.ANY,
+            customer_order_no=f"DYGF{self.task.id}",
+            product_id="gold-product",
+            recharge_arg=[{"templateName": "斗鱼账号", "templateVal": "10001"}],
+        )
+        self.assertEqual(str(supplier.create_order.call_args.kwargs["customer_price"]), "10")
+        self.session.refresh(self.task)
+        self.assertEqual(self.task.status, "success")
+        self.assertEqual(self.task.result["recharge_channel"], "supplier_api")
+        self.assertEqual(self.task.result["supplier_order_status"], 2)
+        self.assertEqual(self.task.result["buy_num"], 10)
+        self.assertEqual(self.task.result["customer_price"], "10")
+        self.assertNotIn("pay_url", self.task.result)
+        self.assertNotIn("sign", self.task.result["supplier_order"])
+
+    @patch("web.backend.services.douyu_runner.FishFinRechargeClient")
+    def test_supplier_channel_marks_non_200_create_response_as_failed(self, client_class):
+        supplier = Mock()
+        supplier.create_order.return_value = {"code": 400, "msg": "商品已下架"}
+        client_class.return_value = supplier
+        config = {
+            "gold_recharge_channel": "supplier_api",
+            "gold_api_product_id": "gold-product",
+            "gold_api_account_template_name": "斗鱼UID",
+        }
+
+        self.runner._execute_create_gold_qr(
+            self.session, self.task, self.account, "acf_uid=10001", config,
+        )
+
+        self.session.refresh(self.task)
+        self.assertEqual(self.task.status, "failed")
+        self.assertEqual(self.task.message, "商品已下架")
+
+
+if __name__ == "__main__":
+    unittest.main()