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)
