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
+5 -3
View File
@@ -1,12 +1,13 @@
import pytest
from types import SimpleNamespace
from unittest.mock import patch
from typing import cast
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from web.backend.database import Base
from web.backend.models import Account, LoginTask, User
from web.backend.models import Account, LoginTask, ProxyConfig, User
from web.backend.services.login_service import (
LoginBatchRunner,
cleanup_orphan_relogin_tasks,
@@ -137,6 +138,7 @@ class TestCookieOperation:
current=admin,
)
assert isinstance(result, dict)
assert result["total"] == 1
assert [item["id"] for item in result["items"]] == [self.owned_task.id]
@@ -200,7 +202,7 @@ class TestCookieOperation:
page=1,
page_size=20,
db=self.session,
current=no_permission_user,
current=cast(User, no_permission_user),
)
def test_relogin_failed_record_stays_visible_and_can_be_checked(self):
@@ -235,7 +237,7 @@ class TestCookieOperation:
assert self.other_task.status == "success"
def test_runner_copies_proxy_config_before_background_execution(self):
proxy = SimpleNamespace(
proxy = ProxyConfig(
enabled=True,
http="http://127.0.0.1:8080",
https="",