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"])
