fix(douyu): reject invalid activity cookies
This commit is contained in:
@@ -12,7 +12,7 @@ from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from web.backend.database import Base
|
||||
from web.backend.models import Account, AuditLog, DouyuTask, User
|
||||
from web.backend.models import Account, AuditLog, DouyuTask, LoginTask, User
|
||||
from web.backend.routers.douyu import get_recharge_channel, supplier_recharge_callback
|
||||
from web.backend.services.douyu_runner import DouyuBatchRunner
|
||||
from web.backend.services.douyu_service import ensure_douyu_config
|
||||
@@ -176,6 +176,46 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
|
||||
self.assertEqual(self.task.status, "failed")
|
||||
self.assertEqual(self.task.message, "商品已下架")
|
||||
|
||||
@patch("web.backend.services.douyu_runner.check_douyu_cookie")
|
||||
@patch("web.backend.services.douyu_runner.SessionLocal")
|
||||
def test_invalid_cookie_skips_activity_and_marks_relogin_required(self, session_local, check_cookie):
|
||||
login_task = LoginTask(
|
||||
batch_id="login-batch",
|
||||
account_id=self.account.id,
|
||||
created_by=self.task.created_by,
|
||||
status="success",
|
||||
cookie="acf_uid=10001",
|
||||
)
|
||||
self.task.status = "planned"
|
||||
self.session.add(login_task)
|
||||
self.session.commit()
|
||||
task_id = self.task.id
|
||||
login_task_id = login_task.id
|
||||
session_local.return_value = self.session
|
||||
check_cookie.return_value = {
|
||||
"valid": False,
|
||||
"message": "鱼丸接口: 未登录;等级接口: 未登录",
|
||||
"fish_ball": None,
|
||||
"nickname": None,
|
||||
"level": None,
|
||||
"checked_at": None,
|
||||
}
|
||||
self.runner._execute_create_gold_qr = Mock()
|
||||
|
||||
self.runner._execute_one(task_id, {}, 1)
|
||||
|
||||
self.runner._execute_create_gold_qr.assert_not_called()
|
||||
verify_session = sessionmaker(bind=self.engine)()
|
||||
try:
|
||||
task = verify_session.get(DouyuTask, task_id)
|
||||
checked_login = verify_session.get(LoginTask, login_task_id)
|
||||
self.assertEqual(task.status, "failed")
|
||||
self.assertIn("Cookie 已失效,请重新登录", task.message)
|
||||
self.assertEqual(checked_login.ck_check_status, "invalid")
|
||||
self.assertEqual(checked_login.ck_check_result["message"], "鱼丸接口: 未登录;等级接口: 未登录")
|
||||
finally:
|
||||
verify_session.close()
|
||||
|
||||
def test_supplier_callback_verifies_signature_and_updates_terminal_task(self):
|
||||
self.task.supplier_out_order_id = f"DYGFBATCHT{self.task.id}"
|
||||
self.task.result = {"recharge_channel": "supplier_api", "out_order_id": self.task.supplier_out_order_id}
|
||||
|
||||
Reference in New Issue
Block a user