"""鱼翅直充供应商 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_nested_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_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=""))


if __name__ == "__main__":
    unittest.main()