support suboptiondescription in TiramisuCmdLineParser
This commit is contained in:
@ -50,6 +50,8 @@ optional arguments:
|
||||
-h, --help show this help message and exit
|
||||
|
||||
leader:
|
||||
leader
|
||||
|
||||
--leader.leader [LEADER [LEADER ...]]
|
||||
Leader var
|
||||
--leader.follower INDEX [FOLLOWER]
|
||||
|
||||
121
test/test_optiondescription.py
Normal file
121
test/test_optiondescription.py
Normal file
@ -0,0 +1,121 @@
|
||||
from io import StringIO
|
||||
from contextlib import redirect_stdout, redirect_stderr
|
||||
import pytest
|
||||
|
||||
|
||||
from tiramisu_cmdline_parser import TiramisuCmdlineParser
|
||||
from tiramisu import IntOption, StrOption, BoolOption, ChoiceOption, \
|
||||
SymLinkOption, OptionDescription, Config
|
||||
from tiramisu_json_api import Config as JsonConfig
|
||||
|
||||
|
||||
def get_config(json, has_tree=False, default_verbosity=False, add_long=False, add_store_false=False):
|
||||
choiceoption = ChoiceOption('cmd',
|
||||
'choice the sub argument',
|
||||
('str', 'list', 'int', 'none'),
|
||||
properties=('mandatory',
|
||||
'positional'))
|
||||
booloption = BoolOption('verbosity',
|
||||
'increase output verbosity',
|
||||
default=default_verbosity)
|
||||
short_booloption = SymLinkOption('v', booloption)
|
||||
|
||||
od0 = OptionDescription('od0',
|
||||
'Sub-Tree 1',
|
||||
[choiceoption,
|
||||
booloption,
|
||||
short_booloption,
|
||||
])
|
||||
od1 = OptionDescription('od1',
|
||||
'First OptionDescription',
|
||||
[od0])
|
||||
before = StrOption('before',
|
||||
'Before',
|
||||
properties=('mandatory',))
|
||||
after = StrOption('after',
|
||||
'After',
|
||||
properties=('mandatory',))
|
||||
str_ = StrOption('str',
|
||||
'string option 2',
|
||||
properties=('mandatory',))
|
||||
subtree = OptionDescription('subtree',
|
||||
'Sub-Tree 2',
|
||||
[str_])
|
||||
od2 = OptionDescription('od2',
|
||||
'Second OptionDescription',
|
||||
[before, subtree, after])
|
||||
root = OptionDescription('root',
|
||||
'root',
|
||||
[od1, od2])
|
||||
config = Config(root)
|
||||
config.property.read_write()
|
||||
if json == 'tiramisu':
|
||||
return config
|
||||
jconfig = JsonConfig(config.option.dict())
|
||||
return jconfig
|
||||
|
||||
|
||||
@pytest.fixture(params=['tiramisu', 'tiramisu-json'])
|
||||
def json(request):
|
||||
return request.param
|
||||
|
||||
|
||||
def test_optiondescription_help(json):
|
||||
output = """usage: prog.py [-h] [-v] [-nv] --od2.subtree.str STR --od2.before BEFORE
|
||||
--od2.after AFTER
|
||||
{str,list,int,none}
|
||||
|
||||
optional arguments:
|
||||
-h, --help show this help message and exit
|
||||
|
||||
od1:
|
||||
First OptionDescription
|
||||
|
||||
od1.od0:
|
||||
Sub-Tree 1
|
||||
|
||||
{str,list,int,none} choice the sub argument
|
||||
-v, --od1.od0.verbosity
|
||||
increase output verbosity
|
||||
-nv, --od1.od0.no-verbosity
|
||||
|
||||
od2:
|
||||
Second OptionDescription
|
||||
|
||||
--od2.before BEFORE Before
|
||||
--od2.after AFTER After
|
||||
|
||||
od2.subtree:
|
||||
Sub-Tree 2
|
||||
|
||||
--od2.subtree.str STR
|
||||
string option 2
|
||||
"""
|
||||
parser = TiramisuCmdlineParser(get_config(json), 'prog.py')
|
||||
f = StringIO()
|
||||
with redirect_stdout(f):
|
||||
parser.print_help()
|
||||
assert f.getvalue() == output
|
||||
|
||||
|
||||
def test_optiondescription_help_subtree(json):
|
||||
output = """usage: prog.py [-h] --od2.subtree.str STR --od2.before BEFORE --od2.after
|
||||
AFTER
|
||||
|
||||
optional arguments:
|
||||
-h, --help show this help message and exit
|
||||
--od2.before BEFORE Before
|
||||
--od2.after AFTER After
|
||||
|
||||
od2.subtree:
|
||||
Sub-Tree 2
|
||||
|
||||
--od2.subtree.str STR
|
||||
string option 2
|
||||
"""
|
||||
config = get_config(json)
|
||||
parser = TiramisuCmdlineParser(config, 'prog.py', root='od2')
|
||||
f = StringIO()
|
||||
with redirect_stdout(f):
|
||||
parser.print_help()
|
||||
assert f.getvalue() == output
|
||||
@ -97,6 +97,8 @@ optional arguments:
|
||||
-h, --help show this help message and exit
|
||||
|
||||
root:
|
||||
root
|
||||
|
||||
{str,list,int,none} choice the sub argument
|
||||
-v, --root.verbosity increase output verbosity
|
||||
-nv, --root.no-verbosity
|
||||
@ -115,6 +117,8 @@ optional arguments:
|
||||
-h, --help show this help message and exit
|
||||
|
||||
root:
|
||||
root
|
||||
|
||||
{str,list,int,none} choice the sub argument
|
||||
-v, --verbosity increase output verbosity
|
||||
-nv, --no-verbosity
|
||||
|
||||
@ -14,6 +14,86 @@ def json(request):
|
||||
return request.param
|
||||
|
||||
|
||||
def test_short(json):
|
||||
def get_config():
|
||||
list_ = StrOption('list',
|
||||
'list string option')
|
||||
slist_ = SymLinkOption('l', list_)
|
||||
root = OptionDescription('root',
|
||||
'root',
|
||||
[list_,
|
||||
slist_,
|
||||
])
|
||||
config = Config(root)
|
||||
config.property.read_write()
|
||||
if json != 'tiramisu':
|
||||
config = JsonConfig(config.option.dict())
|
||||
return config
|
||||
#
|
||||
output = {'list': None, 'l': None}
|
||||
config = get_config()
|
||||
parser = TiramisuCmdlineParser(config, 'prog.py')
|
||||
parser.parse_args([])
|
||||
assert config.value.dict() == output
|
||||
#
|
||||
output = {'list': 'a', 'l': 'a'}
|
||||
config = get_config()
|
||||
parser = TiramisuCmdlineParser(config, 'prog.py')
|
||||
parser.parse_args(['--list', 'a'])
|
||||
assert config.value.dict() == output
|
||||
#
|
||||
output = {'list': 'a', 'l': 'a'}
|
||||
config = get_config()
|
||||
parser = TiramisuCmdlineParser(config, 'prog.py')
|
||||
parser.parse_args(['-l', 'a'])
|
||||
assert config.value.dict() == output
|
||||
|
||||
|
||||
def test_short_mandatory(json):
|
||||
def get_config():
|
||||
list_ = StrOption('list',
|
||||
'list string option',
|
||||
properties=('mandatory',))
|
||||
slist_ = SymLinkOption('l', list_)
|
||||
root = OptionDescription('root',
|
||||
'root',
|
||||
[list_,
|
||||
slist_,
|
||||
])
|
||||
config = Config(root)
|
||||
config.property.read_write()
|
||||
if json != 'tiramisu':
|
||||
config = JsonConfig(config.option.dict())
|
||||
return config
|
||||
#
|
||||
output = """usage: prog.py [-h] -l LIST
|
||||
prog.py: error: the following arguments are required: --list
|
||||
"""
|
||||
config = get_config()
|
||||
parser = TiramisuCmdlineParser(config, 'prog.py')
|
||||
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
|
||||
#
|
||||
output = {'list': 'a', 'l': 'a'}
|
||||
config = get_config()
|
||||
parser = TiramisuCmdlineParser(config, 'prog.py')
|
||||
parser.parse_args(['--list', 'a'])
|
||||
assert config.value.dict() == output
|
||||
#
|
||||
output = {'list': 'a', 'l': 'a'}
|
||||
config = get_config()
|
||||
parser = TiramisuCmdlineParser(config, 'prog.py')
|
||||
parser.parse_args(['-l', 'a'])
|
||||
assert config.value.dict() == output
|
||||
|
||||
|
||||
def test_short_multi(json):
|
||||
def get_config():
|
||||
list_ = StrOption('list',
|
||||
|
||||
Reference in New Issue
Block a user