def test_parse(self, gethostname):
p = NamespacedOptionParser([
'-c:jerry,elaine', '5',
'--loglevel:kramer=DEBUG',
'--flag',
'--logfile=foo', '-Q', 'bar', 'jerry',
'elaine', 'kramer',
'--', '.disable_rate_limits=1',
])
it = multi_args(p, cmd='COMMAND', append='*AP*',
prefix='*P*', suffix='*S*')
names = list(it)
def assert_line_in(name, args):
self.assertIn(name, [tup[0] for tup in names])
argv = None
for item in names:
if item[0] == name:
argv = item[1]
self.assertTrue(argv)
for arg in args:
self.assertIn(arg, argv)
assert_line_in(
'*P*jerry@*S*',
['COMMAND', '-n *P*jerry@*S*', '-Q bar',
'-c 5', '--flag', '--logfile=foo',
'-- .disable_rate_limits=1', '*AP*'],
)
assert_line_in(
'*P*elaine@*S*',
['COMMAND', '-n *P*elaine@*S*', '-Q bar',
'-c 5', '--flag', '--logfile=foo',
'-- .disable_rate_limits=1', '*AP*'],
)
assert_line_in(
'*P*kramer@*S*',
['COMMAND', '--loglevel=DEBUG', '-n *P*kramer@*S*',
'-Q bar', '--flag', '--logfile=foo',
'-- .disable_rate_limits=1', '*AP*'],
)
expand = names[0][2]
self.assertEqual(expand('%h'), '*P*jerry@*S*')
self.assertEqual(expand('%n'), 'jerry')
names2 = list(multi_args(p, cmd='COMMAND', append='',
prefix='*P*', suffix='*S*'))
self.assertEqual(names2[0][1][-1], '-- .disable_rate_limits=1')
gethostname.return_value = 'example.com'
p2 = NamespacedOptionParser(['10', '-c:1', '5'])
names3 = list(multi_args(p2, cmd='COMMAND'))
self.assertEqual(len(names3), 10)
self.assertEqual(
names3[0][0:2],
('*****@*****.**',
['COMMAND', '-n [email protected]', '-c 5', '']),
)
for i, worker in enumerate(names3[1:]):
self.assertEqual(
worker[0:2],
('*****@*****.**' % (i + 2),
['COMMAND', '-n celery%[email protected]' % (i + 2), '']),
)
names4 = list(multi_args(p2, cmd='COMMAND', suffix='""'))
self.assertEqual(len(names4), 10)
self.assertEqual(
names4[0][0:2],
('celery1@',
['COMMAND', '-n celery1@', '-c 5', '']),
)
p3 = NamespacedOptionParser(['foo@', '-c:foo', '5'])
names5 = list(multi_args(p3, cmd='COMMAND', suffix='""'))
self.assertEqual(
names5[0][0:2],
('foo@',
['COMMAND', '-n foo@', '-c 5', '']),
)
def test_parse(self, gethostname):
gethostname.return_value = 'example.com'
p = NamespacedOptionParser([
'-c:jerry,elaine', '5',
'--loglevel:kramer=DEBUG',
'--flag',
'--logfile=foo', '-Q', 'bar', 'jerry',
'elaine', 'kramer',
'--', '.disable_rate_limits=1',
])
it = multi_args(p, cmd='COMMAND', append='*AP*',
prefix='*P*', suffix='*S*')
names = list(it)
def assert_line_in(name, args):
self.assertIn(name, [tup[0] for tup in names])
argv = None
for item in names:
if item[0] == name:
argv = item[1]
self.assertTrue(argv)
for arg in args:
self.assertIn(arg, argv)
assert_line_in(
'*P*jerry@*S*',
['COMMAND', '-n *P*jerry@*S*', '-Q bar',
'-c 5', '--flag', '--logfile=foo',
'-- .disable_rate_limits=1', '*AP*'],
)
assert_line_in(
'*P*elaine@*S*',
['COMMAND', '-n *P*elaine@*S*', '-Q bar',
'-c 5', '--flag', '--logfile=foo',
'-- .disable_rate_limits=1', '*AP*'],
)
assert_line_in(
'*P*kramer@*S*',
['COMMAND', '--loglevel=DEBUG', '-n *P*kramer@*S*',
'-Q bar', '--flag', '--logfile=foo',
'-- .disable_rate_limits=1', '*AP*'],
)
expand = names[0][2]
self.assertEqual(expand('%h'), '*P*jerry@*S*')
self.assertEqual(expand('%n'), '*P*jerry')
names2 = list(multi_args(p, cmd='COMMAND', append='',
prefix='*P*', suffix='*S*'))
self.assertEqual(names2[0][1][-1], '-- .disable_rate_limits=1')
p2 = NamespacedOptionParser(['10', '-c:1', '5'])
names3 = list(multi_args(p2, cmd='COMMAND'))
self.assertEqual(len(names3), 10)
self.assertEqual(
names3[0][0:2],
('*****@*****.**',
['COMMAND', '-n [email protected]', '-c 5', '']),
)
for i, worker in enumerate(names3[1:]):
self.assertEqual(
worker[0:2],
('*****@*****.**' % (i + 2),
['COMMAND', '-n celery%[email protected]' % (i + 2), '']),
)
names4 = list(multi_args(p2, cmd='COMMAND', suffix='""'))
self.assertEqual(len(names4), 10)
self.assertEqual(
names4[0][0:2],
('celery1@',
['COMMAND', '-n celery1@', '-c 5', '']),
)
p3 = NamespacedOptionParser(['foo@', '-c:foo', '5'])
names5 = list(multi_args(p3, cmd='COMMAND', suffix='""'))
self.assertEqual(
names5[0][0:2],
('foo@',
['COMMAND', '-n foo@', '-c 5', '']),
)
