dorian
/
paramiko
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix module import in tests

This commit is contained in:
Jeff Forcier 2012-09-25 18:11:30 -07:00
parent 71456b8990
commit 2d7355da5e
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_util.py
View File

@ -172,15 +172,15 @@ Host *
Port 3333 Port 3333
""" """
f = cStringIO.StringIO(test_config_file) f = cStringIO.StringIO(test_config_file)
config = ssh.util.parse_ssh_config(f) config = paramiko.util.parse_ssh_config(f)
host = 'www13.example.com' host = 'www13.example.com'
self.assertEquals( self.assertEquals(
ssh.util.lookup_ssh_host_config(host, config), paramiko.util.lookup_ssh_host_config(host, config),
{'hostname': host, 'port': '22'} {'hostname': host, 'port': '22'}
) )
def test_8_eintr_retry(self): def test_8_eintr_retry(self):
self.assertEquals('foo', ssh.util.retry_on_signal(lambda: 'foo')) self.assertEquals('foo', paramiko.util.retry_on_signal(lambda: 'foo'))
# Variables that are set by raises_intr # Variables that are set by raises_intr
intr_errors_remaining = [3] intr_errors_remaining = [3]
@ -190,16 +190,16 @@ Host *
if intr_errors_remaining[0] > 0: if intr_errors_remaining[0] > 0:
intr_errors_remaining[0] -= 1 intr_errors_remaining[0] -= 1
raise IOError(errno.EINTR, 'file', 'interrupted system call') raise IOError(errno.EINTR, 'file', 'interrupted system call')
self.assertTrue(ssh.util.retry_on_signal(raises_intr) is None) self.assertTrue(paramiko.util.retry_on_signal(raises_intr) is None)
self.assertEquals(0, intr_errors_remaining[0]) self.assertEquals(0, intr_errors_remaining[0])
self.assertEquals(4, call_count[0]) self.assertEquals(4, call_count[0])
def raises_ioerror_not_eintr(): def raises_ioerror_not_eintr():
raise IOError(errno.ENOENT, 'file', 'file not found') raise IOError(errno.ENOENT, 'file', 'file not found')
self.assertRaises(IOError, self.assertRaises(IOError,
lambda: ssh.util.retry_on_signal(raises_ioerror_not_eintr)) lambda: paramiko.util.retry_on_signal(raises_ioerror_not_eintr))
def raises_other_exception(): def raises_other_exception():
raise AssertionError('foo') raise AssertionError('foo')
self.assertRaises(AssertionError, self.assertRaises(AssertionError,
lambda: ssh.util.retry_on_signal(raises_other_exception)) lambda: paramiko.util.retry_on_signal(raises_other_exception))