import unittest

from backend.app.services.arr import RootFolderNotFoundError, resolve_root_folder_path


class _ArrClient:
    async def get_root_folders(self):
        return [{"id": 7, "path": "/media/tv"}]


class ArrHelperTests(unittest.IsolatedAsyncioTestCase):
    async def test_resolves_numeric_root_folder_id(self) -> None:
        self.assertEqual(await resolve_root_folder_path(_ArrClient(), "7", "Sonarr"), "/media/tv")

    async def test_preserves_configured_path(self) -> None:
        self.assertEqual(await resolve_root_folder_path(_ArrClient(), "/media/movies", "Radarr"), "/media/movies")

    async def test_rejects_missing_root_folder_id(self) -> None:
        with self.assertRaises(RootFolderNotFoundError):
            await resolve_root_folder_path(_ArrClient(), "8", "Sonarr")


if __name__ == "__main__":
    unittest.main()