def _ssh(self):
from dvc.remote.ssh.connection import SSHConnection
conn = SSHConnection(
host=self.host,
port=self.port,
username=TEST_SSH_USER,
key_filename=TEST_SSH_KEY_PATH,
)
try:
yield conn
finally:
conn.close()
def ssh(ssh_server):
yield SSHConnection(
ssh_server.host,
username=user,
port=ssh_server.port,
key_filename=key_path,
)
def ssh():
users = {user: key_path}
with mockssh.Server(users) as s:
yield SSHConnection(s.host,
username=user,
port=s.port,
key_filename=key_path)
def ssh(self, path_info):
host, user, port = path_info.host, path_info.user, path_info.port
logger.debug(
"Establishing ssh connection with '{host}' "
"through port '{port}' as user '{user}'".format(
host=host, user=user, port=port
)
)
if self.ask_password and not self.password:
self.password = prompt.password(
"Enter a private key passphrase or a password for "
"host '{host}' port '{port}' user '{user}'".format(
host=host, port=port, user=user
)
)
return SSHConnection(
host,
username=user,
port=port,
key_filename=self.keyfile,
timeout=self.timeout,
password=self.password,
)
def ssh_connection(ssh_server):
from dvc.remote.ssh.connection import SSHConnection
yield SSHConnection(
host=ssh_server.host,
port=ssh_server.port,
username=TEST_SSH_USER,
key_filename=TEST_SSH_KEY_PATH,
)
def ssh(ssh_server):
yield SSHConnection(**ssh_server.test_creds)
def _client(server):
client = SSHConnection(server.host,
username=user,
port=server.port,
key_filename=key_path)
return client
