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, [])