mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-05 07:27:15 +08:00
feat: expose download save paths in API
Return configured download directories as API-ready save_path values so external integrations can choose download destinations without guessing local or remote path syntax. Fixes #5737
This commit is contained in:
@@ -0,0 +1,91 @@
|
||||
import sys
|
||||
import unittest
|
||||
from types import ModuleType
|
||||
from unittest.mock import patch
|
||||
|
||||
|
||||
def _stub_module(name: str, **attrs):
|
||||
module = sys.modules.get(name)
|
||||
if module is None:
|
||||
module = ModuleType(name)
|
||||
sys.modules[name] = module
|
||||
for key, value in attrs.items():
|
||||
setattr(module, key, value)
|
||||
return module
|
||||
|
||||
|
||||
class _Dummy:
|
||||
def __init__(self, *args, **kwargs):
|
||||
pass
|
||||
|
||||
def __getattr__(self, _name):
|
||||
return lambda *args, **kwargs: None
|
||||
|
||||
|
||||
for _module_name in ("pillow_avif", "aiofiles", "psutil"):
|
||||
_stub_module(_module_name)
|
||||
|
||||
_stub_module("app.chain.download", DownloadChain=_Dummy)
|
||||
_stub_module("app.chain.media", MediaChain=_Dummy)
|
||||
_stub_module("app.core.context", MediaInfo=_Dummy, Context=_Dummy, TorrentInfo=_Dummy)
|
||||
_stub_module("app.core.metainfo", MetaInfo=_Dummy)
|
||||
_stub_module("app.core.security", verify_token=_Dummy)
|
||||
_stub_module("app.db.models.user", User=_Dummy)
|
||||
_stub_module("app.db.systemconfig_oper", SystemConfigOper=_Dummy)
|
||||
_stub_module("app.db.user_oper", get_current_active_user=_Dummy)
|
||||
_stub_module(
|
||||
"app.log",
|
||||
logger=_Dummy(),
|
||||
log_settings=_Dummy(),
|
||||
LogConfigModel=type("LogConfigModel", (), {}),
|
||||
)
|
||||
_stub_module("version", APP_VERSION="test")
|
||||
|
||||
from app.api.endpoints import download as download_endpoint
|
||||
|
||||
|
||||
class DownloadPathsEndpointTest(unittest.TestCase):
|
||||
def test_paths_returns_api_ready_save_paths(self):
|
||||
mocked_dirs = [
|
||||
download_endpoint.schemas.TransferDirectoryConf(
|
||||
name="电影目录",
|
||||
priority=1,
|
||||
storage="local",
|
||||
download_path="/downloads/movies",
|
||||
media_type="movie",
|
||||
),
|
||||
download_endpoint.schemas.TransferDirectoryConf(
|
||||
name="动漫远程目录",
|
||||
priority=2,
|
||||
storage="rclone",
|
||||
download_path="/media/anime",
|
||||
media_type="tv",
|
||||
media_category="动漫",
|
||||
),
|
||||
]
|
||||
|
||||
with patch.object(download_endpoint.DirectoryHelper, "get_download_dirs", return_value=mocked_dirs):
|
||||
ret = download_endpoint.paths(_=download_endpoint.schemas.TokenPayload())
|
||||
|
||||
self.assertEqual(len(ret), 2)
|
||||
self.assertEqual(ret[0].name, "电影目录")
|
||||
self.assertEqual(ret[0].storage, "local")
|
||||
self.assertEqual(ret[0].download_path, "/downloads/movies")
|
||||
self.assertEqual(ret[0].save_path, "/downloads/movies")
|
||||
self.assertEqual(ret[0].priority, 1)
|
||||
self.assertEqual(ret[0].media_type, "movie")
|
||||
self.assertIsNone(ret[0].media_category)
|
||||
|
||||
self.assertEqual(ret[1].name, "动漫远程目录")
|
||||
self.assertEqual(ret[1].storage, "rclone")
|
||||
self.assertEqual(ret[1].download_path, "/media/anime")
|
||||
self.assertEqual(ret[1].save_path, "rclone:/media/anime")
|
||||
self.assertEqual(ret[1].priority, 2)
|
||||
self.assertEqual(ret[1].media_type, "tv")
|
||||
self.assertEqual(ret[1].media_category, "动漫")
|
||||
|
||||
def test_paths_returns_empty_list_when_unconfigured(self):
|
||||
with patch.object(download_endpoint.DirectoryHelper, "get_download_dirs", return_value=[]):
|
||||
ret = download_endpoint.paths(_=download_endpoint.schemas.TokenPayload())
|
||||
|
||||
self.assertEqual(ret, [])
|
||||
Reference in New Issue
Block a user