def test_process_doesnt_proxy_update_config_when_nothing_to_process(self): service = RegionControllerService(sentinel.listener) service.needsProxyUpdate = False mock_proxy_update_config = self.patch(region_controller, "proxy_update_config") service.startProcessing() yield service.processingDefer self.assertThat(mock_proxy_update_config, MockNotCalled())
def test_process_updates_proxy_logs_failure(self): service = RegionControllerService(sentinel.listener) service.needsProxyUpdate = True mock_proxy_update_config = self.patch(region_controller, "proxy_update_config") mock_proxy_update_config.return_value = fail(factory.make_exception()) mock_err = self.patch(region_controller.log, "err") service.startProcessing() yield service.processingDefer self.assertThat(mock_proxy_update_config, MockCalledOnceWith(reload_proxy=True)) self.assertThat(mock_err, MockCalledOnceWith(ANY, "Failed configuring proxy."))
def test_process_updates_proxy(self): service = RegionControllerService(sentinel.listener) service.needsProxyUpdate = True mock_proxy_update_config = self.patch(region_controller, "proxy_update_config") mock_proxy_update_config.return_value = succeed(None) mock_msg = self.patch(region_controller.log, "msg") service.startProcessing() yield service.processingDefer self.assertThat(mock_proxy_update_config, MockCalledOnceWith(reload_proxy=True)) self.assertThat(mock_msg, MockCalledOnceWith("Successfully configured proxy."))
def test_process_updates_bind_and_proxy(self): service = RegionControllerService(sentinel.listener) service.needsDNSUpdate = True service.needsProxyUpdate = True mock_dns_update_all_zones = self.patch( region_controller, "dns_update_all_zones") mock_proxy_update_config = self.patch( region_controller, "proxy_update_config") mock_proxy_update_config.return_value = succeed(None) service.startProcessing() yield service.processingDefer self.assertThat( mock_dns_update_all_zones, MockCalledOnceWith()) self.assertThat( mock_proxy_update_config, MockCalledOnceWith(reload_proxy=True))
def test_process_updates_bind_and_proxy(self): service = RegionControllerService(sentinel.listener) service.needsDNSUpdate = True service.needsProxyUpdate = True dns_result = (random.randint(1, 1000), [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_proxy_update_config = self.patch(region_controller, "proxy_update_config") mock_proxy_update_config.return_value = succeed(None) service.startProcessing() yield service.processingDefer self.assertThat(mock_dns_update_all_zones, MockCalledOnceWith()) self.assertThat(mock_check_serial, MockCalledOnceWith(dns_result)) self.assertThat(mock_proxy_update_config, MockCalledOnceWith(reload_proxy=True))