mirror of
https://salsa.debian.org/freeipa-team/freeipa.git
synced 2026-10-25 22:50:12 -05:00
Sergey Orlov
7363c4b203
The pexpect module can be used for controlling and testing interactive command-line programs. The wrapper adds testing-oriented features like logging and automatic process termination and default check for process exit status. Reviewed-By: Alexander Bokovoy <abokovoy@redhat.com>
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
#
|
|
# Copyright (C) 2020 FreeIPA Contributors see COPYING for license
|
|
#
|
|
"""Enhanced SSH transport for pytest multihost
|
|
|
|
Provides SSH password login for OpenSSH transport
|
|
"""
|
|
import os
|
|
|
|
from .expect import IpaTestExpect
|
|
|
|
from pytest_multihost.transport import OpenSSHTransport
|
|
|
|
|
|
class IPAOpenSSHTransport(OpenSSHTransport):
|
|
def _get_ssh_argv(self):
|
|
"""Return the path to SSH and options needed for every call"""
|
|
control_file = os.path.join(self.control_dir.path, "control")
|
|
known_hosts_file = os.path.join(self.control_dir.path, "known_hosts")
|
|
|
|
argv = [
|
|
"ssh",
|
|
"-l",
|
|
self.host.ssh_username,
|
|
"-o",
|
|
"ControlPath=%s" % control_file,
|
|
"-o",
|
|
"StrictHostKeyChecking=no",
|
|
"-o",
|
|
"UserKnownHostsFile=%s" % known_hosts_file,
|
|
]
|
|
|
|
if self.host.ssh_key_filename:
|
|
key_filename = os.path.expanduser(self.host.ssh_key_filename)
|
|
argv.extend(["-i", key_filename])
|
|
elif self.host.ssh_password:
|
|
password_file = os.path.join(self.control_dir.path, "password")
|
|
with open(password_file, "w") as f:
|
|
os.fchmod(f.fileno(), 0o600)
|
|
f.write(self.host.ssh_password)
|
|
f.write("\n")
|
|
argv = ["sshpass", f"-f{password_file}"] + argv
|
|
else:
|
|
self.log.critical("No SSH credentials configured")
|
|
raise RuntimeError("No SSH credentials configured")
|
|
|
|
argv.append(self.host.external_hostname)
|
|
self.log.debug("SSH invocation: %s", argv)
|
|
|
|
return argv
|
|
|
|
def spawn_expect(self, argv, default_timeout, encoding):
|
|
self.log.debug('Starting pexpect ssh session')
|
|
if isinstance(argv, str):
|
|
argv = [argv]
|
|
argv = self._get_ssh_argv() + ['-t', '-q'] + argv
|
|
return IpaTestExpect(argv, default_timeout, encoding)
|