add test for README example

This commit is contained in:
Emmanuel Garette 2018-12-01 09:53:31 +01:00 committed by Emmanuel Garette
parent b026f9755c
commit 3372a42524
3 changed files with 246 additions and 9 deletions

View file

@ -49,6 +49,7 @@ int_ = IntOption('int',
'expected': 'int', 'expected': 'int',
'action': 'disabled', 'action': 'disabled',
'inverse': True}]) 'inverse': True}])
# Now build Config
config = Config(OptionDescription('root', config = Config(OptionDescription('root',
'root', 'root',
[choiceoption, [choiceoption,
@ -110,6 +111,15 @@ usage: prog.py [-h] [-v] --str STR --list LIST [LIST ...] --int INT
prog.py: error: the following arguments are required: --str prog.py: error: the following arguments are required: --str
``` ```
If 'cmd' is 'str', cannot set --int argument:
```bash
[gnunux@localhost tiramisu-parser]$ python3 prog.py str --int 3
usage: prog.py [-h] [-v] --str STR --list LIST [LIST ...] --int INT
{str,list,int}
prog.py: error: unrecognized arguments: --int
```
With all mandatories arguments: With all mandatories arguments:
```bash ```bash
@ -138,3 +148,20 @@ result:
- v (increase output verbosity): False - v (increase output verbosity): False
- list (list string option): ['a', 'b', 'c'] - list (list string option): ['a', 'b', 'c']
``` ```
Add --verbosity argument:
```bash
[gnunux@localhost tiramisu-parser]$ python3 prog.py list --list a b c -v
result:
- cmd (choice the sub argument): list
- verbosity (increase output verbosity): True
- v (increase output verbosity): True
- list (list string option): ['a', 'b', 'c']
[gnunux@localhost tiramisu-parser]$ python3 prog.py list --list a b c --verbosity
result:
- cmd (choice the sub argument): list
- verbosity (increase output verbosity): True
- v (increase output verbosity): True
- list (list string option): ['a', 'b', 'c']
```

210
test/test_readme.py Normal file
View file

@ -0,0 +1,210 @@
from io import StringIO
from contextlib import redirect_stdout, redirect_stderr
from tiramisu_cmdline_parser import TiramisuCmdlineParser
from tiramisu import IntOption, StrOption, BoolOption, ChoiceOption, \
SymLinkOption, OptionDescription, Config
def get_config():
choiceoption = ChoiceOption('cmd',
'choice the sub argument',
('str', 'list', 'int'),
properties=('mandatory',
'positional'))
booloption = BoolOption('verbosity',
'increase output verbosity',
default=False)
short_booloption = SymLinkOption('v', booloption)
str_ = StrOption('str',
'string option',
properties=('mandatory',),
requires=[{'option': choiceoption,
'expected': 'str',
'action': 'disabled',
'inverse': True}])
list_ = StrOption('list',
'list string option',
multi=True,
properties=('mandatory',),
requires=[{'option': choiceoption,
'expected': 'list',
'action': 'disabled',
'inverse': True}])
int_ = IntOption('int',
'int option',
properties=('mandatory',),
requires=[{'option': choiceoption,
'expected': 'int',
'action': 'disabled',
'inverse': True}])
return Config(OptionDescription('root',
'root',
[choiceoption,
booloption,
short_booloption,
str_,
list_,
int_
]))
def test_readme_help():
output = """usage: prog.py [-h] [-v] --str STR --list LIST [LIST ...] --int INT
{str,list,int}
positional arguments:
{str,list,int} choice the sub argument
optional arguments:
-h, --help show this help message and exit
-v, --verbosity increase output verbosity
--str STR string option
--list LIST [LIST ...]
list string option
--int INT int option
"""
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(get_config())
f = StringIO()
with redirect_stdout(f):
parser.print_help()
assert f.getvalue() == output
def test_readme_positional_mandatory():
output = """usage: prog.py [-h] [-v] --str STR --list LIST [LIST ...] --int INT
{str,list,int}
prog.py: error: the following arguments are required: cmd
"""
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(get_config())
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args([])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_mandatory():
output = """usage: prog.py [-h] [-v] --str STR --list LIST [LIST ...] --int INT
{str,list,int}
prog.py: error: the following arguments are required: --str
"""
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(get_config())
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args(['str'])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_cross():
output = """usage: prog.py [-h] [-v] --str STR --list LIST [LIST ...] --int INT
{str,list,int}
prog.py: error: unrecognized arguments: --int
"""
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(get_config())
f = StringIO()
with redirect_stderr(f):
try:
parser.parse_args(['str', '--int', '3'])
except SystemExit as err:
assert str(err) == "2"
else:
raise Exception('must raises')
assert f.getvalue() == output
def test_readme_int():
output = {'cmd': 'int',
'int': 3,
'verbosity': False,
'v': False}
config = get_config()
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(config)
parser.parse_args(['int', '--int', '3'])
assert config.value.dict() == output
def test_readme_int_verbosity():
output = {'cmd': 'int',
'int': 3,
'verbosity': True,
'v': True}
config = get_config()
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(config)
parser.parse_args(['int', '--int', '3', '--verbosity'])
assert config.value.dict() == output
def test_readme_int_verbosity_short():
output = {'cmd': 'int',
'int': 3,
'verbosity': True,
'v': True}
config = get_config()
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(config)
parser.parse_args(['int', '--int', '3', '-v'])
assert config.value.dict() == output
def test_readme_str():
output = {'cmd': 'str',
'str': 'value',
'verbosity': False,
'v': False}
config = get_config()
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(config)
parser.parse_args(['str', '--str', 'value'])
assert config.value.dict() == output
def test_readme_str_int():
output = {'cmd': 'str',
'str': '3',
'verbosity': False,
'v': False}
config = get_config()
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(config)
parser.parse_args(['str', '--str', '3'])
assert config.value.dict() == output
def test_readme_list():
output = {'cmd': 'list',
'list': ['a', 'b', 'c'],
'verbosity': False,
'v': False}
config = get_config()
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(config)
parser.parse_args(['list', '--list', 'a', 'b', 'c'])
assert config.value.dict() == output
def test_readme_list_uniq():
output = {'cmd': 'list',
'list': ['a'],
'verbosity': False,
'v': False}
config = get_config()
parser = TiramisuCmdlineParser('prog.py')
parser.add_arguments(config)
parser.parse_args(['list', '--list', 'a'])
assert config.value.dict() == output

View file

@ -158,21 +158,21 @@ class TiramisuCmdlineParser(ArgumentParser):
def parse_args(self, *args, **kwargs): def parse_args(self, *args, **kwargs):
kwargs['namespace'] = TiramisuNamespace(self.config) kwargs['namespace'] = TiramisuNamespace(self.config)
namespaces = super().parse_args(*args, **kwargs)
try: try:
namespaces = super().parse_args(*args, **kwargs)
del namespaces.__dict__['_config'] del namespaces.__dict__['_config']
except PropertiesOptionError as err: except PropertiesOptionError as err:
if err.proptype == ['mandatory']:
name = err._option_bag.option.impl_getname() name = err._option_bag.option.impl_getname()
properties = self.config.option(name).property.get() properties = self.config.unrestraint.option(name).property.get()
if 'positional' not in properties: if 'positional' not in properties:
if len(name) == 1 and 'longargument' not in properties: if len(name) == 1 and 'longargument' not in properties:
name = self.prefix_chars + name name = self.prefix_chars + name
else: else:
name = self.prefix_chars * 2 + name name = self.prefix_chars * 2 + name
if err.proptype == ['mandatory']:
self.error('the following arguments are required: {}'.format(name)) self.error('the following arguments are required: {}'.format(name))
else: else:
self.error('unexpected error: {}'.format(err)) self.error('unrecognized arguments: {}'.format(name))
return namespaces return namespaces
def format_usage(self, *args, _forhelp=False, **kwargs): def format_usage(self, *args, _forhelp=False, **kwargs):