diff options
author | Vitiko <[email protected]> | 2023-06-13 02:13:42 -0400 |
---|---|---|
committer | Vitiko <[email protected]> | 2023-06-13 02:13:42 -0400 |
commit | ee1506ed54b5a610dca7ca47ff6a09f13773c84e (patch) | |
tree | 87aaecde0a249329245dc6ffd608637639503675 /tests | |
parent | e3c4def89acc185bdbb67f8dcb7a12531e209df7 (diff) | |
download | bazarr-ee1506ed54b5a610dca7ca47ff6a09f13773c84e.tar.gz bazarr-ee1506ed54b5a610dca7ca47ff6a09f13773c84e.zip |
Subf2m provider: add support for IMDB IDs
Diffstat (limited to 'tests')
-rw-r--r-- | tests/subliminal_patch/test_subf2m.py | 39 |
1 files changed, 26 insertions, 13 deletions
diff --git a/tests/subliminal_patch/test_subf2m.py b/tests/subliminal_patch/test_subf2m.py index cdf201734..2df17af2d 100644 --- a/tests/subliminal_patch/test_subf2m.py +++ b/tests/subliminal_patch/test_subf2m.py @@ -1,10 +1,10 @@ import pytest from subliminal_patch.providers import subf2m +from subliminal_patch.providers.subf2m import ConfigurationError from subliminal_patch.providers.subf2m import Subf2mProvider from subliminal_patch.providers.subf2m import Subf2mSubtitle from subzero.language import Language - _U_A = "Mozilla/5.0 (Linux; Android 10; SM-G996U Build/QP1A.190711.020; wv) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Mobile Safari/537.36" @@ -26,13 +26,15 @@ def provider(): ("Cure", 1997, "/subtitles/cure-kyua"), ], ) -def test_search_movie(provider, movies, title, year, expected_url): - movie = list(movies.values())[0] - movie.title = title - movie.year = year +def test_search_movie(provider, title, year, expected_url): + result = provider._search_movie(title, year) + assert expected_url in result + - result = provider._search_movie(movie.title, movie.year) - assert result == expected_url +def test_init_empty_user_agent_raises_configurationerror(): + with pytest.raises(ConfigurationError): + with Subf2mProvider(user_agent=" ") as provider: + assert provider @pytest.mark.parametrize( @@ -52,27 +54,37 @@ def test_search_movie(provider, movies, title, year, expected_url): ) def test_search_tv_show_season(provider, series_title, season, year, expected_url): result = provider._search_tv_show_season(series_title, season, year) - assert result == expected_url + assert expected_url in result @pytest.mark.parametrize("language", [Language.fromalpha2("en"), Language("por", "BR")]) -def test_find_movie_subtitles(provider, language): +def test_find_movie_subtitles(provider, language, movies): path = "/subtitles/dune-2021" - for sub in provider._find_movie_subtitles(path, language): + for sub in provider._find_movie_subtitles(path, language, movies["dune"].imdb_id): assert sub.language == language @pytest.mark.parametrize("language", [Language.fromalpha2("en"), Language("por", "BR")]) -def test_find_episode_subtitles(provider, language): +def test_find_episode_subtitles(provider, language, episodes): path = "/subtitles/breaking-bad-first-season" - for sub in provider._find_episode_subtitles(path, 1, 1, language): + subs = provider._find_episode_subtitles( + path, 1, 1, language, imdb_id=episodes["breaking_bad_s01e01"].series_imdb_id + ) + assert subs + + for sub in subs: assert sub.language == language def test_find_episode_subtitles_from_complete_series_path(provider): path = "/subtitles/courage-the-cowardly-dog" - for sub in provider._find_episode_subtitles(path, 1, 1, Language.fromalpha2("en")): + subs = provider._find_episode_subtitles( + path, 1, 1, Language.fromalpha2("en"), imdb_id="tt0220880" + ) + assert subs + + for sub in subs: assert sub.language == Language.fromalpha2("en") @@ -82,6 +94,7 @@ def test_list_and_download_subtitles_complete_series_pack(provider, episodes): episode.series = "Sam & Max: Freelance Police" episode.name = "The Glazed McGuffin Affair" episode.title = "The Glazed McGuffin Affair" + episode.series_imdb_id = "tt0125646" episode.season = 1 episode.episode = 21 |