mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-11-10 15:50:24 +08:00
129 lines
4.4 KiB
Python
129 lines
4.4 KiB
Python
"""数据库备份与宿主调度器的接入合同。"""
|
|
|
|
import ast
|
|
import threading
|
|
from dataclasses import replace
|
|
from pathlib import Path
|
|
from unittest.mock import Mock
|
|
|
|
from app.application.configuration import SchedulerRuntimeConfig
|
|
from app.scheduler import catalog as scheduler_catalog
|
|
from app.scheduler import maintenance as scheduler_maintenance
|
|
from app.scheduler.facade import Scheduler
|
|
from app.scheduler.registry import ExecutionRegistry
|
|
|
|
|
|
class _SchedulerStub:
|
|
"""记录 Scheduler 注册结果的最小调度器替身。"""
|
|
|
|
def __init__(self) -> None:
|
|
"""初始化空作业表。"""
|
|
self.jobs = {}
|
|
|
|
def add_job(self, func, *, trigger, id, **kwargs) -> None:
|
|
"""按作业 ID 保存最近一次注册参数。"""
|
|
self.jobs[id] = {"func": func, "trigger": trigger, **kwargs}
|
|
|
|
|
|
def _scheduler() -> Scheduler:
|
|
"""构造不启动后台线程的 Scheduler。"""
|
|
scheduler = object.__new__(Scheduler)
|
|
scheduler._scheduler = _SchedulerStub()
|
|
scheduler._jobs = {}
|
|
scheduler._lock = threading.RLock()
|
|
scheduler._lifecycle_state = "running"
|
|
scheduler._registry = ExecutionRegistry(scheduler._lock)
|
|
return scheduler
|
|
|
|
|
|
def _config(**changes) -> SchedulerRuntimeConfig:
|
|
"""构造数据库备份测试所需的最小调度配置快照。"""
|
|
config = SchedulerRuntimeConfig(
|
|
dev=False,
|
|
timezone="Asia/Shanghai",
|
|
scheduler_workers=1,
|
|
db_backup_enable=False,
|
|
db_backup_cron="",
|
|
cookiecloud_interval=None,
|
|
mediaserver_sync_interval=None,
|
|
subscribe_search=False,
|
|
subscribe_search_interval=24,
|
|
subscribe_mode="rss",
|
|
subscribe_rss_interval=30,
|
|
data_cleanup_enable=False,
|
|
sitedata_refresh_interval=None,
|
|
memory_gc_interval=None,
|
|
ai_agent_enable=False,
|
|
ai_agent_job_interval=None,
|
|
usage_statistic_share=False,
|
|
site_link=None,
|
|
)
|
|
return replace(config, **changes)
|
|
|
|
|
|
def test_database_backup_schedule_only_watches_job_shape() -> None:
|
|
assert Scheduler.CONFIG_WATCH.intersection({
|
|
"DB_BACKUP_ENABLE",
|
|
"DB_BACKUP_CRON",
|
|
"DB_BACKUP_ON_UPGRADE",
|
|
"DB_BACKUP_PATH",
|
|
"DB_BACKUP_RETENTION_DAYS",
|
|
"DB_BACKUP_MAX_COUNT",
|
|
}) == {"DB_BACKUP_ENABLE", "DB_BACKUP_CRON"}
|
|
|
|
|
|
def test_disabled_database_backup_does_not_register_job() -> None:
|
|
"""关闭备份时不注册作业。"""
|
|
scheduler = _scheduler()
|
|
|
|
scheduler._register_database_backup_job(_config())
|
|
|
|
assert scheduler._scheduler.jobs == {}
|
|
|
|
|
|
def test_enabled_database_backup_without_cron_does_not_register_job() -> None:
|
|
"""总开关开启但未配置周期时，不启用定时备份。"""
|
|
scheduler = _scheduler()
|
|
scheduler._register_database_backup_job(_config(db_backup_enable=True))
|
|
|
|
assert scheduler._scheduler.jobs == {}
|
|
|
|
|
|
def test_enabled_database_backup_registers_single_replaceable_job(monkeypatch) -> None:
|
|
scheduler = _scheduler()
|
|
trigger = object()
|
|
monkeypatch.setattr(scheduler_catalog.TimerUtils, "build_schedule_trigger", Mock(return_value=trigger))
|
|
config = _config(db_backup_enable=True, db_backup_cron="0 3 * * *")
|
|
|
|
scheduler._register_database_backup_job(config)
|
|
scheduler._register_database_backup_job(config)
|
|
|
|
assert list(scheduler._scheduler.jobs) == ["database_backup"]
|
|
assert scheduler._scheduler.jobs["database_backup"]["replace_existing"] is True
|
|
|
|
|
|
def test_scheduled_backup_uses_registered_database_governance(monkeypatch) -> None:
|
|
governance = Mock()
|
|
monkeypatch.setattr(scheduler_maintenance, "get_database_governance", lambda: governance)
|
|
|
|
result = Scheduler.database_backup()
|
|
|
|
assert result is governance.create_backup.return_value
|
|
governance.create_backup.assert_called_once_with()
|
|
|
|
|
|
def test_scheduler_database_dependencies_are_explicit_module_imports() -> None:
|
|
scheduler_root = Path(__file__).parents[1] / "app" / "scheduler"
|
|
trees = [ast.parse(path.read_text(encoding="utf-8")) for path in scheduler_root.glob("*.py")]
|
|
function_imports = [
|
|
node
|
|
for tree in trees
|
|
for function in ast.walk(tree)
|
|
if isinstance(function, (ast.FunctionDef, ast.AsyncFunctionDef))
|
|
for node in ast.walk(function)
|
|
if isinstance(node, (ast.Import, ast.ImportFrom))
|
|
and getattr(node, "module", "")
|
|
and str(getattr(node, "module", "")).startswith("app.application.database")
|
|
]
|
|
assert function_imports == []
|