def test_get_hostname_raises_ioerror_when_getfqdn_raises_ioerror(self, mock_hostname):
""" Tests invocation of API get_hostname raises IOError when getfqdn raises IOError"""
# arrange
mock_hostname.side_effect = IOError('getfqdn IO error')
# act, assert
with self.assertRaises(IOError):
Utils.get_hostname()
def test_get_hostname_valid(self, mock_hostname):
""" Test a valid invocation of API get_hostname """
# arrange
hostname = 'test_hostname'
mock_hostname.return_value = hostname
# act
result = Utils.get_hostname()
# assert
mock_hostname.assert_called_with()
self.assertEqual(hostname, result)