yml/live-hub-py
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: 全面迁移 pytest 并加入格式门禁

Browse Source
This commit is contained in:
yml2213
2026-08-30 19:19:32 +08:00
parent 3ce1c7a51b
commit 13b5aedd1d
26 changed files with 1374 additions and 888 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_fish_fin_recharge_api.py
+103 -67
View File
@@ -1,6 +1,7 @@
"""鱼翅直充供应商 API 客户端测试。"""
import unittest
import pytest
from unittest.mock import Mock
from core.douyu.recharge_api import (
@@ -10,8 +11,8 @@ from core.douyu.recharge_api import (
)
class FishFinRechargeClientTests(unittest.TestCase):
def setUp(self):
class TestFishFinRechargeClient:
def setup_method(self):
self.config = FishFinRechargeConfig(
base_url="https://supplier.example/",
app_id="15945681",
@@ -21,18 +22,21 @@ class FishFinRechargeClientTests(unittest.TestCase):
self.client = FishFinRechargeClient(self.config, session=self.session)
def test_sign_matches_document_example(self):
sign = self.client.sign({
"app_id": "15945681 ",
"timestamp": 1740916504,
"charge_account": "18888888888",
"buy_num": 2,
"customer_price": 1.21,
"customer_order_no": "DD202507010010208888",
"product_id": "66668888",
"empty": " \n",
"sign": "ignored",
}, "POST")
self.assertEqual(sign, "4087a959f3488ecb13efe6ef58e3bc67")
sign = self.client.sign(
{
"app_id": "15945681 ",
"timestamp": 1740916504,
"charge_account": "18888888888",
"buy_num": 2,
"customer_price": 1.21,
"customer_order_no": "DD202507010010208888",
"product_id": "66668888",
"empty": " \n",
"sign": "ignored",
},
"POST",
)
assert sign == "4087a959f3488ecb13efe6ef58e3bc67"
def test_create_order_posts_new_document_payload(self):
response = Mock()
@@ -48,40 +52,52 @@ class FishFinRechargeClientTests(unittest.TestCase):
ext_arg={"skuid": 12},
)
self.assertEqual(result["code"], 200)
assert result["code"] == 200
kwargs = self.session.post.call_args.kwargs
self.assertEqual(self.session.post.call_args.args[0], "https://supplier.example/adapter-apiaccess/open/api/createOrderV2")
self.assertEqual(kwargs["json"]["pay_amount"], 1.2)
self.assertEqual(kwargs["json"]["out_order_id"], "OUT202503010010208888")
self.assertEqual(kwargs["json"]["order_type"], 0)
self.assertEqual(kwargs["json"]["recharge_arg"], '[{"templateName":"斗鱼账号","templateVal":"10001"}]')
self.assertEqual(kwargs["json"]["ext_arg"], '{"skuid":12}')
self.assertNotIn("charge_account", kwargs["json"])
self.assertNotIn("customer_price", kwargs["json"])
self.assertNotIn("customer_order_no", kwargs["json"])
self.assertEqual(kwargs["json"]["sign"], self.client.sign(kwargs["json"], "POST"))
self.assertEqual(kwargs["timeout"], (8, 20))
assert (
self.session.post.call_args.args[0]
== "https://supplier.example/adapter-apiaccess/open/api/createOrderV2"
)
assert kwargs["json"]["pay_amount"] == 1.2
assert kwargs["json"]["out_order_id"] == "OUT202503010010208888"
assert kwargs["json"]["order_type"] == 0
assert (
kwargs["json"]["recharge_arg"]
== '[{"templateName":"斗鱼账号","templateVal":"10001"}]'
)
assert kwargs["json"]["ext_arg"] == '{"skuid":12}'
assert "charge_account" not in kwargs["json"]
assert "customer_price" not in kwargs["json"]
assert "customer_order_no" not in kwargs["json"]
assert kwargs["json"]["sign"] == self.client.sign(kwargs["json"], "POST")
assert kwargs["timeout"] == (8, 20)
def test_create_order_omits_empty_optional_fields_and_serializes_integer_price(self):
def test_create_order_omits_empty_optional_fields_and_serializes_integer_price(
self,
):
response = Mock()
response.json.return_value = {"code": 200}
self.session.post.return_value = response
self.client.create_order(
buy_num=1, pay_amount="1.0",
out_order_id="merchant-001", product_id="111570",
buy_num=1,
pay_amount="1.0",
out_order_id="merchant-001",
product_id="111570",
recharge_arg=[{"templateName": "斗鱼UID", "templateVal": "10001"}],
)
body = self.session.post.call_args.kwargs["json"]
self.assertEqual(body["pay_amount"], 1)
self.assertNotIn("notify_url", body)
self.assertEqual(body["recharge_arg"], '[{"templateName":"斗鱼UID","templateVal":"10001"}]')
self.assertNotIn("ext_arg", body)
self.assertEqual(body["sign"], self.client.sign(body, "POST"))
assert body["pay_amount"] == 1
assert "notify_url" not in body
assert (
body["recharge_arg"] == '[{"templateName":"斗鱼UID","templateVal":"10001"}]'
)
assert "ext_arg" not in body
assert body["sign"] == self.client.sign(body, "POST")
def test_query_order_requires_identifier_and_uses_get_params(self):
with self.assertRaisesRegex(ValueError, "out_order_id"):
with pytest.raises(ValueError, match="out_order_id"):
self.client.query_order("")
response = Mock()
@@ -90,13 +106,18 @@ class FishFinRechargeClientTests(unittest.TestCase):
self.client.query_order("OUT202503010010208888")
kwargs = self.session.get.call_args.kwargs
self.assertEqual(self.session.get.call_args.args[0], "https://supplier.example/adapter-apiaccess/open/api/queryOrderV2")
self.assertEqual(kwargs["params"]["out_order_id"], "OUT202503010010208888")
self.assertEqual(kwargs["params"]["sign"], self.client.sign(kwargs["params"], "GET"))
assert (
self.session.get.call_args.args[0]
== "https://supplier.example/adapter-apiaccess/open/api/queryOrderV2"
)
assert kwargs["params"]["out_order_id"] == "OUT202503010010208888"
assert kwargs["params"]["sign"] == self.client.sign(kwargs["params"], "GET")
def test_missing_configuration_is_rejected_before_request(self):
with self.assertRaises(FishFinRechargeConfigError):
FishFinRechargeClient(FishFinRechargeConfig(base_url="", app_id="", app_secret=""))
with pytest.raises(FishFinRechargeConfigError):
FishFinRechargeClient(
FishFinRechargeConfig(base_url="", app_id="", app_secret="")
)
def test_trace_excludes_replayable_signature_and_nested_account_data(self):
response = Mock()
@@ -104,47 +125,62 @@ class FishFinRechargeClientTests(unittest.TestCase):
response.json.return_value = {"code": 200, "result": {"order_status": 0}}
self.session.post.return_value = response
events = []
client = FishFinRechargeClient(self.config, session=self.session, trace=events.append)
client = FishFinRechargeClient(
self.config, session=self.session, trace=events.append
)
client.create_order(
buy_num=1, pay_amount="1",
out_order_id="merchant-001", product_id="111570",
buy_num=1,
pay_amount="1",
out_order_id="merchant-001",
product_id="111570",
recharge_arg=[{"templateName": "斗鱼UID", "templateVal": "10001"}],
)
self.assertEqual([event["stage"] for event in events], ["request", "response"])
self.assertNotIn("sign", events[0]["params"])
self.assertNotIn("recharge_arg", events[0]["params"])
self.assertEqual(events[0]["params"]["pay_amount"], 1)
assert [event["stage"] for event in events] == ["request", "response"]
assert "sign" not in events[0]["params"]
assert "recharge_arg" not in events[0]["params"]
assert events[0]["params"]["pay_amount"] == 1
def test_debug_trace_includes_complete_request_and_response_bodies(self):
response = Mock()
response.status_code = 200
response.headers = {"Content-Type": "application/json", "X-Request-Id": "request-1"}
response.text = '{"code":1000,"message":"未传递支付金额","sign":"response-sign"}'
response.json.return_value = {"code": 1000, "message": "未传递支付金额", "sign": "response-sign"}
response.headers = {
"Content-Type": "application/json",
"X-Request-Id": "request-1",
}
response.text = (
'{"code":1000,"message":"未传递支付金额","sign":"response-sign"}'
)
response.json.return_value = {
"code": 1000,
"message": "未传递支付金额",
"sign": "response-sign",
}
self.session.post.return_value = response
events = []
config = FishFinRechargeConfig(
base_url=self.config.base_url, app_id=self.config.app_id,
app_secret=self.config.app_secret, debug=True,
base_url=self.config.base_url,
app_id=self.config.app_id,
app_secret=self.config.app_secret,
debug=True,
)
client = FishFinRechargeClient(
config, session=self.session, trace=events.append
)
client = FishFinRechargeClient(config, session=self.session, trace=events.append)
client.create_order(
buy_num=1, pay_amount="1",
out_order_id="merchant-001", product_id="111570",
buy_num=1,
pay_amount="1",
out_order_id="merchant-001",
product_id="111570",
recharge_arg=[{"templateName": "斗鱼UID", "templateVal": "10001"}],
)
self.assertEqual(events[0]["json_body"]["sign"], "<redacted>")
self.assertEqual(events[0]["json_body"]["pay_amount"], 1)
self.assertNotIn("sign", events[0]["sign_params"])
self.assertNotIn(self.config.app_secret, str(events[0]))
self.assertIn("sign_source_digest", events[0])
self.assertEqual(events[1]["response_body"]["sign"], "<redacted>")
self.assertNotIn(self.config.app_secret, str(events[1]))
if __name__ == "__main__":
unittest.main()
assert events[0]["json_body"]["sign"] == "<redacted>"
assert events[0]["json_body"]["pay_amount"] == 1
assert "sign" not in events[0]["sign_params"]
assert self.config.app_secret not in str(events[0])
assert "sign_source_digest" in events[0]
assert events[1]["response_body"]["sign"] == "<redacted>"
assert self.config.app_secret not in str(events[1])