执行 Ruff 安全自动修复
This commit is contained in:
@@ -1,8 +1,7 @@
|
||||
import pytest
|
||||
|
||||
from fastapi import HTTPException
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
from fastapi import HTTPException
|
||||
|
||||
from web.backend.database import Base
|
||||
from web.backend.models import Account, User
|
||||
|
||||
@@ -1,9 +1,8 @@
|
||||
"""充值审计日志的权限、查询和脱敏测试。"""
|
||||
|
||||
import pytest
|
||||
|
||||
import json
|
||||
|
||||
import pytest
|
||||
from fastapi import HTTPException
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
@@ -1,4 +1,4 @@
|
||||
from datetime import datetime, timedelta, timezone
|
||||
from datetime import UTC, datetime, timedelta
|
||||
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import Session
|
||||
@@ -17,7 +17,7 @@ class TestCustomCookieOrder:
|
||||
self.db.add(self.user)
|
||||
self.db.flush()
|
||||
|
||||
now = datetime.now(timezone.utc)
|
||||
now = datetime.now(UTC)
|
||||
accounts = {
|
||||
name: Account(username=name, password="p", email="e", email_password="ep")
|
||||
for name in ("account-a", "account-b", "account-c")
|
||||
|
||||
@@ -1,18 +1,13 @@
|
||||
import pytest
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import patch
|
||||
from typing import cast
|
||||
from unittest.mock import patch
|
||||
|
||||
import pytest
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from web.backend.database import Base
|
||||
from web.backend.models import Account, LoginTask, ProxyConfig, User
|
||||
from web.backend.services.login_service import (
|
||||
LoginBatchRunner,
|
||||
cleanup_orphan_relogin_tasks,
|
||||
)
|
||||
from web.backend.services.cookie_check_service import check_douyu_cookie
|
||||
from web.backend.routers.cookies import (
|
||||
check_cookie_operations,
|
||||
get_cookie,
|
||||
@@ -21,6 +16,11 @@ from web.backend.routers.cookies import (
|
||||
list_cookies,
|
||||
relogin_invalid_cookie_operations,
|
||||
)
|
||||
from web.backend.services.cookie_check_service import check_douyu_cookie
|
||||
from web.backend.services.login_service import (
|
||||
LoginBatchRunner,
|
||||
cleanup_orphan_relogin_tasks,
|
||||
)
|
||||
|
||||
|
||||
class TestCookieOperation:
|
||||
|
||||
@@ -1,9 +1,9 @@
|
||||
"""精英手册兑换:csrf 复用、新链路接口与浏览器状态机路由的测试。"""
|
||||
|
||||
import pytest
|
||||
|
||||
from unittest.mock import Mock, call
|
||||
|
||||
import pytest
|
||||
|
||||
from core.douyu.activity_client import DouyuActivityClient, DouyuActivityError
|
||||
|
||||
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
import pytest
|
||||
from unittest.mock import Mock
|
||||
|
||||
import pytest
|
||||
import requests
|
||||
|
||||
from core.douyu.activity_client import DouyuActivityClient, DouyuActivityError
|
||||
|
||||
@@ -7,13 +7,13 @@ from unittest.mock import ANY, AsyncMock, Mock, patch
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from core.douyu import FishFinRechargeClient, FishFinRechargeConfig
|
||||
from web.backend.database import Base
|
||||
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.audit_service import record_audit
|
||||
from web.backend.services.douyu_runner import DouyuBatchRunner
|
||||
from web.backend.services.douyu_service import ensure_douyu_config
|
||||
from web.backend.services.audit_service import record_audit
|
||||
from core.douyu import FishFinRechargeClient, FishFinRechargeConfig
|
||||
|
||||
|
||||
class TestDouyuGoldRechargeChannel:
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import Mock
|
||||
from typing import Any, cast
|
||||
from unittest.mock import Mock
|
||||
|
||||
from core.douyu.login import DouyuLogin
|
||||
from core.douyu.login_api_wgapi import WgapiLoginAPI
|
||||
|
||||
@@ -3,12 +3,12 @@
|
||||
from unittest.mock import Mock
|
||||
|
||||
from core.douyu.activity_client import DouyuActivityClient
|
||||
from web.backend.models import ProxyConfig
|
||||
from web.backend.services.douyu_runner import DouyuBatchRunner
|
||||
from web.backend.services.douyu_runner_core import (
|
||||
DouyuBatchRunnerCore,
|
||||
DOUYU_PROXY_TASK_TYPES,
|
||||
DouyuBatchRunnerCore,
|
||||
)
|
||||
from web.backend.models import ProxyConfig
|
||||
|
||||
|
||||
def _cfg(**overrides) -> ProxyConfig:
|
||||
|
||||
@@ -1,5 +1,4 @@
|
||||
import pytest
|
||||
|
||||
from sqlalchemy import create_engine, event
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
|
||||
@@ -1,9 +1,10 @@
|
||||
import pytest
|
||||
import json
|
||||
import re
|
||||
from types import SimpleNamespace
|
||||
from unittest.mock import Mock
|
||||
|
||||
import pytest
|
||||
|
||||
from core.douyu.activity_client import DouyuActivityClient, DouyuActivityError
|
||||
|
||||
|
||||
|
||||
@@ -1,9 +1,9 @@
|
||||
"""鱼翅直充供应商 API 客户端测试。"""
|
||||
|
||||
import pytest
|
||||
|
||||
from unittest.mock import Mock
|
||||
|
||||
import pytest
|
||||
|
||||
from core.douyu.recharge_api import (
|
||||
FishFinRechargeClient,
|
||||
FishFinRechargeConfig,
|
||||
|
||||
@@ -1,21 +1,15 @@
|
||||
"""虎牙 App 密码登录及相关组件测试。"""
|
||||
|
||||
import pytest
|
||||
|
||||
import base64
|
||||
import os
|
||||
import struct
|
||||
from unittest.mock import patch, MagicMock
|
||||
from unittest.mock import MagicMock, patch
|
||||
|
||||
import pytest
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from core.huya import (
|
||||
HuyaAppLoginError,
|
||||
HuyaAppPasswordLogin,
|
||||
HuyaAppQrAuthRequiredError,
|
||||
login_huya_app_password,
|
||||
login_huya_password,
|
||||
)
|
||||
from core.huya.app_login import (
|
||||
DEFAULT_GOLDEN_DEV,
|
||||
@@ -30,17 +24,16 @@ from core.huya.login import HuyaLoginResult
|
||||
from core.huya.nonce_forge import K1_DEFAULT, gen_nonce
|
||||
from core.huya.udb_aes import udb_decrypt, udb_encrypt
|
||||
from core.huya.wup_encoder import build_password_login_wup
|
||||
|
||||
from web.backend.database import Base
|
||||
from web.backend.models import User, HuyaAccount
|
||||
from web.backend.schemas import (
|
||||
HuyaAppPasswordLoginRequest,
|
||||
HuyaPasswordLoginSelectedRequest,
|
||||
)
|
||||
from web.backend.models import HuyaAccount, User
|
||||
from web.backend.routers.huya import (
|
||||
app_password_login_account,
|
||||
app_password_login_selected_accounts,
|
||||
)
|
||||
from web.backend.schemas import (
|
||||
HuyaAppPasswordLoginRequest,
|
||||
HuyaPasswordLoginSelectedRequest,
|
||||
)
|
||||
|
||||
|
||||
class TestHuyaAppLogin:
|
||||
@@ -193,9 +186,8 @@ class TestHuyaAppLogin:
|
||||
with patch(
|
||||
"core.huya.app_login.register_device",
|
||||
side_effect=DfpRegistrationError("注册链 HTTP 500"),
|
||||
):
|
||||
with pytest.raises(HuyaAppLoginError, match="注册失败"):
|
||||
login_cred_with_flow("300023887", "pw")
|
||||
), pytest.raises(HuyaAppLoginError, match="注册失败"):
|
||||
login_cred_with_flow("300023887", "pw")
|
||||
|
||||
def test_router_functions(self):
|
||||
mock_res = HuyaLoginResult(
|
||||
|
||||
@@ -7,9 +7,9 @@ from datetime import datetime, timedelta
|
||||
from pathlib import Path
|
||||
|
||||
from utils.logger import (
|
||||
_parse_size,
|
||||
_SensitiveDataFilter,
|
||||
_SizeAndDayRotatingFileHandler,
|
||||
_parse_size,
|
||||
)
|
||||
|
||||
|
||||
|
||||
@@ -68,8 +68,8 @@ class TestMigrationSmoke:
|
||||
|
||||
def test_full_chain_applies_on_fresh_sqlite_and_matches_models(self):
|
||||
"""空库执行整条迁移链,校验表/列/索引与模型元数据对齐。"""
|
||||
from web.backend.database import Base
|
||||
import web.backend.database as database_module
|
||||
from web.backend.database import Base
|
||||
|
||||
original_url = database_module.DATABASE_URL
|
||||
with tempfile.TemporaryDirectory() as tmp:
|
||||
|
||||
Reference in New Issue
Block a user