def test_ambiguity():
c = Command('ndcli',
Argument('var', nargs='*', stop_at='end'),
Group(Token('end'), Argument('endval'), nargs='?'))
assert c.parse(['a']).values == {'var': ['a'], 'endval': None}
assert c.parse(['a', 'end', '1']).values == {'var': ['a'], 'endval': '1'}
assert c.parse('a end b end c'.split()).values == {'var': ['a', 'end', 'b'], 'endval': 'c'}
def test_star_star():
c = Command('ndcli',
Group(Argument('fields', choices=['a', 'b'], action='append_unique'), Argument('values', action='append'), nargs='*'),
Argument('second', nargs='*'))
assert c.parse(['a', '1', '2', '3']).values == {'fields': ['a'], 'values': ['1'], 'second': ['2', '3']}
assert c.parse(['a', '1', 'b', 'c', '2', '3']).values == {'fields': ['a', 'b'], 'second': ['2', '3'], 'values': ['1', 'c']}
assert c.parse([]).values == {'fields': [], 'second': [], 'values': []}
def setUp(self):
self.c = Command(
'ndcli',
Command(
'create',
Command('pool', Argument('poolname'),
Group(Token('vlan'), Argument('vlan'), nargs='?'),
Argument('attrs', metavar='KEY:VALUE', nargs='*'))))
def test_plus():
c = Command('ndcli',
Group(Token('val'), Argument('val', action='append'), nargs='+'),
Group(Token('endval'), Argument('endval'), nargs='?'))
assert c.parse([]).errors
assert c.parse(['endval', 'a']).errors
assert c.parse(['val', 'a']).values == {'val': ['a'], 'endval': None}
assert c.parse(['val', 'a', 'val', 'b']).values == {'val': ['a', 'b'], 'endval': None}
assert c.parse(['val', 'a', 'val', 'b', 'endval', 'c']).values == {'val': ['a', 'b'], 'endval': 'c'}
def setUp(self):
self.c = Command(
'ndcli', Option('D', 'detailed', help='detailed return codes'),
Option('s', 'server', help='Dim server URL', action='store'),
Command(
'set',
Group(Argument('field',
choices=('master', 'mail', 'export_file'),
action='append_unique'),
Argument('value', action='append'),
nargs='*')))
def test_completion2():
def completions(line):
return c.complete(line, len(line))
c = Command('ndcli',
Group(Argument('field', choices=('f1', 'f2', 'f3'), action='append_unique'),
Argument('value', action='append'),
nargs='*'))
assert completions('ndcli ') == set(['f1', 'f2', 'f3'])
assert completions('ndcli f1 ') == set()
assert completions('ndcli f1 1 ') == set(['f2', 'f3'])
assert completions('ndcli f1 1 f') == set(['f2', 'f3'])
assert completions('ndcli f1 1 f2 2 ') == set(['f3'])
def test_plus_options():
c = Command('ndcli',
Option('q', 'quiet'),
Command('sub',
Argument('var', nargs='+', stop_at='end')))
assert c.parse(['sub', 'a']).values == {'var': ['a'], 'quiet': None}
assert c.parse(['sub', 'a', '-q']).values == {'var': ['a'], 'quiet': True}
def test_default_subcommand():
c = Command('ndcli',
Command('sub1', Argument('food'), Option('', 'drink'), defaults={'run': 'sub1'}),
Command('sub2', defaults={'run': 'sub2'}),
default_subcommand='sub1')
assert c.parse([]).values == {'run': 'sub1', 'food': None, 'drink': None}
assert c.parse(['sub1']).values.run == 'sub1'
assert c.parse(['sub2']).values.run == 'sub2'
def test_completion_quoted():
def completions(line):
return c.complete(line, len(line))
c = Command('ndcli', Group(Argument('field', choices=('a b', 'a!$\\'), action='append_unique')))
assert completions('ndcli ') == set(['a\ b', 'a\'!\'\$\\\\'])
assert completions("ndcli 'a") == set(["'a b'", "'a!$\\'"])
assert completions('ndcli "a') == set(['"a b"', '"a"\'!\'"\$\\\\"'])
assert completions('ndcli a\ ') == set(['a\\ b'])
def test_groups(self):
c = Command('ndcli', Group(Token('ttl'), Argument('ttl'), nargs='?'),
Command('test'))
assert c.parse(('ttl', '10', 'test')).values == {'ttl': '10'}
assert c.parse(('test', )).values == {'ttl': None}
assert c.parse(('bbl', '10', 'test')).errors
assert c.parse(('ttl', '10')).errors
def test_qual(self):
c = Command('ndcli',
Argument('1', choices=('a', 'b'), nargs='?'),
Argument('2'),
Argument('3', choices=('c', 'd'), nargs='*'),
Argument('4', choices=('e', 'f')),
Argument('5', nargs='*'))
assert c.parse(('xxxx', 'c', 'd', 'e', 'extra')).values == \
{'1': None,
'2': 'xxxx',
'3': ['c', 'd'],
'4': 'e',
'5': ['extra']}
assert c.parse(('x', 'f')).values == {'1': None,
'2': 'x',
'3': [],
'4': 'f',
'5': []}
def test_completion(self):
def completions(line):
return c.complete(line, len(line))
c = Command(
'ndcli', Option('D', 'detailed'), Option('q', 'quiet'),
Command(
'create',
Command('pool', Argument('poolname'), Option('p', 'perfect'),
Group(Token('vlan'), Argument('vlan'), nargs='?'),
Command('a'))))
assert completions('ndcli ') == set(['create'])
assert completions('ndcli -') == set(
['--detailed', '-D', '--quiet', '-q'])
assert completions('prog -- -') == set()
assert completions('prog create ') == set(['pool'])
assert completions('prog create pool ') == set()
assert completions('prog create pool -') == set(
['--detailed', '-D', '--quiet', '-q', '--perfect', '-p'])
assert completions('prog create pool test ') == set(['vlan', 'a'])
assert completions('prog create pool test vlan ') == set()
assert completions('prog create pool test vlan 1 ') == set(['a'])