优化斗鱼充值

tests/test_douyu_gold_recharge_channel.py

@@ -1,8 +1,9 @@
 """斗鱼鱼翅充值渠道分流测试。"""
 
+import asyncio
 import os
 import unittest
-from unittest.mock import Mock, patch
+from unittest.mock import AsyncMock, Mock, patch
 
 os.environ.setdefault("DATABASE_URL", "sqlite://")
 os.environ.setdefault("APP_ENCRYPTION_KEY", "MDEyMzQ1Njc4OWFiY2RlZjAxMjM0NTY3ODlhYmNkZWY=")
@@ -12,7 +13,9 @@ from sqlalchemy.orm import sessionmaker
 
 from web.backend.database import Base
 from web.backend.models import Account, DouyuTask, User
+from web.backend.routers.douyu import supplier_recharge_callback
 from web.backend.services.douyu_runner import DouyuBatchRunner
+from core.douyu import FishFinRechargeClient, FishFinRechargeConfig
 
 
 class DouyuGoldRechargeChannelTests(unittest.TestCase):
@@ -39,6 +42,9 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
         self.session.add(self.task)
         self.session.commit()
         self.runner = DouyuBatchRunner(self.session, "batch", "create_gold_qr")
+        os.environ["FISH_FIN_RECHARGE_BASE_URL"] = "https://supplier.example"
+        os.environ["FISH_FIN_RECHARGE_APP_ID"] = "test-app"
+        os.environ["FISH_FIN_RECHARGE_APP_SECRET"] = "test-secret"
 
     def tearDown(self):
         self.session.close()
@@ -50,7 +56,7 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
         supplier = Mock()
         supplier.create_order.return_value = {
             "code": 200,
-            "result": {"order_status": 2, "order_no": "supplier-001"},
+            "result": {"order_status": 2, "order_id": "supplier-001"},
             "sign": "not-stored",
         }
         client_class.return_value = supplier
@@ -67,20 +73,23 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
         )
 
         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}",
+            pay_amount=unittest.mock.ANY,
+            out_order_id=f"DYGF{self.task.id}",
             product_id="gold-product",
             recharge_arg=[{"templateName": "斗鱼账号", "templateVal": "10001"}],
+            order_type=0,
+            notify_url=unittest.mock.ANY,
         )
-        self.assertEqual(str(supplier.create_order.call_args.kwargs["customer_price"]), "10")
+        self.assertEqual(str(supplier.create_order.call_args.kwargs["pay_amount"]), "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.assertEqual(self.task.result["out_order_id"], f"DYGF{self.task.id}")
+        self.assertEqual(self.task.result["order_id"], "supplier-001")
+        self.assertEqual(self.task.result["pay_amount"], "10")
         self.assertNotIn("pay_url", self.task.result)
         self.assertNotIn("sign", self.task.result["supplier_order"])
 
@@ -103,6 +112,43 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
         self.assertEqual(self.task.status, "failed")
         self.assertEqual(self.task.message, "商品已下架")
 
+    def test_supplier_callback_verifies_signature_and_updates_terminal_task(self):
+        self.task.supplier_out_order_id = f"DYGF{self.task.id}"
+        self.task.result = {"recharge_channel": "supplier_api", "out_order_id": self.task.supplier_out_order_id}
+        self.session.commit()
+        client = FishFinRechargeClient(FishFinRechargeConfig(
+            base_url="https://supplier.example", app_id="test-app", app_secret="test-secret",
+        ))
+        payload = {
+            "app_id": "test-app",
+            "timestamp": 1786629000,
+            "out_order_id": self.task.supplier_out_order_id,
+            "order_id": "supplier-001",
+            "order_type": 0,
+            "order_status": 2,
+            "create_time": "2026/8/13 21:30",
+        }
+        payload["sign"] = client.sign(payload, "POST")
+        request = Mock()
+        request.json = AsyncMock(return_value=payload)
+
+        response = asyncio.run(supplier_recharge_callback(request, self.session))
+
+        self.assertEqual(response["code"], 200)
+        self.assertEqual(response["message"], "success")
+        self.assertTrue(client.verify_response_sign(response, "POST"))
+        self.session.refresh(self.task)
+        self.assertEqual(self.task.status, "success")
+        self.assertEqual(self.task.message, "供应商直充成功（异步通知）")
+        self.session.refresh(self.account)
+        self.assertEqual(self.account.bind_status, "gold_recharged")
+        self.assertEqual(self.task.result["order_id"], "supplier-001")
+        self.assertEqual(self.task.result["supplier_order_status"], 2)
+        # 同一终态回调可重复投递，不会改写已结束任务。
+        asyncio.run(supplier_recharge_callback(request, self.session))
+        self.session.refresh(self.task)
+        self.assertEqual(self.task.status, "success")
+
 
 if __name__ == "__main__":
     unittest.main()