type: 收窄测试与响应模型类型
This commit is contained in:
@@ -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="",
|
||||
|
||||
Reference in New Issue
Block a user