def test_expected(self):
values = [self.GOOD_LIST_OUTPUT, self.GOOD_KILL_OUTPUT]
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = (values.pop(0), "")
return m
with patch.object(ssh, "Popen", side_effect=fake_popen) as m:
ssh.ssh_terminate_single_job(["ssh_bin"], "address", "key.pem")
self.assertEqual(m.call_args_list[0], call(self.EXPECTED_LIST_CALL, stdin=PIPE, stdout=PIPE, stderr=PIPE))
self.assertEqual(m.call_args_list[1], call(self.EXPECTED_KILL_CALL, stdin=PIPE, stdout=PIPE, stderr=PIPE))
def test_dance_floor_is_empty(self):
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = ("0 jobs currently running\n", "")
return m
with patch.object(ssh, "Popen", side_effect=fake_popen):
self.assertEqual(None, ssh.ssh_terminate_single_job(["ssh_bin"], "address", "key.pem"))
def test_dance_floor_is_empty(self):
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = (b"0 jobs currently running\n", b'')
return m
with patch.object(ssh, 'Popen', side_effect=fake_popen):
self.assertEqual(
None, ssh.ssh_terminate_single_job(
['ssh_bin'], 'address', 'key.pem'))
def test_dance_floor_is_empty(self):
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = (b"0 jobs currently running\n", b'')
return m
with patch.object(ssh, 'Popen', side_effect=fake_popen):
self.assertEqual(
None,
ssh.ssh_terminate_single_job(['ssh_bin'], 'address',
'key.pem'))
def test_junk_kill_output(self):
values = [self.GOOD_LIST_OUTPUT, "yah output, its gahbage\n"]
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = (values.pop(0), "")
return m
with patch.object(ssh, "Popen", side_effect=fake_popen):
self.assertEqual(
ssh.ssh_terminate_single_job(["ssh_bin"], "address", "key.pem"), "yah output, its gahbage\n"
)
def test_junk_kill_output(self):
values = [self.GOOD_LIST_OUTPUT, "yah output, its gahbage\n"]
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = (values.pop(0), '')
return m
with patch.object(ssh, 'Popen', side_effect=fake_popen):
self.assertEqual(
ssh.ssh_terminate_single_job(
['ssh_bin'], 'address', 'key.pem'),
'yah output, its gahbage\n')
def test_junk_kill_output(self):
values = [self.GOOD_LIST_OUTPUT, b"yah output, its gahbage\n"]
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = (values.pop(0), b'')
return m
with patch.object(ssh, 'Popen', side_effect=fake_popen):
self.assertEqual(
ssh.ssh_terminate_single_job(
['ssh_bin'], 'address', 'key.pem'),
'yah output, its gahbage\n')
def test_expected(self):
values = [self.GOOD_LIST_OUTPUT, self.GOOD_KILL_OUTPUT]
def fake_popen(*args, **kwargs):
m = Mock()
m.communicate.return_value = (values.pop(0), b'')
return m
with patch.object(ssh, 'Popen', side_effect=fake_popen) as m:
ssh.ssh_terminate_single_job(['ssh_bin'], 'address', 'key.pem')
self.assertEqual(
m.call_args_list[0],
call(self.EXPECTED_LIST_CALL,
stdin=PIPE,
stdout=PIPE,
stderr=PIPE))
self.assertEqual(
m.call_args_list[1],
call(self.EXPECTED_KILL_CALL,
stdin=PIPE,
stdout=PIPE,
stderr=PIPE))
