def test_updateresult(self):
r1 = Result("result1")
r1.return_code = 0
r2 = Result("result2")
r2.return_code = 127
r3 = Result("result3")
r3.return_code = 1
values = [(r1, 'no', 'ls'), (r2, 'yes', '@@ sudo reboot'), (r3, 'dontcare', '## ping foo')]
for res, negative, command in values:
tunirlib.update_result(res, command, negative)
res = [True, True, False]
for out, result in zip(tunirlib.STR.iteritems(), res):
self.assertEqual(out[1]['status'], result)
def test_execute_nongating(self, t_run):
config = {"host_string": "127.0.0.1",
"user": "******"}
r1 = Result("result1")
r1.return_code = 127
t_run.return_value=r1
res, negative = tunirlib.execute(config, '## ls')
self.assertEqual(res, "result1")
self.assertEqual(negative, "dontcare")
def test_main(self,p_br, p_run,p_kill, p_exit, p_sleep):
res = StupidProcess()
p_br.return_value = res
r1 = Result("result1")
r1.return_code = 0
r2 = Result("result2")
r2.return_code = 0
values = [r1, r2]
p_run.side_effect = values
# Now let us construct the args
args = StupidArgs()
with captured_output() as (out, err):
main(args)
self.assertIn("Passed:1", out.getvalue())
self.assertIn("Job status: True", out.getvalue())
self.assertTrue(p_kill.called)
self.assertTrue(p_exit.called)