"""鱼翅直充供应商 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_new_document_payload(self):
        response = Mock()
        response.json.return_value = {"code": 200, "order_status": 0}
        self.session.post.return_value = response

        result = self.client.create_order(
            buy_num=2,
            pay_amount="1.20",
            out_order_id="OUT202503010010208888",
            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"]["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))

    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",
            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"))

    def test_query_order_requires_identifier_and_uses_get_params(self):
        with self.assertRaisesRegex(ValueError, "out_order_id"):
            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("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"))

    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(
            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)

    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(
            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()