1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
|
import pytest
from subliminal_patch.core import Episode
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"
@pytest.fixture
def provider():
with Subf2mProvider(user_agent=_U_A) as provider:
yield provider
@pytest.mark.parametrize(
"title,year,expected_url",
[
(
"Dead Man's Chest",
2006,
"/subtitles/pirates-of-the-caribbean-2-dead-mans-chest",
),
("Dune", 2021, "/subtitles/dune-2021"),
("Cure", 1997, "/subtitles/cure-kyua"),
],
)
def test_search_movie(provider, title, year, expected_url):
result = provider._search_movie(title, year)
assert expected_url in result
def test_init_empty_user_agent_raises_configurationerror():
with pytest.raises(ConfigurationError):
with Subf2mProvider(user_agent=" ") as provider:
assert provider
@pytest.mark.parametrize(
"series_title,season,year,expected_url",
[
("Breaking Bad", 1, None, "/subtitles/breaking-bad-first-season"),
("House Of The Dragon", 1, None, "/subtitles/house-of-the-dragon-first-season"),
("The Bear", 1, None, "/subtitles/the-bear-first-season"),
("Courage the Cowardly Dog", 1, None, "/subtitles/courage-the-cowardly-dog"),
(
"The Twilight Zone",
2,
1959,
"/subtitles/the-twilight-zone-the-complete-original-series",
),
],
)
def test_search_tv_show_season(provider, series_title, season, year, expected_url):
result = provider._search_tv_show_season(series_title, season, year)
assert expected_url in result
@pytest.mark.parametrize("language", [Language.fromalpha2("en"), Language("por", "BR")])
def test_find_movie_subtitles(provider, language, movies):
path = "/subtitles/dune-2021"
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, episodes):
path = "/subtitles/breaking-bad-first-season"
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"
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")
def test_list_and_download_subtitles_complete_series_pack(provider, episodes):
episode = list(episodes.values())[0]
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
subtitles = provider.list_subtitles(episode, {Language.fromalpha2("en")})
assert subtitles
subtitle = subtitles[0]
provider.download_subtitle(subtitle)
assert subtitle.is_valid()
@pytest.fixture
def subtitle():
release_info = """Dune-2021.All.WEBDLL
Dune.2021.WEBRip.XviD.MP3-XVID
Dune.2021.WEBRip.XviD.MP3-SHITBOX
Dune.2021.WEBRip.x264-SHITBOX
Dune.2021.WEBRip.x264-ION10
Dune.2021.HDRip.XviD-EVO[TGx]
Dune.2021.HDRip.XviD-EVO
Dune.2021.720p.HDRip.900MB.x264-GalaxyRG
Dune.2021.1080p.HDRip.X264-EVO
Dune.2021.1080p.HDRip.1400MB.x264-GalaxyRG"""
return Subf2mSubtitle(
Language.fromalpha3b("per"),
"https://subf2m.co/subtitles/dune-2021/farsi_persian/2604701",
release_info,
)
@pytest.fixture
def subtitle_episode():
return Subf2mSubtitle(
Language.fromalpha2("en"),
"https://subf2m.co/subtitles/breaking-bad-first-season/english/161227",
"Breaking.Bad.S01E01-7.DSR-HDTV.eng",
7,
)
def test_subtitle_get_matches(subtitle, movies):
matches = subtitle.get_matches(movies["dune"]) # type: set
assert matches.issuperset(
("title", "year", "source", "video_codec", "resolution", "release_group")
)
def test_subtitle_get_matches_episode(subtitle_episode, episodes):
matches = subtitle_episode.get_matches(episodes["breaking_bad_s01e01"]) # type: set
assert matches.issuperset(("title", "series", "season", "episode"))
assert "source" not in matches
def test_list_subtitles_movie(provider, movies):
assert provider.list_subtitles(movies["dune"], {Language.fromalpha2("en")})
def test_list_subtitles_inexistent_movie(provider, movies):
assert (
provider.list_subtitles(movies["inexistent"], {Language.fromalpha2("en")}) == []
)
def test_list_subtitles_episode(provider, episodes):
assert provider.list_subtitles(
episodes["breaking_bad_s01e01"], {Language.fromalpha2("en")}
)
def test_list_subtitles_inexistent_episode(provider, episodes):
assert (
provider.list_subtitles(episodes["inexistent"], {Language.fromalpha2("en")})
== []
)
def test_download_subtitle(provider, subtitle):
provider.download_subtitle(subtitle)
assert subtitle.is_valid()
def test_download_subtitle_episode(provider, subtitle_episode):
provider.download_subtitle(subtitle_episode)
assert subtitle_episode.is_valid()
def test_download_subtitle_episode_with_title(provider):
sub = Subf2mSubtitle(
Language.fromalpha2("en"),
"https://subf2m.co/subtitles/courage-the-cowardly-dog/english/2232402",
"Season 3 complete.",
13,
)
sub.episode_title = "Feast of the Bullfrogs"
provider.download_subtitle(sub)
assert sub.is_valid()
def test_get_episode_from_release():
assert subf2m._get_episode_from_release(
"Vinland Saga Season 2 - 05 [Crunchyroll][Crunchyroll] Vinland Saga Season 2 - 05"
) == {"season": [2], "episode": [5]}
def test_get_episode_from_release_return_none():
assert subf2m._get_episode_from_release("Vinland Saga Season 2 - Foo") is None
def test_get_episode_from_release_w_empty_match_return_none():
assert subf2m._get_episode_from_release("Vinland Saga - 02") is None
def test_complex_episode_name(provider):
episode = Episode(
**{
"name": "Dr.Romantic.S03E16.SBS.x265.1080p-thon.mkv",
"source": "HDTV",
"release_group": "thon",
"resolution": "1080p",
"video_codec": "H.265",
"audio_codec": "AAC",
"subtitle_languages": set(),
"original_name": "Dr. Romantic - S03E16.mkv",
"other": None,
"series": "Dr. Romantic",
"season": 3,
"episode": 16,
"title": "Dreamers",
"year": 2016,
"series_imdb_id": "tt6157190",
"alternative_series": ["Romantic Doctor Teacher Kim"],
}
)
assert provider.list_subtitles(episode, {Language.fromietf("en")})
|
