danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

error tests

Browse source
This commit is contained in:
jbannon 2022-06-04 23:55:26 +00:00
parent d5e669679a
commit 03008acde9
1 changed files with 34 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
tests/unit/cli/test_download_args_parser.py
View file

@ -1,6 +1,8 @@
import shlex import shlex
from argparse import ArgumentError
from typing import Callable from typing import Callable
from typing import Dict from typing import Dict
from typing import List
from typing import Optional from typing import Optional
import pytest import pytest
@ -12,7 +14,7 @@ from ytdl_sub.config.config_file import ConfigOptions
@pytest.fixture @pytest.fixture
def config_options_generator() -> Callable: def config_options_generator() -> Callable:
def _config_options_generator(dl_aliases: Optional[Dict[str, str]]): def _config_options_generator(dl_aliases: Optional[Dict[str, str]] = None):
config_options_dict = { config_options_dict = {
"working_directory": ".", "working_directory": ".",
} }
@ -24,6 +26,18 @@ def config_options_generator() -> Callable:
return _config_options_generator return _config_options_generator
@pytest.fixture
def argument_error_msg() -> str:
return "dl arguments must be in the form of --subscription.option.name 'value'"
def _get_extra_arguments(cmd_string: str) -> List[str]:
sys_argv = shlex.split(cmd_string)
_, extra_args = parser.parse_known_args(args=sys_argv)
return extra_args
class TestDownloadArgsParser: class TestDownloadArgsParser:
@pytest.mark.parametrize( @pytest.mark.parametrize(
"aliases, cmd, expected_sub_dict", "aliases, cmd, expected_sub_dict",
@ -52,12 +66,28 @@ class TestDownloadArgsParser:
) )
def test_successful_args(self, config_options_generator, aliases, cmd, expected_sub_dict): def test_successful_args(self, config_options_generator, aliases, cmd, expected_sub_dict):
config_options = config_options_generator(dl_aliases=aliases) config_options = config_options_generator(dl_aliases=aliases)
sys_argv = shlex.split(cmd) extra_args = _get_extra_arguments(cmd_string=cmd)
args, extra_args = parser.parse_known_args(args=sys_argv)
output_sub_dict = DownloadArgsParser( output_sub_dict = DownloadArgsParser(
extra_arguments=extra_args, config_options=config_options extra_arguments=extra_args, config_options=config_options
).to_subscription_dict() ).to_subscription_dict()
assert output_sub_dict == expected_sub_dict assert output_sub_dict == expected_sub_dict
def test_error_uneven_args(self, config_options_generator, argument_error_msg):
config_options = config_options_generator()
extra_args = _get_extra_arguments(cmd_string="dl --preset")
with pytest.raises(ArgumentError, match=argument_error_msg):
DownloadArgsParser(
extra_arguments=extra_args, config_options=config_options
).to_subscription_dict()
def test_error_adjacent_values(self, config_options_generator, argument_error_msg):
config_options = config_options_generator()
extra_args = _get_extra_arguments(cmd_string="dl --preset buttered toast --test")
with pytest.raises(ArgumentError, match=argument_error_msg):
DownloadArgsParser(
extra_arguments=extra_args, config_options=config_options
).to_subscription_dict()