def test_execute_timeout(self, ssh_client):
ssh = SSHRetry("host", "user", "passwd", read_file_as_string("../files/privatekey.pem"))
client = MagicMock()
ssh_client.return_value = client
client.exec_command.return_value = "", ["out"], ["err"]
res_stdout, res_stderr = ssh.execute_timeout("ls", 5)
self.assertEqual(res_stdout, "out")
self.assertEqual(res_stderr, "err")
def test_execute_timeout(self, ssh_client):
ssh = SSHRetry("host", "user", "passwd",
read_file_as_string("../files/privatekey.pem"))
client = MagicMock()
ssh_client.return_value = client
client.exec_command.return_value = "", ["out"], ["err"]
res_stdout, res_stderr = ssh.execute_timeout("ls", 5)
self.assertEqual(res_stdout, "out")
self.assertEqual(res_stderr, "err")
def test_execute_timeout(self, ssh_client):
ssh = SSHRetry("host", "user", "passwd",
read_file_as_string("../files/privatekey.pem"))
client = MagicMock()
ssh_client.return_value = client
tansport = MagicMock()
client.get_transport.return_value = tansport
channel = MagicMock()
tansport.open_session.return_value = channel
channel.makefile.return_value = "out"
channel.makefile_stderr.return_value = "err"
channel.recv_exit_status.return_value = 0
res_stdout, res_stderr, code = ssh.execute_timeout("ls", 5)
self.assertEqual(res_stdout, "out")
self.assertEqual(res_stderr, "err")
self.assertEqual(code, 0)
