mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-11-11 16:21:09 +08:00
264 lines
10 KiB
Python
264 lines
10 KiB
Python
from types import SimpleNamespace
|
|
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
|
|
from app.application.history import (
|
|
DownloadHistoryMutationCommand,
|
|
TransferHistoryMutationCommand,
|
|
)
|
|
from app.application.transfer.execution import (
|
|
TransferExecutionState,
|
|
TransferFailureDiscardResult,
|
|
)
|
|
|
|
|
|
def _transfer_command(*, history=None, delete_result=True, commit_error=None):
|
|
"""构造可观察整理历史事务和外部副作用的命令。"""
|
|
repository = Mock()
|
|
repository.get.return_value = history
|
|
download_repository = Mock()
|
|
transfer_execution_repository = Mock()
|
|
unit_of_work = Mock()
|
|
unit_of_work.commit.side_effect = commit_error
|
|
dependencies = {
|
|
"repository": repository,
|
|
"download_repository": download_repository,
|
|
"transfer_execution_repository": transfer_execution_repository,
|
|
"unit_of_work": unit_of_work,
|
|
"file_item_factory": lambda payload: SimpleNamespace(**payload),
|
|
"delete_media_file": Mock(return_value=delete_result),
|
|
"publish_download_file_deleted": Mock(),
|
|
"clear_failures": Mock(),
|
|
}
|
|
return TransferHistoryMutationCommand(**dependencies), dependencies
|
|
|
|
|
|
def _history():
|
|
"""构造包含源和目标文件信息的整理历史快照。"""
|
|
return SimpleNamespace(
|
|
id=7,
|
|
src="/downloads/demo.mkv",
|
|
src_storage="local",
|
|
download_hash="abc",
|
|
src_fileitem={"path": "/downloads/demo.mkv"},
|
|
dest_fileitem={"path": "/media/demo.mkv"},
|
|
transfer_task_id=None,
|
|
transfer_settlement_revision=None,
|
|
)
|
|
|
|
|
|
def test_download_history_delete_rolls_back_commit_failure():
|
|
"""下载历史提交失败时必须回滚请求级事务。"""
|
|
repository = Mock()
|
|
unit_of_work = Mock()
|
|
unit_of_work.commit.side_effect = RuntimeError("commit failed")
|
|
command = DownloadHistoryMutationCommand(
|
|
repository=repository,
|
|
unit_of_work=unit_of_work,
|
|
)
|
|
|
|
with pytest.raises(RuntimeError, match="commit failed"):
|
|
command.delete(8)
|
|
|
|
repository.stage_delete_history.assert_called_once_with(8)
|
|
unit_of_work.rollback.assert_called_once_with()
|
|
|
|
|
|
def test_transfer_source_delete_failure_keeps_database_unchanged():
|
|
"""源文件删除失败时不得删除历史或更新下载文件状态。"""
|
|
command, dependencies = _transfer_command(
|
|
history=_history(),
|
|
delete_result=False,
|
|
)
|
|
|
|
result = command.delete(7, delete_source=True)
|
|
|
|
assert result.success is False
|
|
assert result.message == "/downloads/demo.mkv 删除失败"
|
|
assert result.source.status == "failed"
|
|
assert result.destination.status == "not_requested"
|
|
dependencies["repository"].stage_delete.assert_not_called()
|
|
dependencies["download_repository"].stage_delete_file_by_fullpath.assert_not_called()
|
|
dependencies["unit_of_work"].commit.assert_not_called()
|
|
|
|
|
|
def test_transfer_delete_commits_before_event_and_retry_cleanup():
|
|
"""整理记录提交成功后才发送文件删除事件并清理失败计数。"""
|
|
calls = []
|
|
command, dependencies = _transfer_command(history=_history())
|
|
dependencies["unit_of_work"].commit.side_effect = lambda: calls.append("commit")
|
|
dependencies["publish_download_file_deleted"].side_effect = (
|
|
lambda _payload: calls.append("event")
|
|
)
|
|
dependencies["clear_failures"].side_effect = lambda *_args: calls.append("clear")
|
|
|
|
result = command.delete(
|
|
7,
|
|
delete_source=True,
|
|
delete_destination=True,
|
|
)
|
|
|
|
assert result.success is True
|
|
assert result.source.status == "deleted"
|
|
assert result.destination.status == "deleted"
|
|
dependencies["download_repository"].stage_delete_file_by_fullpath.assert_called_once_with(
|
|
"/downloads/demo.mkv"
|
|
)
|
|
dependencies["repository"].stage_delete.assert_called_once_with(7)
|
|
assert calls == ["commit", "event", "clear"]
|
|
|
|
|
|
def test_transfer_commit_failure_suppresses_event_and_retry_cleanup():
|
|
"""数据库提交失败时不得发布已删除事件或清除重试状态。"""
|
|
command, dependencies = _transfer_command(
|
|
history=_history(),
|
|
commit_error=RuntimeError("commit failed"),
|
|
)
|
|
|
|
with pytest.raises(RuntimeError, match="commit failed"):
|
|
command.delete(7, delete_source=True)
|
|
|
|
dependencies["unit_of_work"].rollback.assert_called_once_with()
|
|
dependencies["publish_download_file_deleted"].assert_not_called()
|
|
dependencies["clear_failures"].assert_not_called()
|
|
|
|
|
|
def test_transfer_truncate_uses_single_transaction():
|
|
"""清空整理历史只暂存一次并统一提交。"""
|
|
command, dependencies = _transfer_command()
|
|
|
|
result = command.truncate()
|
|
|
|
assert result.success is True
|
|
assert result.message == "已清空旧整理记录，失败任务记录已保留"
|
|
dependencies["repository"].stage_truncate.assert_called_once_with()
|
|
dependencies["unit_of_work"].commit.assert_called_once_with()
|
|
|
|
|
|
def test_transfer_delete_rejects_nonfailed_durable_receipt_before_file_side_effects():
|
|
"""非 FAILED durable 回执不能被历史 API 连同文件一起删除。"""
|
|
history = _history()
|
|
history.transfer_task_id = "task-durable"
|
|
history.transfer_settlement_revision = 2
|
|
command, dependencies = _transfer_command(history=history)
|
|
dependencies["transfer_execution_repository"].discard_failed.return_value = (
|
|
TransferFailureDiscardResult(
|
|
discarded=False,
|
|
state=TransferExecutionState.MANUAL_REVIEW,
|
|
message="人工复核任务必须先完成专门判定",
|
|
)
|
|
)
|
|
|
|
result = command.delete(7, delete_source=True, delete_destination=True)
|
|
|
|
assert result.success is False
|
|
assert result.message == "人工复核任务必须先完成专门判定"
|
|
assert result.history == "retained"
|
|
dependencies["delete_media_file"].assert_not_called()
|
|
dependencies["repository"].stage_delete.assert_not_called()
|
|
dependencies["unit_of_work"].commit.assert_not_called()
|
|
|
|
|
|
def test_transfer_delete_discards_failed_durable_receipt_before_cleanup():
|
|
"""FAILED durable 回执应先放弃任务，再按普通历史执行文件和记录删除。"""
|
|
history = _history()
|
|
history.transfer_task_id = "task-durable"
|
|
history.transfer_settlement_revision = 3
|
|
command, dependencies = _transfer_command(history=history)
|
|
dependencies["transfer_execution_repository"].discard_failed.return_value = (
|
|
TransferFailureDiscardResult(
|
|
discarded=True,
|
|
state=TransferExecutionState.FAILED,
|
|
message="已放弃失败整理任务",
|
|
)
|
|
)
|
|
|
|
result = command.delete(7, delete_destination=True)
|
|
|
|
assert result.success is True
|
|
dependencies["transfer_execution_repository"].discard_failed.assert_called_once_with(
|
|
task_id="task-durable",
|
|
history_id=7,
|
|
settlement_revision=3,
|
|
)
|
|
dependencies["delete_media_file"].assert_called_once()
|
|
dependencies["repository"].stage_delete.assert_called_once_with(7)
|
|
dependencies["unit_of_work"].commit.assert_called_once_with()
|
|
|
|
|
|
def test_transfer_destination_delete_failure_keeps_history_and_reports_step():
|
|
"""目标文件删除失败时必须保留历史，并向调用方报告失败步骤。"""
|
|
command, dependencies = _transfer_command(
|
|
history=_history(),
|
|
delete_result=False,
|
|
)
|
|
|
|
result = command.delete(7, delete_destination=True)
|
|
|
|
assert result.success is False
|
|
assert result.destination.status == "failed"
|
|
assert result.history == "retained"
|
|
assert result.source.status == "not_requested"
|
|
dependencies["repository"].stage_delete.assert_not_called()
|
|
dependencies["unit_of_work"].commit.assert_not_called()
|
|
|
|
|
|
def test_transfer_delete_commits_completed_source_when_destination_fails():
|
|
"""一项文件已完成而另一项失败时，已完成源文件状态仍应提交并保留历史。"""
|
|
repository = Mock()
|
|
repository.get.return_value = _history()
|
|
dependencies = {
|
|
"repository": repository,
|
|
"download_repository": Mock(),
|
|
"transfer_execution_repository": Mock(),
|
|
"unit_of_work": Mock(),
|
|
"file_item_factory": lambda payload: SimpleNamespace(**payload),
|
|
"file_exists": Mock(return_value=True),
|
|
"delete_media_file": Mock(side_effect=[False, True]),
|
|
"publish_download_file_deleted": Mock(),
|
|
"clear_failures": Mock(),
|
|
}
|
|
command = TransferHistoryMutationCommand(**dependencies)
|
|
|
|
result = command.delete(7, delete_source=True, delete_destination=True)
|
|
|
|
assert result.success is False
|
|
assert result.destination.status == "failed"
|
|
assert result.source.status == "deleted"
|
|
assert result.history == "retained"
|
|
repository.stage_delete.assert_not_called()
|
|
dependencies["download_repository"].stage_delete_file_by_fullpath.assert_called_once_with(
|
|
"/downloads/demo.mkv"
|
|
)
|
|
dependencies["unit_of_work"].commit.assert_called_once_with()
|
|
dependencies["publish_download_file_deleted"].assert_called_once()
|
|
dependencies["clear_failures"].assert_not_called()
|
|
|
|
|
|
def test_transfer_delete_treats_missing_requested_file_as_completed():
|
|
"""重试时已被前一次尝试删除的文件应跳过外部删除并允许清理历史。"""
|
|
command, dependencies = _transfer_command(history=_history())
|
|
dependencies["delete_media_file"].reset_mock()
|
|
dependencies["file_exists"] = Mock(return_value=False)
|
|
command = TransferHistoryMutationCommand(
|
|
repository=dependencies["repository"],
|
|
download_repository=dependencies["download_repository"],
|
|
transfer_execution_repository=dependencies[
|
|
"transfer_execution_repository"
|
|
],
|
|
unit_of_work=dependencies["unit_of_work"],
|
|
file_item_factory=dependencies["file_item_factory"],
|
|
file_exists=dependencies["file_exists"],
|
|
delete_media_file=dependencies["delete_media_file"],
|
|
publish_download_file_deleted=dependencies["publish_download_file_deleted"],
|
|
clear_failures=dependencies["clear_failures"],
|
|
)
|
|
|
|
result = command.delete(7, delete_source=True, delete_destination=True)
|
|
|
|
assert result.success is True
|
|
assert result.source.status == "already_missing"
|
|
assert result.destination.status == "already_missing"
|
|
dependencies["delete_media_file"].assert_not_called()
|