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))