def test_tmdb_find__imdb_success(): expected_top_level_keys = { "movie_results", "person_results", "tv_episode_results", "tv_results", "tv_season_results", } expected_movie_results_keys = { "adult", "backdrop_path", "genre_ids", "id", "original_language", "original_title", "overview", "poster_path", "popularity", "release_date", "title", "video", "vote_average", "vote_count", } result = tmdb_find(Tmdb.api_key, "imdb_id", GOONIES_IMDB_ID) assert isinstance(result, dict) assert set(result.keys()) == expected_top_level_keys assert len(result.get("movie_results", {})) > 0 assert expected_movie_results_keys == set( result.get("movie_results", {})[0].keys())
def test_tmdb_find__not_found(): with pytest.raises(MnamerNotFoundException): tmdb_find(Tmdb.api_key, "imdb_id", JUNK_IMDB_ID)
def test_tmdb_find__invalid_id_imdb(): with pytest.raises(MnamerException): tmdb_find(Tmdb.api_key, "imdb_id", JUNK_TEXT, cache=False)
def test_tmdb_find__api_key_fail(): with pytest.raises(MnamerException): tmdb_find(JUNK_TEXT, "imdb_id", GOONIES_IMDB_ID, cache=False)
def test_tmdb_find__invalid_source(): with pytest.raises(MnamerException): tmdb_find(Tmdb.api_key, "abc123", GOONIES_IMDB_ID, cache=False)
def test_tmdb_find__language(): results = tmdb_find(Tmdb.api_key, "imdb_id", GOONIES_IMDB_ID, RUSSIAN_LANG) assert any(result["title"] == "Балбесы" for result in results["movie_results"])