mirror of
https://hubproxy.babadafafafafa.cn/https://github.com/jxxghp/MoviePilot.git
synced 2026-09-20 08:03:34 +08:00
64 lines
3.1 KiB
Python
64 lines
3.1 KiB
Python
"""整理恢复用例的清理边界与错误证据保留测试。"""
|
|
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from app.application.transfer.execution import TransferExecutionConflictError
|
|
from app.application.transfer.recovery import TransferRecoveryCommand
|
|
from app.chain.transfer.settlement import _discard_corrupt_transfer_task
|
|
|
|
|
|
@pytest.mark.parametrize("message", ["记录已失效", "记录不完整", "版本不一致", "检查点", "恢复状态不完整"])
|
|
def test_corrupt_conflict_discards_owned_task(message):
|
|
"""不可恢复的规划冲突必须带当前租约原子清理持久任务。"""
|
|
repository = Mock()
|
|
repository.discard_corrupt_task.return_value = True
|
|
result = TransferRecoveryCommand(repository).discard_conflict(
|
|
task_id="task", lease_token="lease", error=TransferExecutionConflictError(message),
|
|
)
|
|
assert result is True
|
|
repository.discard_corrupt_task.assert_called_once_with(
|
|
task_id="task", lease_token="lease", error=message,
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("error,task_id,lease_token", [
|
|
(ValueError("记录不完整"), "task", "lease"),
|
|
(TransferExecutionConflictError("暂时失败"), "task", "lease"),
|
|
(TransferExecutionConflictError("记录不完整"), None, "lease"),
|
|
(TransferExecutionConflictError("记录不完整"), "task", None),
|
|
])
|
|
def test_recoverable_or_unowned_failure_preserves_evidence(error, task_id, lease_token):
|
|
"""普通错误、可重试冲突和缺少租约的任务都不得删除恢复证据。"""
|
|
repository = Mock()
|
|
assert TransferRecoveryCommand(repository).discard_conflict(
|
|
task_id=task_id, lease_token=lease_token, error=error,
|
|
) is False
|
|
repository.discard_corrupt_task.assert_not_called()
|
|
|
|
|
|
@pytest.mark.parametrize("preview", [True, False])
|
|
def test_cleanup_preserves_preview_and_tolerates_repository_failure(preview):
|
|
"""预览不清理任务；持久清理失败不阻止内存作业收口。"""
|
|
repository = Mock()
|
|
repository.discard_corrupt_task.side_effect = RuntimeError("存储失败")
|
|
task = Mock(preview=preview, admission_task_id="task", lease_token="lease")
|
|
_discard_corrupt_transfer_task(repository, task, TransferExecutionConflictError("记录不完整"))
|
|
assert repository.discard_corrupt_task.call_count == (0 if preview else 1)
|
|
|
|
|
|
@pytest.mark.parametrize("method,kwargs", [
|
|
("discard_corrupt_task", {"task_id": "", "lease_token": "lease", "error": "损坏"}),
|
|
("discard_corrupt_task", {"task_id": "task", "lease_token": "", "error": "损坏"}),
|
|
("discard_corrupt_task", {"task_id": "task", "lease_token": "lease", "error": ""}),
|
|
("discard_corrupt_by_history", {"task_id": "task", "history_id": 0}),
|
|
("discard_failed", {"task_id": "task", "history_id": 7, "settlement_revision": 0}),
|
|
])
|
|
def test_invalid_cleanup_identity_never_reaches_repository(method, kwargs):
|
|
"""缺少身份或结算证据时，必须在写持久层之前拒绝清理。"""
|
|
repository = Mock()
|
|
with pytest.raises(ValueError):
|
|
getattr(TransferRecoveryCommand(repository), method)(**kwargs)
|
|
assert repository.mock_calls == []
|