def test_caches_region_model_object(self):
hostname = factory.make_hostname()
self.set_fake_twisted_dns_reply([hostname])
service = ReverseDNSService()
yield service.startService()
self.assertThat(service.region, Equals(self.region))
service.stopService()
def test__registers_and_unregisters_listener(self):
listener = Mock()
listener.register = Mock()
listener.unregister = Mock()
service = ReverseDNSService(postgresListener=listener)
yield service.startService()
self.assertThat(
listener.register,
MockCalledOnceWith('neighbour', service.consumeNeighbourEvent))
service.stopService()
self.assertThat(
listener.unregister,
MockCalledOnceWith('neighbour', service.consumeNeighbourEvent))
def test_ignores_timeouts_when_consuming_neighbour_event(self):
reverseResolve = self.patch(reverse_dns_module, "reverseResolve")
reverseResolve.return_value = defer.fail(defer.TimeoutError())
ip = factory.make_ip_address(ipv6=False)
service = ReverseDNSService()
yield callWithServiceRunning(service, service.consumeNeighbourEvent,
"create", "%s/32" % ip)
self.assertThat(reverseResolve, MockCalledOnceWith(ip))
result = yield deferToDatabase(RDNS.objects.first)
self.assertThat(result, Is(None))
def test_updates_rdns_entry(self):
hostname = factory.make_hostname()
hostname2 = factory.make_hostname()
self.set_fake_twisted_dns_reply([hostname])
service = ReverseDNSService()
yield service.startService()
ip = factory.make_ip_address(ipv6=False)
yield service.consumeNeighbourEvent("create", "%s/32" % ip)
self.set_fake_twisted_dns_reply([hostname2])
yield service.consumeNeighbourEvent("update", "%s/32" % ip)
service.stopService()
result = yield deferToDatabase(RDNS.objects.first)
self.assertThat(result.ip, Equals(ip))
self.assertThat(result.hostname, Equals(hostname2))
def make_ReverseDNSService(postgresListener):
from maasserver.regiondservices.reverse_dns import (ReverseDNSService)
return ReverseDNSService(postgresListener)
