1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#!/usr/bin/env python
import os
import unittest
from pexpect import pxssh
class SSHTestBase(unittest.TestCase):
def setUp(self):
self.orig_path = os.environ.get('PATH')
fakessh_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), 'fakessh'))
os.environ['PATH'] = fakessh_dir + \
((os.pathsep + self.orig_path) if self.orig_path else '')
def tearDown(self):
if self.orig_path:
os.environ['PATH'] = self.orig_path
else:
del os.environ['PATH']
class PxsshTestCase(SSHTestBase):
def test_fake_ssh(self):
ssh = pxssh.pxssh()
#ssh.logfile_read = sys.stdout # DEBUG
ssh.login('server', 'me', password='s3cret')
ssh.sendline('ping')
ssh.expect('pong', timeout=10)
assert ssh.prompt(timeout=10)
ssh.logout()
def test_wrong_pw(self):
ssh = pxssh.pxssh()
try:
ssh.login('server', 'me', password='wr0ng')
except pxssh.ExceptionPxssh:
pass
else:
assert False, 'Password should have been refused'
def test_failed_set_unique_prompt(self):
ssh = pxssh.pxssh()
ssh.set_unique_prompt = lambda: False
try:
ssh.login('server', 'me', password='s3cret',
auto_prompt_reset=True)
except pxssh.ExceptionPxssh:
pass
else:
assert False, 'should have raised exception, pxssh.ExceptionPxssh'
def test_connection_refused(self):
ssh = pxssh.pxssh()
try:
ssh.login('noserver', 'me', password='s3cret')
except pxssh.ExceptionPxssh:
pass
else:
assert False, 'should have raised exception, pxssh.ExceptionPxssh'
def test_ssh_tunnel_string(self):
ssh = pxssh.pxssh(debug_tunnel_command=True)
tunnels = { 'local': ['2424:localhost:22'],'remote': ['2525:localhost:22'],
'dynamic': [8888] }
confirmation_string = 'ssh -q -R 2525:localhost:22 -L 2424:localhost:22 -D 8888 -l me server'
string = ssh.login('server', 'me', password='s3cret', ssh_tunnels=tunnels)
if string!=confirmation_string:
print(string)
assert False, 'String generated from tunneling is potientally incorrect.'
if __name__ == '__main__':
unittest.main()
|
