def test_get_last_output(self): host = '127.0.0.9' server = OpenSSHServer(listen_ip=host, port=self.port) server.start_server() try: hosts = [self.host, host] client = ParallelSSHClient(hosts, port=self.port, pkey=self.user_key) self.assertTrue(client.cmds is None) self.assertTrue(client.get_last_output() is None) client.run_command(self.cmd) self.assertTrue(client.cmds is not None) self.assertEqual(len(client.cmds), len(hosts)) expected_stdout = [self.resp] expected_stderr = [] output = client.get_last_output() self.assertIsInstance(output, list) self.assertEqual(len(output), len(hosts)) self.assertIsInstance(output[0], HostOutput) client.join(output) for i, host in enumerate(hosts): self.assertEqual(output[i].host, host) exit_code = output[i].exit_code _stdout = list(output[i].stdout) _stderr = list(output[i].stderr) self.assertEqual(exit_code, 0) self.assertListEqual(expected_stdout, _stdout) self.assertListEqual(expected_stderr, _stderr) finally: server.stop()
def test_get_last_output(self): host = '127.0.0.9' server = OpenSSHServer(listen_ip=host, port=self.port) server.start_server() try: hosts = [self.host, host] client = ParallelSSHClient(hosts, port=self.port, pkey=self.user_key) self.assertTrue(client.cmds is None) self.assertTrue(client.get_last_output() is None) client.run_command(self.cmd) self.assertTrue(client.cmds is not None) self.assertEqual(len(client.cmds), len(hosts)) output = client.get_last_output() self.assertTrue(len(output), len(hosts)) client.join(output) for i, host in enumerate(hosts): self.assertEqual(output[i].host, host) exit_code = output[i].exit_code self.assertEqual(exit_code, 0) finally: server.stop()