mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-06-11 18:50:59 +08:00
81 lines
2.9 KiB
Python
81 lines
2.9 KiB
Python
from types import SimpleNamespace
|
|
|
|
from app.schemas import ActionContext, DownloadTask, FileItem
|
|
from app.workflow.actions import fetch_downloads as fetch_downloads_module
|
|
from app.workflow.actions import scrape_file as scrape_file_module
|
|
from app.workflow.actions.fetch_downloads import FetchDownloadsAction
|
|
from app.workflow.actions.scrape_file import ScrapeFileAction
|
|
|
|
|
|
def test_fetch_downloads_updates_context_downloads(monkeypatch):
|
|
"""获取下载任务动作应更新上游上下文中的下载任务。"""
|
|
calls = []
|
|
|
|
class FakeActionChain:
|
|
"""模拟下载器查询链。"""
|
|
|
|
def list_torrents(self, hashs=None, downloader=None, **kwargs):
|
|
calls.append((hashs, downloader))
|
|
return [SimpleNamespace(path="/downloads/movie.mkv", progress=100)]
|
|
|
|
monkeypatch.setattr(fetch_downloads_module, "ActionChain", FakeActionChain)
|
|
monkeypatch.setattr(fetch_downloads_module.global_vars, "is_workflow_stopped", lambda workflow_id: False)
|
|
|
|
context = ActionContext(
|
|
downloads=[
|
|
DownloadTask(download_id="hash-1", downloader="qbittorrent"),
|
|
]
|
|
)
|
|
|
|
result = FetchDownloadsAction("fetch-downloads").execute(
|
|
workflow_id=1,
|
|
params={},
|
|
context=context,
|
|
)
|
|
|
|
assert calls == [(["hash-1"], "qbittorrent")]
|
|
assert result.downloads[0].completed is True
|
|
assert result.downloads[0].path == "/downloads/movie.mkv"
|
|
|
|
|
|
def test_scrape_file_keeps_workflow_action_context(monkeypatch):
|
|
"""刮削文件动作不应将工作流上下文替换为媒体识别上下文。"""
|
|
scraped = []
|
|
|
|
class FakeStorageChain:
|
|
"""模拟存储链。"""
|
|
|
|
def exists(self, fileitem):
|
|
return True
|
|
|
|
class FakeMediaChain:
|
|
"""模拟媒体识别和刮削链。"""
|
|
|
|
def recognize_by_path(self, path, obtain_images=False):
|
|
return SimpleNamespace(meta_info="meta", media_info="media")
|
|
|
|
def scrape_metadata(self, fileitem, meta=None, mediainfo=None):
|
|
scraped.append((fileitem.path, meta, mediainfo))
|
|
|
|
monkeypatch.setattr(scrape_file_module, "StorageChain", FakeStorageChain)
|
|
monkeypatch.setattr(scrape_file_module, "MediaChain", FakeMediaChain)
|
|
monkeypatch.setattr(scrape_file_module.global_vars, "is_workflow_stopped", lambda workflow_id: False)
|
|
monkeypatch.setattr(ScrapeFileAction, "check_cache", lambda self, workflow_id, key: False)
|
|
monkeypatch.setattr(ScrapeFileAction, "save_cache", lambda self, workflow_id, data: None)
|
|
|
|
context = ActionContext(
|
|
fileitems=[
|
|
FileItem(path="/library/movie.mkv", storage="local", type="file"),
|
|
]
|
|
)
|
|
|
|
result = ScrapeFileAction("scrape-file").execute(
|
|
workflow_id=1,
|
|
params={},
|
|
context=context,
|
|
)
|
|
|
|
assert result is context
|
|
assert result.fileitems[0].path == "/library/movie.mkv"
|
|
assert scraped == [("/library/movie.mkv", "meta", "media")]
|