""" Unit tests """ import unittest import aiohttp from mautrix.util.logging.trace import TraceLogger from sqlalchemy import create_engine from tmdb.database import Database from tmdb.tmdb import TmdbBot from tmdb.tmdb_api import Movie, MoviePopular, TvShow # pylint: disable=missing-class-docstring,missing-function-docstring async def api_requests(command): api_key = "51d75c00dc1502dc894b7773ec3e7a15" base_url = "https://api.themoviedb.org/3/" url = base_url + command.lstrip("/") params = {"api_key": api_key} params.update({"language": "en"}) async with aiohttp.ClientSession() as client: async with client.get(url, params=params) as resp: return await resp.json() class TestTmdbMethods(unittest.IsolatedAsyncioTestCase): # TMDB API async def test_search_item(self): movie = Movie() await movie.load_parameters() movie_id = await movie.search_title("Breakfast Club") self.assertEqual(movie_id, 2108) self.assertEqual(movie.valid, True) await movie.close_session() async def test_vote(self): movie = Movie() await movie.load_parameters() await movie.search_title("Dune") vote = movie.vote_average self.assertGreaterEqual(vote, 7) await movie.close_session() async def test_cast(self): movie = Movie() await movie.load_parameters() await movie.search_title("Breakfast Club") self.assertEqual("Emilio Estevez", movie.cast[0]) await movie.close_session() async def test_title(self): movie = Movie() await movie.load_parameters() await movie.search_title("Breakfast Club") self.assertEqual("The Breakfast Club", movie.title) await movie.close_session() async def test_overview(self): movie = Movie() await movie.load_parameters() await movie.search_title("Breakfast Club") description = "Five high school students from different walks of" self.assertEqual(description, movie.overview[: len(description)]) await movie.close_session() async def test_change_language(self): movie = Movie() await movie.load_parameters() movie.set_language("en") await movie.search_title("Breakfast Club") description = ( "Five high school students from different walks of " "life endure a Saturday detention" ) self.assertEqual(description, movie.overview[: len(description)]) await movie.close_session() def test_database_language(self): engine = create_engine("sqlite:///test.db", echo=True) db = Database(engine) db.set_language("@testuser:example.com", "de") self.assertEqual(str(db.get_language("@testuser:example.com")), "de") db.set_language("@testuser:example.com", "en") self.assertEqual(str(db.get_language("@testuser:example.com")), "en") async def test_id_lookup(self): movie = Movie() await movie.load_parameters() await movie.query_details("2108") self.assertEqual("The Breakfast Club", movie.title) await movie.close_session() async def test_search_fails(self): movie = Movie() await movie.load_parameters() movie_id = await movie.search_title("Breakfast Club 2019") self.assertEqual(movie_id, 0) self.assertEqual("", movie.title) self.assertEqual(movie.valid, False) await movie.close_session() async def test_search_year(self): movie = Movie() await movie.load_parameters() movie_id = await movie.search_title("Dune") self.assertEqual(movie_id, 438631) movie_id = await movie.search_title("Dune", 1984) self.assertEqual(movie_id, 841) await movie.close_session() def test_split_year(self): tmdb = TmdbBot( client=None, # pyright: ignore[reportArgumentType] loop=None, # pyright: ignore[reportArgumentType] http=None, # pyright: ignore[reportArgumentType] instance_id="", log=TraceLogger(""), config=None, database=None, webapp=None, webapp_url="", loader=None, # pyright: ignore[reportArgumentType] ) title, year = tmdb.split_title_year("Dune") self.assertEqual("Dune", title) self.assertEqual(0, year) title, year = tmdb.split_title_year("Dune y:2020 ") self.assertEqual("Dune", title) self.assertEqual(2020, year) async def test_set_poster_size(self): movie = Movie() await movie.load_parameters() size = movie.set_poster_size("w500") self.assertEqual(size, "w500") self.assertEqual(movie.base_url_poster, f"{movie.base_url_images}w500") size = movie.set_poster_size("w666") self.assertEqual(size, None) await movie.close_session() async def test_year_no_y(self): movie = Movie() await movie.load_parameters() movie_id = await movie.search_title("infinite y:2021") self.assertEqual(movie_id, 0) self.assertEqual(movie.valid, False) # TV Shows async def test_search_tvshow(self): movie = TvShow() await movie.load_parameters() movie_id = await movie.search_title("The Flash") self.assertEqual(movie_id, 60735) await movie.close_session() async def test_tv_title(self): movie = TvShow() await movie.load_parameters() await movie.search_title("The Flash") self.assertEqual("The Flash", movie.title) await movie.close_session() async def test_search_tvshow_v(self): show = TvShow() await show.load_parameters() mid = await show.search_title("V 2009") self.assertEqual(mid, 21494) await show.close_session() async def test_tv_title_v(self): movie = TvShow() await movie.load_parameters() await movie.search_title("V 1983") self.assertEqual("V", movie.title) await movie.close_session() async def test_search_tvshow_v_1983(self): movie = TvShow() await movie.load_parameters() mid = await movie.search_title("V", 1983) self.assertEqual(mid, 14141) await movie.close_session() async def test_tv_title_v_1983(self): movie = TvShow() await movie.load_parameters() await movie.search_title("V", 1983) self.assertEqual("V", movie.title) await movie.close_session() async def test_cast_2(self): movie = TvShow() await movie.load_parameters() await movie.search_title("The Flash") self.assertEqual("Danielle Panabaker", movie.cast[2]) await movie.close_session() # async def test_poster_path(self): # movie = Movie() # await movie.load_parameters() # await movie.search_title("Dune") # self.assertEqual( # movie.poster_url, # "http://image.tmdb.org/t/p/w92/d5NXSklXo0qyIYkgV94XAgMIckC.jpg", # ) # await movie.close_session() async def test_movie_popular_length(self): results = await api_requests("/movie/popular") movie_list = MoviePopular() await movie_list.load_parameters() text = await movie_list.query() self.assertEqual(text, results["total_results"]) await movie_list.close_session() async def test_movie_popular_id(self): results = await api_requests("/movie/popular") movie_list = MoviePopular() await movie_list.load_parameters() await movie_list.query() self.assertEqual(movie_list.list[2]["id"], results["results"][2]["id"]) await movie_list.close_session() async def test_movie_popular_text(self): results = await api_requests("/movie/popular") movie_list = MoviePopular() await movie_list.load_parameters() await movie_list.query() test_result = results["results"][-1]["title"] tested = movie_list.getListText() tested = tested[(len(results["results"][-1]["title"])) * -1 :] self.assertEqual(tested, test_result) await movie_list.close_session() if __name__ == "__main__": unittest.main()