mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-11-01 21:07:01 +08:00
73 lines
2.0 KiB
Python
73 lines
2.0 KiB
Python
"""
|
||
验证站点重置任务的调度注册和执行。
|
||
"""
|
||
|
||
from unittest.mock import AsyncMock, Mock
|
||
|
||
import pytest
|
||
|
||
from app.api.endpoints import site as site_endpoint
|
||
from app.application.site.mutation import SiteMutationResult
|
||
from app.runtime.tasks import TaskRegistry
|
||
|
||
|
||
class _TaskRegistry(TaskRegistry):
|
||
"""
|
||
记录站点重置提交的同步后台任务。
|
||
"""
|
||
|
||
def __init__(self) -> None:
|
||
"""
|
||
初始化任务记录。
|
||
"""
|
||
super().__init__()
|
||
self.calls: list[tuple] = []
|
||
|
||
def create_sync(self, function, *args, owner: str, **kwargs) -> None:
|
||
"""
|
||
保存函数、参数和 owner，避免端点测试启动真实任务。
|
||
"""
|
||
self.calls.append((function, args, kwargs, owner))
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_reset_submits_cookiecloud_after_site_transaction(monkeypatch):
|
||
"""
|
||
站点重置提交 CookieCloud 后台任务，不在请求事件循环内直接执行。
|
||
"""
|
||
command = Mock()
|
||
command.reset = AsyncMock(return_value=SiteMutationResult(success=True))
|
||
task_registry = _TaskRegistry()
|
||
scheduler = Mock()
|
||
system_config = Mock()
|
||
system_config.async_set = AsyncMock()
|
||
|
||
monkeypatch.setattr(site_endpoint, "get_scheduler", Mock(return_value=scheduler))
|
||
monkeypatch.setattr(
|
||
site_endpoint,
|
||
"get_configured_system_config",
|
||
Mock(return_value=system_config),
|
||
)
|
||
|
||
response = await site_endpoint.reset(
|
||
task_registry=task_registry,
|
||
command=command,
|
||
_=Mock(),
|
||
)
|
||
|
||
assert response.success is True
|
||
command.reset.assert_awaited_once_with()
|
||
scheduler.start.assert_not_called()
|
||
system_config.async_set.assert_any_await(
|
||
site_endpoint.SystemConfigKey.IndexerSites, []
|
||
)
|
||
system_config.async_set.assert_any_await(site_endpoint.SystemConfigKey.RssSites, [])
|
||
assert task_registry.calls == [
|
||
(
|
||
scheduler.start,
|
||
(),
|
||
{"job_id": "cookiecloud", "manual": True},
|
||
"api.site.reset",
|
||
)
|
||
]
|