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)