class SshTestCase(unittest.TestCase):
# Always run before any test
def setUp(self):
self.ssh = Ssh(ip='127.0.0.1',
user='******',
password='',
debug=True)
# Always run after any test
def tearDown(self):
pass
# Reminder : test name must start with test ...
def test_connect_and_close(self):
self.ssh.connect()
self.assertTrue(self.ssh.connected)
self.ssh.close()
self.assertFalse(self.ssh.connected)
def test_connect_key_authentication_failure(self):
self.ssh.private_key_file = "whateverfile"
self.ssh.mock(exception=paramiko.AuthenticationException)
self.ssh.connect()
self.ssh.close()
self.assertRaises(paramiko.AuthenticationException)
def test_ssh_connection_failure(self):
self.ssh.mock(exception=paramiko.SSHException)
self.ssh.connect()
self.ssh.close()
self.assertRaises(paramiko.SSHException)
def test_socket_timeout(self):
self.ssh.mock(exception=socket.timeout)
self.ssh.connect()
self.ssh.close()
self.assertRaises(socket.timeout)
def test_genuine_exception(self):
self.ssh.mock(exception=Exception)
self.ssh.connect()
self.ssh.close()
self.assertRaises(Exception)
def test_sshcmd_single_command(self):
self.ssh.mock(context='default')
self.ssh.connect()
self.ssh.commands(["uptime"])
# Check we can see the work average in the output : 12:09:34 up 2:49, 1 user, load average: 0.01, 0.05, 0.0
self.ssh.close()
self.assertNotEqual(self.ssh.output.find("load average"), -1)
def test_sshcmd_default_command(self):
self.ssh.connect()
self.ssh.commands(["whatever"])
self.ssh.close()
self.assertTrue(True)
#@unittest.skip("by-passed for now")
def test_sshcmd_double_command(self):
self.ssh.mock(context='default')
self.ssh.connect()
self.ssh.commands(["uptime", "ps -ef"])
self.ssh.close()
self.assertNotEqual(self.ssh.output.find("load average"), -1)
def test_sshcmd_commands_timeout(self):
self.ssh.mock(exception=socket.timeout)
self.ssh.connect()
self.ssh.commands(["whatever"])
self.ssh.close()
self.assertRaises(socket.timeout)
def test_sshcmd_commands_failure(self):
self.ssh.mock(exception=paramiko.SSHException)
self.ssh.connect()
self.ssh.commands(["whatever"])
self.ssh.close()
self.assertRaises(paramiko.SSHException)
def test_sshcmd_execute(self):
self.ssh.mock(context='default')
self.ssh.connect()
self.ssh.execute(["uptime", "ps -ef"])
self.ssh.close()
self.assertNotEqual(self.ssh.output.find("load average"), -1)
def test_shell_send_single(self):
self.ssh.mock(context='default')
self.ssh.connect()
self.ssh.commands(["uptime"])
self.ssh.close()
self.assertNotEqual(self.ssh.output.find("load average"), -1)
def test_ssh_ouputfile(self):
self.ssh.trace_open(filename="myTraceFile.log")
self.ssh.trace_write("\n*** This is test mark line 1 ***\n")
self.ssh.trace_write("\n*** This is test mark line 2 ***\n")
self.ssh.trace_mark("MARK1 - command sample")
self.ssh.mock(context='default')
self.ssh.connect()
self.ssh.commands(["uptime"])
self.ssh.trace_mark("MARK2 - execute sample")
self.ssh.mock(context='ps')
self.ssh.execute(["ps -ef"])
self.ssh.shell_read()
self.ssh.close()
from ssh import Ssh
import time
ssh = Ssh(ip='127.0.0.1',
user='******',
password='',
private_key_file='/home/cgustave/.ssh/id_rsa',
port=22,
debug=True)
ssh.trace_open(filename="myTraceFileChannel.log")
#ssh.connect()
#ssh.invoke_channel()
ssh.channel_send('ls -la\n')
data = ssh.channel_read()
print("received {}".format(data))
ssh.close()
from ssh import Ssh myssh = Ssh(ip='127.0.0.1', user='******', password='', debug=True) myssh.trace_open(filename="test4File.log") myssh.mock(context='default') myssh.connect() myssh.execute(["ps -ef"]) myssh.shell_read() myssh.close()
