def test_ssh_client_shell(self):
"""Test that running command sans shell works as expected
and that shell commands fail accordingly"""
client = SSHClient(self.host,
port=self.listen_port,
pkey=self.user_key)
channel, host, stdout, stderr, stdin = client.exec_command(
self.fake_cmd, use_shell=False)
output = list(client.read_output_buffer(stdout))
stderr = list(stderr)
expected = []
exit_code = channel.recv_exit_status()
self.assertEqual(expected,
output,
msg="Got unexpected command output - %s" % (output, ))
self.assertTrue(
exit_code == 127,
msg="Expected cmd not found error code 127, got %s instead" %
(exit_code, ))
channel, host, stdout, stderr, stdin = client.exec_command(
'id', use_shell=False)
output = list(client.read_output_buffer(stdout))
exit_code = channel.recv_exit_status()
self.assertTrue(output,
msg="Got no output from cmd executed without shell")
self.assertTrue(
exit_code == 0,
msg="Cmd executed with shell failed with error code %s" %
(exit_code, ))
del client
def test_ssh_client_utf_encoding(self):
"""Test that unicode output works"""
client = SSHClient(self.host,
port=self.listen_port,
pkey=self.user_key)
expected = [u'é']
cmd = u"echo 'é'"
channel, host, stdout, stderr, stdin = client.exec_command(cmd)
output = list(client.read_output_buffer(stdout))
self.assertEqual(expected,
output,
msg="Got unexpected unicode output %s - expected %s" %
(
output,
expected,
))
del client
def test_ssh_agent_authentication(self):
"""Test authentication via SSH agent.
Do not provide public key to use when creating SSHClient,
instead override the client's agent with our own fake SSH agent,
add our to key to agent and try to login to server.
Key should be automatically picked up from the overriden agent"""
agent = SSHAgent()
agent.add_key(USER_KEY)
client = SSHClient(self.host, port=self.listen_port, agent=agent)
channel, host, stdout, stderr, stdin = client.exec_command(
self.fake_cmd)
output = list(client.read_output_buffer(stdout))
stderr = list(client.read_output_buffer(stderr))
expected = [self.fake_resp]
self.assertEqual(expected,
output,
msg="Got unexpected command output - %s" % (output, ))
del client
agent._connect(None)
agent._close()
del agent
