def run_toolkit(self,
subcmd,
_args=None,
server=None,
verbose=False,
**kwargs):
def do_args(args):
a = (x[1] if (type(x) is tuple) and x[0] else x for x in args)
b = (x for x in a if x and type(x) is not tuple)
return list(b)
_args = _args if _args else []
args = do_args(['-v', (server, '-s'), (server, server), subcmd] +
_args)
print(kwargs)
if verbose:
self.out = sh.stackhut(args,
_out=lambda x: print(x.strip()),
**kwargs)
else:
self.out = sh.stackhut(args, **kwargs)
self.assertEqual(0, self.out.exit_code)
return self.out
def test_3_run(self):
# out = self.run_toolkit('run', verbose=True, _bg=True)
out = sh.stackhut('-v',
'run',
_out=lambda x: print(x.strip()),
_bg=True)
time.sleep(5)
# use the client lib to send some requests
sh_client = client.SHService('mands',
'test-service',
host='http://localhost:6000')
# valid request
res = sh_client.add(1, 2)
self.assertEqual(res, 3)
# invalid request
try:
res = sh_client.sub(1, 2)
except client.SHRPCError as e:
self.assertEqual(e.code, rpc.ERR_METHOD_NOT_FOUND)
self.assertIn('sub', e.message)
out.process.signal(2)
out.wait()
self.assertEqual(0, out.exit_code)
def run_toolkit(self, subcmd, _args=None, server=None, verbose=False, **kwargs):
def do_args(args):
a = (x[1] if (type(x) is tuple) and x[0] else x for x in args)
b = (x for x in a if x and type(x) is not tuple)
return list(b)
_args = _args if _args else []
args = do_args(['-v', (server, '-s'), (server, server), subcmd] + _args)
print(kwargs)
if verbose:
self.out = sh.stackhut(args, _out=lambda x: print(x.strip()), **kwargs)
else:
self.out = sh.stackhut(args, **kwargs)
self.assertEqual(0, self.out.exit_code)
return self.out
def test_3_run(self):
# out = self.run_toolkit('run', verbose=True, _bg=True)
out = sh.stackhut('-v', 'run', _out=lambda x:print(x.strip()), _bg=True)
time.sleep(5)
# use the client lib to send some requests
sh_client = client.SHService('mands', 'test-service', host='http://localhost:6000')
# valid request
res = sh_client.add(1,2)
self.assertEqual(res, 3)
# invalid request
try:
res = sh_client.sub(1,2)
except client.SHRPCError as e:
self.assertEqual(e.code, rpc.ERR_METHOD_NOT_FOUND)
self.assertIn('sub', e.message)
out.process.signal(2)
out.wait()
self.assertEqual(0, out.exit_code)
