kodik tests, type annotation for id_type

moe_parsers/providers/kodik.py

@@ -236,11 +236,11 @@ async def search(
 
         return animes
 
-    async def translations(self, anime_id: str, id_type: str) -> list:
+    async def translations(self, anime_id: str, id_type: Literal["shikimori", "kinopoisk", "imdb"]) -> list:
         data = await self.get_info(anime_id, id_type)
         return data["translations"]
 
-    async def episode_count(self, anime_id: str, id_type: str) -> int:
+    async def episode_count(self, anime_id: str, id_type: Literal["shikimori", "kinopoisk", "imdb"]) -> int:
         data = await self.get_info(anime_id, id_type)
         return data["episode_count"]
 
@@ -264,7 +264,7 @@ async def _link_to_info(
             raise Exceptions.PlayerBlocked(f'Нет данных по {id_type} id "{id}"')
         return "https:" + data["link"]
 
-    async def get_info(self, anime_id: str, id_type: str) -> dict:
+    async def get_info(self, anime_id: str, id_type: Literal["shikimori", "kinopoisk", "imdb"]) -> dict:
         link = await self._link_to_info(anime_id, id_type)
         data = await self.get(link, text=True)
         soup = await self.soup(data)

tests/unit/test_kodik.py

@@ -0,0 +1,20 @@
+import pytest
+from moe_parsers.providers.kodik import KodikParser, KodikAnime, KodikEpisode, KodikIframe
+
+
+@pytest.mark.asyncio
+async def test_kodik_search():
+    parser = KodikParser()
+    res = await parser.search("plastic memories")
+    assert len(res) > 0
+    assert "27775" in res[0].anime_id
+
+@pytest.mark.asyncio
+async def test_kodik_get_video():
+    parser = KodikParser()
+    res = await parser.search("plastic memories")
+    assert len(res) > 0
+    await res[0].get_video(1)
+    assert len(res[0].episodes) > 0
+    assert isinstance(res[0].episodes[0], KodikEpisode)
+