From 62f7e345f1fd2b4ca20177bd0fba624075fc1f51 Mon Sep 17 00:00:00 2001
From: Niranjan M.R <mrniranjan@redhat.com>
Date: Nov 28 2016 16:30:12 +0000
Subject: Add get_pty option to run command with pseudo terminal


Signed-off-by: Niranjan M.R <mrniranjan@redhat.com>

---

diff --git a/pytest_multihost/host.py b/pytest_multihost/host.py
index 826372d..a0fed74 100644
--- a/pytest_multihost/host.py
+++ b/pytest_multihost/host.py
@@ -201,7 +201,7 @@ class BaseHost(object):
 
     def run_command(self, argv, set_env=True, stdin_text=None,
                     log_stdout=True, raiseonerr=True,
-                    cwd=None):
+                    cwd=None, get_pty=False):
         """Run the given command on this host
 
         Returns a Command instance. The command will have already run in the
@@ -218,8 +218,9 @@ class BaseHost(object):
         :param raiseonerr: If true, an exception will be raised if the command
                            does not exit with return code 0
         :param cwd: The working directory for the command
+        :param get_pty: If True, request a pseudo-terminal from the server.
         """
-        command = self.transport.start_shell(argv, log_stdout=log_stdout)
+        command = self.transport.start_shell(argv, log_stdout=log_stdout, get_pty=get_pty)
         # Set working directory
         if cwd is None:
             cwd = self.test_dir
diff --git a/pytest_multihost/transport.py b/pytest_multihost/transport.py
index 8a36f02..5ccda05 100644
--- a/pytest_multihost/transport.py
+++ b/pytest_multihost/transport.py
@@ -248,13 +248,13 @@ class ParamikoTransport(Transport):
         self.log.info('MKDIR %s', path)
         self.sftp.mkdir(path)
 
-    def start_shell(self, argv, log_stdout=True):
+    def start_shell(self, argv, log_stdout=True, get_pty=False):
         logger_name = self.get_next_command_logger_name()
         ssh = self._transport.open_channel('session')
         self.log.info('RUN %s', argv)
         return SSHCommand(ssh, argv, logger_name=logger_name,
                           log_stdout=log_stdout,
-                          get_logger=self.host.config.get_logger)
+                          get_logger=self.host.config.get_logger, get_pty=get_pty)
 
     def get_file(self, remotepath, localpath):
         self.log.debug('GET %s', remotepath)
@@ -432,7 +432,7 @@ class SSHCallWrapper(object):
 class SSHCommand(Command):
     """Command implementation for ParamikoTransport and OpenSSHTranspport"""
     def __init__(self, ssh, argv, logger_name, log_stdout=True,
-                 collect_output=True, encoding='utf-8', get_logger=None):
+                 collect_output=True, encoding='utf-8', get_logger=None, get_pty=False):
         super(SSHCommand, self).__init__(argv, logger_name,
                                          log_stdout=log_stdout,
                                          get_logger=get_logger)
@@ -443,6 +443,9 @@ class SSHCommand(Command):
         self._ssh = ssh
 
         self.log.debug('RUN %s', argv)
+        
+        if get_pty:
+            self._ssh.get_pty()
 
         self._ssh.invoke_shell()
         def wrap_file(file, encoding):