update unit tests
This commit is contained in:
parent
7781827711
commit
3b0869b2f6
4 changed files with 26 additions and 25 deletions
|
|
@ -48,6 +48,6 @@ def output_options() -> Dict:
|
|||
@pytest.fixture
|
||||
def youtube_video() -> Dict:
|
||||
return {
|
||||
"download_strategy": "video",
|
||||
"video_url": "youtube.com/watch?v=123abc",
|
||||
"download_strategy": "url",
|
||||
"url": "youtube.com/watch?v=123abc",
|
||||
}
|
||||
|
|
|
|||
|
|
@ -66,12 +66,13 @@ class TestConfigFilePartiallyValidatesPresets:
|
|||
f"Allowed fields: {', '.join(sorted(PRESET_KEYS))}",
|
||||
)
|
||||
|
||||
def test_error__multiple_sources(self):
|
||||
self._partial_validate(
|
||||
preset_dict={"youtube": {}, "download": {}},
|
||||
expected_error_message="Validation error in partial_preset: "
|
||||
"Contains the sources download, youtube but can only have one",
|
||||
)
|
||||
# TODO: Update with future source
|
||||
# def test_error__multiple_sources(self):
|
||||
# self._partial_validate(
|
||||
# preset_dict={"youtube": {}, "download": {}},
|
||||
# expected_error_message="Validation error in partial_preset: "
|
||||
# "Contains the sources download, youtube but can only have one",
|
||||
# )
|
||||
|
||||
def test_error__no_download_strategy(self):
|
||||
self._partial_validate(
|
||||
|
|
|
|||
|
|
@ -10,18 +10,18 @@ class TestPreset:
|
|||
@pytest.mark.parametrize(
|
||||
"source, download_strategy",
|
||||
[
|
||||
("youtube", {"download_strategy": "video", "video_url": "youtube.com/watch?v=123abc"}),
|
||||
("download", {"download_strategy": "url", "url": "youtube.com/watch?v=123abc"}),
|
||||
(
|
||||
"youtube",
|
||||
"download",
|
||||
{
|
||||
"download_strategy": "playlist",
|
||||
"playlist_url": "youtube.com/playlist?list=123abc",
|
||||
"download_strategy": "url",
|
||||
"url": "youtube.com/playlist?list=123abc",
|
||||
},
|
||||
),
|
||||
("youtube", {"download_strategy": "channel", "channel_url": "youtube.com/c/123abc"}),
|
||||
("download", {"download_strategy": "url", "url": "youtube.com/c/123abc"}),
|
||||
(
|
||||
"soundcloud",
|
||||
{"download_strategy": "albums_and_singles", "url": "soundcloud.com/123abc"},
|
||||
"download",
|
||||
{"download_strategy": "url", "url": "soundcloud.com/123abc"},
|
||||
),
|
||||
],
|
||||
)
|
||||
|
|
@ -37,7 +37,7 @@ class TestPreset:
|
|||
config=config_file,
|
||||
name="test",
|
||||
value={
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": {"output_directory": "dir", "file_name": "{dne_var}"},
|
||||
"overrides": {"dne_var": "not dne"},
|
||||
},
|
||||
|
|
@ -49,7 +49,7 @@ class TestPreset:
|
|||
name="test",
|
||||
value={
|
||||
"preset": "parent_preset_1",
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": output_options,
|
||||
"nfo_tags": {"tags": {"key-2": "this-preset"}},
|
||||
},
|
||||
|
|
@ -74,7 +74,7 @@ class TestPreset:
|
|||
name="test",
|
||||
value={
|
||||
"preset": preset_value,
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": output_options,
|
||||
"nfo_tags": {"tags": {"key-3": "this-preset"}},
|
||||
},
|
||||
|
|
@ -97,7 +97,7 @@ class TestPreset:
|
|||
config=config_file,
|
||||
name="test",
|
||||
value={
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": dict(
|
||||
output_options,
|
||||
**{"maintain_download_archive": True, "keep_files_after": "today-{ttl}"},
|
||||
|
|
@ -122,7 +122,7 @@ class TestPreset:
|
|||
name="test",
|
||||
value={
|
||||
"preset": parent_preset,
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": output_options,
|
||||
},
|
||||
)
|
||||
|
|
@ -138,7 +138,7 @@ class TestPreset:
|
|||
config=config_file,
|
||||
name="test",
|
||||
value={
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": {"output_directory": "dir", "file_name": "{dne_var}"},
|
||||
},
|
||||
)
|
||||
|
|
@ -154,7 +154,7 @@ class TestPreset:
|
|||
config=config_file,
|
||||
name="test",
|
||||
value={
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": {"output_directory": "{dne_var}", "file_name": "file"},
|
||||
},
|
||||
)
|
||||
|
|
@ -170,7 +170,7 @@ class TestPreset:
|
|||
config=config_file,
|
||||
name="test",
|
||||
value={
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": {"output_directory": "dir", "file_name": "file"},
|
||||
"nfo_tags": {
|
||||
"nfo_name": "the nfo name",
|
||||
|
|
@ -191,7 +191,7 @@ class TestPreset:
|
|||
config=config_file,
|
||||
name="test",
|
||||
value={
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": {"output_directory": "dir", "file_name": "file"},
|
||||
"output_directory_nfo_tags": {
|
||||
"nfo_name": "the nfo name",
|
||||
|
|
|
|||
|
|
@ -12,7 +12,7 @@ from ytdl_sub.subscriptions.subscription import Subscription
|
|||
def preset_file(youtube_video: Dict, output_options: Dict) -> Dict:
|
||||
return {
|
||||
"__preset__": {
|
||||
"youtube": youtube_video,
|
||||
"download": youtube_video,
|
||||
"output_options": output_options,
|
||||
"nfo_tags": {
|
||||
"tags": {"key-3": "file_preset"},
|
||||
|
|
|
|||
Loading…
Reference in a new issue