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)), )