def test_get_other_addresses_returns_empty_list(self, mocked): servers = ['mocked-1.example.com'] addresses = ['1.1.1.1'] mocked.return_value = 'mocked-1.example.com' result = util.get_other_addresses(servers, addresses) self.assertEquals([], result)
def test_get_other_addresses(self, mocked): addresses = ['1.1.1.1', '2.1.1.1', '3.1.1.1', '4.1.1.1', '5.1.1.1', '6.1.1.1', '7.1.1.1', '8.1.1.1'] mocked.return_value = 'mocked-3.example.com' result = util.get_other_addresses(self._servers, addresses) expected = ['1.1.1.1', '3.1.1.1', '2.1.1.1', '4.1.1.1', '5.1.1.1', '7.1.1.1'] self.assertItemsEqual(expected, result)
def _cleanup(self): # TODO(retr0h): log print 'addresses to cleanup' other_addresses = util.get_other_addresses(self._get_servers(), self._config.aliases) for address in other_addresses: print address interface = self._config.interface util.delete_alias(address, interface)