def test_returns_proper_fqdn(self):
"""
Return the proper FQDN of the host when it's resolvable both
ways.
"""
self.mock_getaddrinfo.return_value = (
[0, 1, 2, 3, ['take_me', 'not_me']],)
self.mock_gethostbyaddr.return_value = ['proper_FQDN']
return_value = sanitize_host('erroneous_host')
self.mock_getaddrinfo.assert_called_once_with(
'erroneous_host', 443, socket.AF_INET, socket.SOCK_STREAM)
self.mock_gethostbyaddr.assert_called_once_with('take_me')
self.assertEqual(return_value, 'proper_FQDN')
def test_returns_proper_fqdn(self):
"""
Return the proper FQDN of the host when it's resolvable both
ways.
"""
self.mock_getaddrinfo.return_value = ([
0, 1, 2, 3, ['take_me', 'not_me']
], )
self.mock_gethostbyaddr.return_value = ['proper_FQDN']
return_value = sanitize_host('erroneous_host')
self.mock_getaddrinfo.assert_called_once_with('erroneous_host', 443,
socket.AF_INET,
socket.SOCK_STREAM)
self.mock_gethostbyaddr.assert_called_once_with('take_me')
self.assertEqual(return_value, 'proper_FQDN')
def test_raises_error_on_hostnotfound(self):
"""
Raise an error when the host is not resolvable.
"""
self.mock_getaddrinfo.side_effect = socket.gaierror('error_message')
return_value = None
with self.assertRaises(CMDLineExit) as cm:
return_value = sanitize_host('erroneous_host')
self.assertIsNone(return_value)
self.assertEqual(cm.exception.args[0],
'Could not resolve hostname \'afp\': error_message')
self.mock_getaddrinfo.assert_called_once_with('erroneous_host', 443,
socket.AF_INET,
socket.SOCK_STREAM)
self.mock_gethostbyaddr.assert_not_called()
def test_raises_error_on_hostnotfound(self):
"""
Raise an error when the host is not resolvable.
"""
self.mock_getaddrinfo.side_effect = socket.gaierror('error_message')
return_value = None
with self.assertRaises(CMDLineExit) as cm:
return_value = sanitize_host('erroneous_host')
self.assertIsNone(return_value)
self.assertEqual(
cm.exception.args[0],
'Could not resolve hostname \'erroneous_host\': error_message')
self.mock_getaddrinfo.assert_called_once_with(
'erroneous_host', 443, socket.AF_INET, socket.SOCK_STREAM)
self.mock_gethostbyaddr.assert_not_called()
def test_raises_error_on_reversenotfound(self):
"""
Raise an error when the host is resolvable, but its reverse is
not.
"""
self.mock_getaddrinfo.return_value = (
[0, 1, 2, 3, ['take_me', 'not_me']],)
self.mock_gethostbyaddr.side_effect = socket.gaierror('error_message')
return_value = None
with self.assertRaises(CMDLineExit) as cm:
return_value = sanitize_host('erroneous_host')
self.assertIsNone(return_value)
self.assertEqual(
cm.exception.args[0],
'DNS reverse lookup failed for IP take_me: error_message')
self.mock_getaddrinfo.assert_called_once_with(
'erroneous_host', 443, socket.AF_INET, socket.SOCK_STREAM)
self.mock_gethostbyaddr.assert_called_once_with('take_me')
def test_raises_error_on_reversenotfound(self):
"""
Raise an error when the host is resolvable, but its reverse is
not.
"""
self.mock_getaddrinfo.return_value = ([
0, 1, 2, 3, ['take_me', 'not_me']
], )
self.mock_gethostbyaddr.side_effect = socket.gaierror('error_message')
return_value = None
with self.assertRaises(CMDLineExit) as cm:
return_value = sanitize_host('erroneous_host')
self.assertIsNone(return_value)
self.assertEqual(
cm.exception.args[0],
'DNS reverse lookup failed for IP take_me: error_message')
self.mock_getaddrinfo.assert_called_once_with('erroneous_host', 443,
socket.AF_INET,
socket.SOCK_STREAM)
self.mock_gethostbyaddr.assert_called_once_with('take_me')