update unit tests

This commit is contained in:
Jesse Bannon 2023-01-11 21:45:08 -08:00
parent 7781827711
commit 3b0869b2f6
4 changed files with 26 additions and 25 deletions

View file

@ -48,6 +48,6 @@ def output_options() -> Dict:
@pytest.fixture @pytest.fixture
def youtube_video() -> Dict: def youtube_video() -> Dict:
return { return {
"download_strategy": "video", "download_strategy": "url",
"video_url": "youtube.com/watch?v=123abc", "url": "youtube.com/watch?v=123abc",
} }

View file

@ -66,12 +66,13 @@ class TestConfigFilePartiallyValidatesPresets:
f"Allowed fields: {', '.join(sorted(PRESET_KEYS))}", f"Allowed fields: {', '.join(sorted(PRESET_KEYS))}",
) )
def test_error__multiple_sources(self): # TODO: Update with future source
self._partial_validate( # def test_error__multiple_sources(self):
preset_dict={"youtube": {}, "download": {}}, # self._partial_validate(
expected_error_message="Validation error in partial_preset: " # preset_dict={"youtube": {}, "download": {}},
"Contains the sources download, youtube but can only have one", # 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): def test_error__no_download_strategy(self):
self._partial_validate( self._partial_validate(

View file

@ -10,18 +10,18 @@ class TestPreset:
@pytest.mark.parametrize( @pytest.mark.parametrize(
"source, download_strategy", "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", "download_strategy": "url",
"playlist_url": "youtube.com/playlist?list=123abc", "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",
{"download_strategy": "albums_and_singles", "url": "soundcloud.com/123abc"}, {"download_strategy": "url", "url": "soundcloud.com/123abc"},
), ),
], ],
) )
@ -37,7 +37,7 @@ class TestPreset:
config=config_file, config=config_file,
name="test", name="test",
value={ value={
"youtube": youtube_video, "download": youtube_video,
"output_options": {"output_directory": "dir", "file_name": "{dne_var}"}, "output_options": {"output_directory": "dir", "file_name": "{dne_var}"},
"overrides": {"dne_var": "not dne"}, "overrides": {"dne_var": "not dne"},
}, },
@ -49,7 +49,7 @@ class TestPreset:
name="test", name="test",
value={ value={
"preset": "parent_preset_1", "preset": "parent_preset_1",
"youtube": youtube_video, "download": youtube_video,
"output_options": output_options, "output_options": output_options,
"nfo_tags": {"tags": {"key-2": "this-preset"}}, "nfo_tags": {"tags": {"key-2": "this-preset"}},
}, },
@ -74,7 +74,7 @@ class TestPreset:
name="test", name="test",
value={ value={
"preset": preset_value, "preset": preset_value,
"youtube": youtube_video, "download": youtube_video,
"output_options": output_options, "output_options": output_options,
"nfo_tags": {"tags": {"key-3": "this-preset"}}, "nfo_tags": {"tags": {"key-3": "this-preset"}},
}, },
@ -97,7 +97,7 @@ class TestPreset:
config=config_file, config=config_file,
name="test", name="test",
value={ value={
"youtube": youtube_video, "download": youtube_video,
"output_options": dict( "output_options": dict(
output_options, output_options,
**{"maintain_download_archive": True, "keep_files_after": "today-{ttl}"}, **{"maintain_download_archive": True, "keep_files_after": "today-{ttl}"},
@ -122,7 +122,7 @@ class TestPreset:
name="test", name="test",
value={ value={
"preset": parent_preset, "preset": parent_preset,
"youtube": youtube_video, "download": youtube_video,
"output_options": output_options, "output_options": output_options,
}, },
) )
@ -138,7 +138,7 @@ class TestPreset:
config=config_file, config=config_file,
name="test", name="test",
value={ value={
"youtube": youtube_video, "download": youtube_video,
"output_options": {"output_directory": "dir", "file_name": "{dne_var}"}, "output_options": {"output_directory": "dir", "file_name": "{dne_var}"},
}, },
) )
@ -154,7 +154,7 @@ class TestPreset:
config=config_file, config=config_file,
name="test", name="test",
value={ value={
"youtube": youtube_video, "download": youtube_video,
"output_options": {"output_directory": "{dne_var}", "file_name": "file"}, "output_options": {"output_directory": "{dne_var}", "file_name": "file"},
}, },
) )
@ -170,7 +170,7 @@ class TestPreset:
config=config_file, config=config_file,
name="test", name="test",
value={ value={
"youtube": youtube_video, "download": youtube_video,
"output_options": {"output_directory": "dir", "file_name": "file"}, "output_options": {"output_directory": "dir", "file_name": "file"},
"nfo_tags": { "nfo_tags": {
"nfo_name": "the nfo name", "nfo_name": "the nfo name",
@ -191,7 +191,7 @@ class TestPreset:
config=config_file, config=config_file,
name="test", name="test",
value={ value={
"youtube": youtube_video, "download": youtube_video,
"output_options": {"output_directory": "dir", "file_name": "file"}, "output_options": {"output_directory": "dir", "file_name": "file"},
"output_directory_nfo_tags": { "output_directory_nfo_tags": {
"nfo_name": "the nfo name", "nfo_name": "the nfo name",

View file

@ -12,7 +12,7 @@ from ytdl_sub.subscriptions.subscription import Subscription
def preset_file(youtube_video: Dict, output_options: Dict) -> Dict: def preset_file(youtube_video: Dict, output_options: Dict) -> Dict:
return { return {
"__preset__": { "__preset__": {
"youtube": youtube_video, "download": youtube_video,
"output_options": output_options, "output_options": output_options,
"nfo_tags": { "nfo_tags": {
"tags": {"key-3": "file_preset"}, "tags": {"key-3": "file_preset"},