mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-04 23:17:20 +08:00
57 lines
2.1 KiB
Python
57 lines
2.1 KiB
Python
"""下载任务应用服务测试。"""
|
|
|
|
from app.application.download.tasks import DownloadTaskService
|
|
from app.application.history import DownloadHistorySnapshot
|
|
from app.schemas.transfer import DownloaderTorrent
|
|
from app.schemas.types import MediaSource
|
|
|
|
|
|
def test_download_task_service_enriches_history_and_controls_task():
|
|
"""下载任务查询应附加历史媒体信息，控制方法只转发规范参数。"""
|
|
torrent = DownloaderTorrent(hash="hash")
|
|
history = DownloadHistorySnapshot(
|
|
id=1,
|
|
path="/downloads/test",
|
|
media_source=MediaSource.TMDB,
|
|
media_id="123",
|
|
type="电影",
|
|
title="测试电影",
|
|
seasons="1",
|
|
episodes="2",
|
|
poster="poster",
|
|
image="backdrop",
|
|
torrent_site="站点",
|
|
userid="1",
|
|
username="alice",
|
|
)
|
|
calls = []
|
|
service = DownloadTaskService(
|
|
list_torrents=lambda **kwargs: [torrent],
|
|
get_history_by_hashes=lambda hashes: {hashes[0]: history},
|
|
start_torrents=lambda **kwargs: calls.append(("start", kwargs)) or True,
|
|
stop_torrents=lambda **kwargs: calls.append(("stop", kwargs)) or True,
|
|
remove_torrents=lambda **kwargs: calls.append(("remove", kwargs)) or True,
|
|
)
|
|
|
|
assert service.downloading("qb") == [torrent]
|
|
assert torrent.media is not None
|
|
assert torrent.media.media_id == "123"
|
|
assert torrent.username == "alice"
|
|
assert service.set_downloading("hash", "start", "qb") is True
|
|
assert service.set_downloading("hash", "stop", "qb") is True
|
|
assert service.remove_downloading("hash", "qb") is True
|
|
assert [call[0] for call in calls] == ["start", "stop", "remove"]
|
|
|
|
|
|
def test_download_task_service_rejects_unknown_operation():
|
|
"""未知操作保持旧的 False 返回语义。"""
|
|
service = DownloadTaskService(
|
|
list_torrents=lambda **_kwargs: [],
|
|
get_history_by_hashes=lambda _hashes: {},
|
|
start_torrents=lambda **_kwargs: True,
|
|
stop_torrents=lambda **_kwargs: True,
|
|
remove_torrents=lambda **_kwargs: True,
|
|
)
|
|
|
|
assert service.set_downloading("hash", "pause") is False
|