def test_process_updates_zones_logs_reason_for_multiple_updates(self):
# Create some fake serial updates with sources for the update.
def _create_publications():
return [
DNSPublication.objects.create(
source=factory.make_name('reason')) for _ in range(3)
]
publications = yield deferToDatabase(_create_publications)
service = RegionControllerService(sentinel.listener)
service.needsDNSUpdate = True
service.previousSerial = publications[0].serial
dns_result = (publications[-1].serial,
[factory.make_name('domain') for _ in range(3)])
mock_dns_update_all_zones = self.patch(region_controller,
"dns_update_all_zones")
mock_dns_update_all_zones.return_value = dns_result
mock_check_serial = self.patch(service, "_checkSerial")
mock_check_serial.return_value = succeed(dns_result)
mock_msg = self.patch(region_controller.log, "msg")
service.startProcessing()
yield service.processingDefer
expected_msg = "Reloaded DNS configuration: \n"
expected_msg += '\n'.join(
' * %s' % publication.source
for publication in reversed(publications[1:]))
self.assertThat(mock_dns_update_all_zones, MockCalledOnceWith())
self.assertThat(mock_check_serial, MockCalledOnceWith(dns_result))
self.assertThat(mock_msg, MockCalledOnceWith(expected_msg))
def test_process_updates_zones_logs_reason_for_single_update(self):
# Create some fake serial updates with sources for the update.
def _create_publications():
return [
DNSPublication.objects.create(
source=factory.make_name("reason")) for _ in range(2)
]
publications = yield deferToDatabase(_create_publications)
service = RegionControllerService(sentinel.listener)
service.needsDNSUpdate = True
service.previousSerial = publications[0].serial
dns_result = (
publications[-1].serial,
True,
[factory.make_name("domain") for _ in range(3)],
)
mock_dns_update_all_zones = self.patch(region_controller,
"dns_update_all_zones")
mock_dns_update_all_zones.return_value = dns_result
mock_check_serial = self.patch(service, "_checkSerial")
mock_check_serial.return_value = succeed(dns_result)
mock_msg = self.patch(region_controller.log, "msg")
service.startProcessing()
yield service.processingDefer
self.assertThat(mock_dns_update_all_zones, MockCalledOnceWith())
self.assertThat(mock_check_serial, MockCalledOnceWith(dns_result))
self.assertThat(
mock_msg,
MockCalledOnceWith("Reloaded DNS configuration; %s" %
(publications[-1].source)),
)