143 lines
6.2 KiB
Python
143 lines
6.2 KiB
Python
"""鱼翅直充供应商 API 客户端测试。"""
|
|
|
|
import unittest
|
|
from unittest.mock import Mock
|
|
|
|
from core.douyu.recharge_api import (
|
|
FishFinRechargeClient,
|
|
FishFinRechargeConfig,
|
|
FishFinRechargeConfigError,
|
|
)
|
|
|
|
|
|
class FishFinRechargeClientTests(unittest.TestCase):
|
|
def setUp(self):
|
|
self.config = FishFinRechargeConfig(
|
|
base_url="https://supplier.example/",
|
|
app_id="15945681",
|
|
app_secret="ac9fa79db418fd2c61d9bd4c60956f2d5db3a6d781d6f91f2e96327c74dfaccd",
|
|
)
|
|
self.session = Mock()
|
|
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")
|
|
|
|
def test_create_order_posts_json_with_signed_optional_arguments(self):
|
|
response = Mock()
|
|
response.json.return_value = {"code": 200, "order_status": 0}
|
|
self.session.post.return_value = response
|
|
|
|
result = self.client.create_order(
|
|
charge_account="10001",
|
|
buy_num=2,
|
|
customer_price="1.20",
|
|
customer_order_no="merchant-001",
|
|
product_id="product-1",
|
|
recharge_arg=[{"templateName": "斗鱼账号", "templateVal": "10001"}],
|
|
ext_arg={"skuid": 12},
|
|
)
|
|
|
|
self.assertEqual(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"]["customer_price"], 1.2)
|
|
self.assertEqual(kwargs["json"]["sign"], self.client.sign(kwargs["json"], "POST"))
|
|
self.assertEqual(kwargs["timeout"], (8, 20))
|
|
|
|
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(
|
|
charge_account="10001", buy_num=1, customer_price="1.0",
|
|
customer_order_no="merchant-001", product_id="111570",
|
|
)
|
|
|
|
body = self.session.post.call_args.kwargs["json"]
|
|
self.assertEqual(body["customer_price"], 1)
|
|
self.assertNotIn("notify_url", body)
|
|
self.assertNotIn("recharge_arg", body)
|
|
self.assertNotIn("ext_arg", body)
|
|
self.assertEqual(body["sign"], self.client.sign(body, "POST"))
|
|
|
|
def test_query_order_requires_identifier_and_uses_get_params(self):
|
|
with self.assertRaisesRegex(ValueError, "订单标识"):
|
|
self.client.query_order()
|
|
|
|
response = Mock()
|
|
response.json.return_value = {"code": 200, "order_status": 2}
|
|
self.session.get.return_value = response
|
|
self.client.query_order(customer_order_no="merchant-001")
|
|
|
|
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"]["customer_order_no"], "merchant-001")
|
|
self.assertEqual(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=""))
|
|
|
|
def test_trace_excludes_replayable_signature_and_nested_account_data(self):
|
|
response = Mock()
|
|
response.status_code = 200
|
|
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.create_order(
|
|
charge_account="10001", buy_num=1, customer_price="0.993",
|
|
customer_order_no="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"]["customer_price"], 0.993)
|
|
|
|
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"}
|
|
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,
|
|
)
|
|
client = FishFinRechargeClient(config, session=self.session, trace=events.append)
|
|
|
|
client.create_order(
|
|
charge_account="10001", buy_num=1, customer_price="0.993",
|
|
customer_order_no="merchant-001", product_id="111570",
|
|
recharge_arg=[{"templateName": "斗鱼UID", "templateVal": "10001"}],
|
|
)
|
|
|
|
self.assertEqual(events[0]["json_body"]["sign"], client.sign(events[0]["json_body"], "POST"))
|
|
self.assertEqual(events[0]["json_body"]["customer_price"], 0.993)
|
|
self.assertNotIn("sign", events[0]["sign_params"])
|
|
self.assertIn(self.config.app_secret, events[0]["sign_source"])
|
|
self.assertEqual(events[1]["response_body"]["sign"], "response-sign")
|
|
self.assertEqual(events[1]["response_text"], response.text)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|