75 lines
1.9 KiB
Python
75 lines
1.9 KiB
Python
|
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):
|
||
|
|
choiceoption = ChoiceOption('cmd',
|
||
|
|
'choice the sub argument',
|
||
|
|
('str', 'list', 'int', 'none'),
|
||
|
|
properties=('mandatory',
|
||
|
|
'positional'))
|
||
|
|
od = OptionDescription('od',
|
||
|
|
'od',
|
||
|
|
[choiceoption])
|
||
|
|
root = OptionDescription('root',
|
||
|
|
'root',
|
||
|
|
[od])
|
||
|
|
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_help(json):
|
||
|
|
output = """usage: prog.py [-h] {str,list,int,none}
|
||
|
|
|
||
|
|
optional arguments:
|
||
|
|
-h, --help show this help message and exit
|
||
|
|
|
||
|
|
od:
|
||
|
|
od
|
||
|
|
|
||
|
|
{str,list,int,none} choice the sub argument
|
||
|
|
"""
|
||
|
|
parser = TiramisuCmdlineParser(get_config(json), 'prog.py')
|
||
|
|
f = StringIO()
|
||
|
|
with redirect_stdout(f):
|
||
|
|
parser.print_help()
|
||
|
|
assert f.getvalue() == output
|
||
|
|
|
||
|
|
|
||
|
|
def test_help_epilog(json):
|
||
|
|
output = """usage: prog.py [-h] {str,list,int,none}
|
||
|
|
|
||
|
|
optional arguments:
|
||
|
|
-h, --help show this help message and exit
|
||
|
|
|
||
|
|
od:
|
||
|
|
od
|
||
|
|
|
||
|
|
{str,list,int,none} choice the sub argument
|
||
|
|
|
||
|
|
two
|
||
|
|
line
|
||
|
|
"""
|
||
|
|
parser = TiramisuCmdlineParser(get_config(json), 'prog.py', epilog="\ntwo\nline")
|
||
|
|
f = StringIO()
|
||
|
|
with redirect_stdout(f):
|
||
|
|
parser.print_help()
|
||
|
|
assert f.getvalue() == output
|