type: 收窄测试与响应模型类型

This commit is contained in:
yml2213
2026-08-30 20:21:14 +08:00
parent 3f97dbe525
commit cbd314f4e5
7 changed files with 96 additions and 69 deletions
+21 -16
View File
@@ -94,16 +94,18 @@ class TestDouyuGoldRechargeChannel:
assert str(supplier.create_order.call_args.kwargs["pay_amount"]) == "10"
self.session.refresh(self.task)
assert self.task.status == "success"
assert self.task.result["recharge_channel"] == "supplier_api"
assert self.task.result["supplier_order_status"] == 2
assert self.task.result["buy_num"] == 10
assert self.task.result["out_order_id"] == expected_order_id
assert self.task.result["recharge_account"] == "罗炅729"
assert self.task.result["douyu_uid"] == "10001"
assert self.task.result["order_id"] == "supplier-001"
assert self.task.result["pay_amount"] == "10"
assert "pay_url" not in self.task.result
assert "sign" not in self.task.result["supplier_order"]
task_result = self.task.result
assert isinstance(task_result, dict)
assert task_result["recharge_channel"] == "supplier_api"
assert task_result["supplier_order_status"] == 2
assert task_result["buy_num"] == 10
assert task_result["out_order_id"] == expected_order_id
assert task_result["recharge_account"] == "罗炅729"
assert task_result["douyu_uid"] == "10001"
assert task_result["order_id"] == "supplier-001"
assert task_result["pay_amount"] == "10"
assert "pay_url" not in task_result
assert "sign" not in task_result["supplier_order"]
@patch("web.backend.services.douyu_runner_gold.FishFinRechargeClient")
def test_supplier_channel_rejects_account_without_nickname(self, client_class):
@@ -226,13 +228,14 @@ class TestDouyuGoldRechargeChannel:
try:
task = verify_session.get(DouyuTask, task_id)
checked_login = verify_session.get(LoginTask, login_task_id)
assert task is not None
assert checked_login is not None
assert task.status == "failed"
assert "Cookie 已失效,请重新登录" in task.message
assert checked_login.ck_check_status == "invalid"
assert (
checked_login.ck_check_result["message"]
== "鱼丸接口: 未登录;等级接口: 未登录"
)
check_result = checked_login.ck_check_result
assert isinstance(check_result, dict)
assert check_result["message"] == "鱼丸接口: 未登录;等级接口: 未登录"
finally:
verify_session.close()
@@ -273,8 +276,10 @@ class TestDouyuGoldRechargeChannel:
assert self.task.message == "供应商直充成功(异步通知)"
self.session.refresh(self.account)
assert self.account.bind_status == "gold_recharged"
assert self.task.result["order_id"] == "supplier-001"
assert self.task.result["supplier_order_status"] == 2
task_result = self.task.result
assert isinstance(task_result, dict)
assert task_result["order_id"] == "supplier-001"
assert task_result["supplier_order_status"] == 2
audit = (
self.session.query(AuditLog)
.filter(AuditLog.action == "recharge:douyu:callback")