class HostTest(unittest.TestCase):
CORRELATE = Host
def setUp(self):
self.host = Host('localhost')
self.host.set_all(dict(testarg=1))
def testConstructor(self):
host = Host('localhost')
self.assertEqual(host.get_protocol(), 'telnet')
host = Host('localhost', default_protocol='foo')
self.assertEqual(host.get_protocol(), 'foo')
for url, result in urls:
host = Host(url)
uri = Url.from_string(url)
self.assertEqual(host.get_name(), uri.hostname)
self.assertEqual(host.get_address(), uri.hostname)
self.assertEqual(host.get_uri().split('&').sort(),
str(uri).split('&').sort())
def testSetUri(self):
for url, result in urls:
self.host.set_uri(url)
uri = Url.from_string(url)
self.assertEqual(self.host.get_name(), uri.hostname)
self.assertEqual(self.host.get_address(), uri.hostname)
def testGetUri(self):
for url, result in urls:
host = Host(url)
uri = Url.from_string(url)
self.assertEqual(host.get_uri().split('&').sort(),
str(uri).split('&').sort())
def testGetDict(self):
host = Host('foo')
host.set_address('foo2')
self.assertEqual(host.get_dict(), {
'hostname': 'foo',
'address': 'foo2',
'protocol': 'telnet',
'port': 23
})
def testSetAddress(self):
self.host.set_protocol('dummy')
self.host.set_address('test.org')
self.assertEqual(self.host.get_protocol(), 'dummy')
self.assertEqual(self.host.get_name(), 'localhost')
self.assertEqual(self.host.get_address(), 'test.org')
self.host.set_address('001.002.003.004')
self.assertEqual(self.host.get_protocol(), 'dummy')
self.assertEqual(self.host.get_name(), 'localhost')
self.assertEqual(self.host.get_address(), '1.2.3.4')
def testGetAddress(self):
self.assertEqual(self.host.get_address(), 'localhost')
# Additional tests are in testSetAddress().
def testSetName(self):
self.assertEqual(self.host.get_name(), 'localhost')
self.host.set_protocol('dummy')
self.host.set_name('test.org')
self.assertEqual(self.host.get_protocol(), 'dummy')
self.assertEqual(self.host.get_name(), 'test.org')
self.assertEqual(self.host.get_address(), 'localhost')
self.host.set_name('testhost')
self.assertEqual(self.host.get_name(), 'testhost')
def testGetName(self):
pass # Tested in testSetName().
def testSetProtocol(self):
self.assertEqual(self.host.get_protocol(), 'telnet')
self.host.set_protocol('dummy')
self.assertEqual(self.host.get_protocol(), 'dummy')
def testGetProtocol(self):
pass # Tested in testSetProtocol().
def testSetOption(self):
self.assertRaises(TypeError, self.host.set_option, 'test', True)
self.assertEqual(self.host.get_options(), {})
self.assertEqual(self.host.get_option('verify_fingerprint'), None)
self.assertEqual(self.host.get_option('verify_fingerprint', False),
False)
self.host.set_option('verify_fingerprint', True)
self.assertEqual(self.host.get_option('verify_fingerprint'), True)
self.assertEqual(self.host.get_options(), {'verify_fingerprint': True})
def testGetOption(self):
pass # Tested in testSetOption().
def testGetOptions(self):
pass # Tested in testSetOption().
def testSetTcpPort(self):
self.assertEqual(self.host.get_tcp_port(), 23)
self.host.set_protocol('ssh')
self.assertEqual(self.host.get_tcp_port(), 23)
self.host.set_tcp_port(123)
self.assertEqual(self.host.get_tcp_port(), 123)
def testGetTcpPort(self):
pass # Tested in testSetTcpPort().
def testSetAccount(self):
account = Account('test')
self.assertEqual(self.host.get_account(), None)
self.host.set_account(account)
self.assertEqual(self.host.get_account(), account)
def testGetAccount(self):
pass # Tested in testSetAccount().
def testSet(self):
self.assertEqual(self.host.get('test'), None)
self.host.set('test', 3)
self.assertEqual(self.host.get('test'), 3)
def testSetAll(self):
self.testSet()
self.host.set_all({'test1': 1, 'test2': 2})
self.assertEqual(self.host.get('test'), None)
self.assertEqual(self.host.get('test1'), 1)
self.assertEqual(self.host.get('test2'), 2)
def testGetAll(self):
self.assertEqual(self.host.get_all(), {'testarg': 1})
self.testSetAll()
self.assertEqual(self.host.get_all(), {'test1': 1, 'test2': 2})
host = Host('localhost')
self.assertEqual(host.get_all(), {})
def testAppend(self):
self.assertEqual(self.host.get('test'), None)
self.host.append('test', 3)
self.assertEqual(self.host.get('test'), [3])
self.host.append('test', 4)
self.assertEqual(self.host.get('test'), [3, 4])
def testSetDefault(self):
self.testSet()
self.assertEqual(self.host.get('test'), 3)
self.assertEqual(self.host.get('test2'), None)
self.host.set_default('test', 5)
self.host.set_default('test2', 1)
self.assertEqual(self.host.get('test'), 3)
self.assertEqual(self.host.get('test2'), 1)
def testHasKey(self):
self.testSet()
self.assertTrue(self.host.has_key('test'))
self.assertFalse(self.host.has_key('test2'))
def testGet(self):
self.testSet()
self.assertEqual(self.host.get('test'), 3)
self.assertEqual(self.host.get('test2'), None)
self.assertEqual(self.host.get('test', 1), 3)
self.assertEqual(self.host.get('test2', 1), 1)
class HostTest(unittest.TestCase):
CORRELATE = Host
def setUp(self):
self.host = Host('localhost')
self.host.set_all(dict(testarg = 1))
def testConstructor(self):
host = Host('localhost')
self.assertEqual(host.get_protocol(), 'telnet')
host = Host('localhost', default_protocol = 'foo')
self.assertEqual(host.get_protocol(), 'foo')
for url, result in urls:
host = Host(url)
uri = Url.from_string(url)
self.assertEqual(host.get_name(), uri.hostname)
self.assertEqual(host.get_address(), uri.hostname)
self.assertEqual(host.get_uri(), str(uri))
def testSetUri(self):
for url, result in urls:
self.host.set_uri(url)
uri = Url.from_string(url)
self.assertEqual(self.host.get_name(), uri.hostname)
self.assertEqual(self.host.get_address(), uri.hostname)
def testGetUri(self):
for url, result in urls:
host = Host(url)
uri = Url.from_string(url)
self.assertEqual(host.get_uri(), str(uri))
def testGetDict(self):
host = Host('foo')
host.set_address('foo2')
self.assertEqual(host.get_dict(), {'hostname': 'foo',
'address': 'foo2',
'protocol': 'telnet',
'port': 23})
def testSetAddress(self):
self.host.set_protocol('dummy')
self.host.set_address('test.org')
self.assertEqual(self.host.get_protocol(), 'dummy')
self.assertEqual(self.host.get_name(), 'localhost')
self.assertEqual(self.host.get_address(), 'test.org')
self.host.set_address('001.002.003.004')
self.assertEqual(self.host.get_protocol(), 'dummy')
self.assertEqual(self.host.get_name(), 'localhost')
self.assertEqual(self.host.get_address(), '1.2.3.4')
def testGetAddress(self):
self.assertEqual(self.host.get_address(), 'localhost')
# Additional tests are in testSetAddress().
def testSetName(self):
self.assertEqual(self.host.get_name(), 'localhost')
self.host.set_protocol('dummy')
self.host.set_name('test.org')
self.assertEqual(self.host.get_protocol(), 'dummy')
self.assertEqual(self.host.get_name(), 'test.org')
self.assertEqual(self.host.get_address(), 'localhost')
self.host.set_name('testhost')
self.assertEqual(self.host.get_name(), 'testhost')
def testGetName(self):
pass # Tested in testSetName().
def testSetProtocol(self):
self.assertEqual(self.host.get_protocol(), 'telnet')
self.host.set_protocol('dummy')
self.assertEqual(self.host.get_protocol(), 'dummy')
def testGetProtocol(self):
pass # Tested in testSetProtocol().
def testSetOption(self):
self.assertRaises(TypeError, self.host.set_option, 'test', True)
self.assertEqual(self.host.get_options(), {})
self.assertEqual(self.host.get_option('verify_fingerprint'), None)
self.assertEqual(self.host.get_option('verify_fingerprint', False), False)
self.host.set_option('verify_fingerprint', True)
self.assertEqual(self.host.get_option('verify_fingerprint'), True)
self.assertEqual(self.host.get_options(), {'verify_fingerprint': True})
def testGetOption(self):
pass # Tested in testSetOption().
def testGetOptions(self):
pass # Tested in testSetOption().
def testSetTcpPort(self):
self.assertEqual(self.host.get_tcp_port(), 23)
self.host.set_protocol('ssh')
self.assertEqual(self.host.get_tcp_port(), 23)
self.host.set_tcp_port(123)
self.assertEqual(self.host.get_tcp_port(), 123)
def testGetTcpPort(self):
pass # Tested in testSetTcpPort().
def testSetAccount(self):
account = Account('test')
self.assertEqual(self.host.get_account(), None)
self.host.set_account(account)
self.assertEqual(self.host.get_account(), account)
def testGetAccount(self):
pass # Tested in testSetAccount().
def testSet(self):
self.assertEqual(self.host.get('test'), None)
self.host.set('test', 3)
self.assertEqual(self.host.get('test'), 3)
def testSetAll(self):
self.testSet()
self.host.set_all({'test1': 1, 'test2': 2})
self.assertEqual(self.host.get('test'), None)
self.assertEqual(self.host.get('test1'), 1)
self.assertEqual(self.host.get('test2'), 2)
def testGetAll(self):
self.assertEqual(self.host.get_all(), {'testarg': 1})
self.testSetAll()
self.assertEqual(self.host.get_all(), {'test1': 1, 'test2': 2})
host = Host('localhost')
self.assertEqual(host.get_all(), {})
def testAppend(self):
self.assertEqual(self.host.get('test'), None)
self.host.append('test', 3)
self.assertEqual(self.host.get('test'), [3])
self.host.append('test', 4)
self.assertEqual(self.host.get('test'), [3, 4])
def testSetDefault(self):
self.testSet()
self.assertEqual(self.host.get('test'), 3)
self.assertEqual(self.host.get('test2'), None)
self.host.set_default('test', 5)
self.host.set_default('test2', 1)
self.assertEqual(self.host.get('test'), 3)
self.assertEqual(self.host.get('test2'), 1)
def testHasKey(self):
self.testSet()
self.assert_(self.host.has_key('test'))
self.failIf(self.host.has_key('test2'))
def testGet(self):
self.testSet()
self.assertEqual(self.host.get('test'), 3)
self.assertEqual(self.host.get('test2'), None)
self.assertEqual(self.host.get('test', 1), 3)
self.assertEqual(self.host.get('test2', 1), 1)
