mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-10-29 19:37:54 +08:00
131 lines
4.0 KiB
Python
131 lines
4.0 KiB
Python
"""
|
|
系统数据库备份策略配置测试。
|
|
"""
|
|
|
|
import asyncio
|
|
from contextlib import asynccontextmanager
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from app.api.endpoints import system as system_endpoint
|
|
from app.application.configuration import (
|
|
get_configured_system_config,
|
|
get_runtime_settings,
|
|
)
|
|
from app.runtime.config import settings
|
|
from app.startup.composition.system import compose_system_service
|
|
|
|
|
|
def _runtime():
|
|
"""
|
|
构造使用真实配置校验逻辑的最小系统运行时。
|
|
"""
|
|
|
|
@asynccontextmanager
|
|
async def rule_group_mutation():
|
|
"""
|
|
提供本组测试不会进入的规则组事务替身。
|
|
"""
|
|
yield SimpleNamespace()
|
|
|
|
return SimpleNamespace(
|
|
system=compose_system_service(
|
|
settings=get_runtime_settings(),
|
|
system_config=get_configured_system_config(),
|
|
rule_group_mutation=rule_group_mutation,
|
|
)
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("env", "message"),
|
|
[
|
|
({"DB_BACKUP_CRON": "0 3 * *"}, "数据库备份周期格式不正确"),
|
|
({"DB_BACKUP_PATH": 123}, "数据库备份目录必须是路径字符串"),
|
|
({"DB_BACKUP_RETENTION_DAYS": -1}, "数据库备份过期天数"),
|
|
({"DB_BACKUP_RETENTION_DAYS": 1.5}, "数据库备份过期天数"),
|
|
({"DB_BACKUP_MAX_COUNT": True}, "数据库备份最大保留份数"),
|
|
({"DB_BACKUP_MAX_COUNT": "many"}, "数据库备份最大保留份数"),
|
|
],
|
|
)
|
|
def test_database_backup_policy_rejects_invalid_values(
|
|
env: dict,
|
|
message: str,
|
|
) -> None:
|
|
"""
|
|
无效策略必须在批量配置写入前被拒绝。
|
|
"""
|
|
env["DB_BACKUP_ENABLE"] = True
|
|
assert message in str(_runtime().system._validate_database_backup_config(env))
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"env",
|
|
[
|
|
{"DB_BACKUP_CRON": ""},
|
|
{"DB_BACKUP_CRON": "0 3 * * *"},
|
|
{"DB_BACKUP_ON_UPGRADE": True},
|
|
{"DB_BACKUP_ON_UPGRADE": False},
|
|
{"DB_BACKUP_PATH": None},
|
|
{"DB_BACKUP_PATH": ""},
|
|
{"DB_BACKUP_PATH": "database_backup"},
|
|
{"DB_BACKUP_RETENTION_DAYS": 0},
|
|
{"DB_BACKUP_MAX_COUNT": "0"},
|
|
],
|
|
)
|
|
def test_database_backup_policy_accepts_supported_boundaries(env: dict) -> None:
|
|
"""
|
|
空目录使用默认路径，两个保留值的零均表示不限制。
|
|
"""
|
|
env["DB_BACKUP_ENABLE"] = True
|
|
assert _runtime().system._validate_database_backup_config(env) is None
|
|
|
|
|
|
@pytest.mark.parametrize("disabled", [False, "false", "0", "off"])
|
|
def test_database_backup_policy_rejects_invalid_hidden_values_when_disabled(disabled) -> None:
|
|
"""
|
|
关闭总开关只暂停调度，不能把无效策略写入持久配置。
|
|
"""
|
|
error = _runtime().system._validate_database_backup_config({
|
|
"DB_BACKUP_ENABLE": disabled,
|
|
"DB_BACKUP_CRON": "invalid",
|
|
"DB_BACKUP_PATH": 123,
|
|
"DB_BACKUP_RETENTION_DAYS": -1,
|
|
"DB_BACKUP_MAX_COUNT": 1.5,
|
|
})
|
|
|
|
assert error is not None
|
|
|
|
|
|
def test_set_env_rejects_invalid_database_backup_policy_without_partial_write() -> None:
|
|
"""
|
|
备份策略校验失败时不得调用 Settings 的批量写入。
|
|
"""
|
|
env = {
|
|
"DB_BACKUP_ENABLE": True,
|
|
"DB_BACKUP_CRON": "invalid",
|
|
"DB_BACKUP_RETENTION_DAYS": 30,
|
|
"DB_BACKUP_MAX_COUNT": 30,
|
|
}
|
|
|
|
with patch.object(type(settings), "update_settings") as update_settings:
|
|
response = asyncio.run(
|
|
system_endpoint.set_env_setting(env=env, _=object(), runtime=_runtime())
|
|
)
|
|
|
|
assert response.success is False
|
|
assert "数据库备份周期格式不正确" in response.message
|
|
update_settings.assert_not_called()
|
|
|
|
|
|
def test_database_backup_default_path_tracks_config_directory(tmp_path, monkeypatch) -> None:
|
|
"""
|
|
未显式配置目录时应跟随当前配置根，而不是写死 Docker 路径。
|
|
"""
|
|
monkeypatch.setattr(settings, "CONFIG_DIR", str(tmp_path))
|
|
monkeypatch.setattr(settings, "DB_BACKUP_PATH", None)
|
|
|
|
assert settings.DATABASE_BACKUP_PATH == tmp_path / "database_backup"
|