def test_process_request_disabled(self): self.config(maintenance_mode=False, group='service:api') request = FakeRequest() app = middleware.MaintenanceMiddleware({}) # Process the request response = app(request) # Ensure request was not blocked self.assertEqual(response, 'FakeResponse')
def test_process_request_enabled_reject_no_context(self): self.config(maintenance_mode=True, maintenance_mode_role='admin', group='service:api') request = FakeRequest() app = middleware.MaintenanceMiddleware({}) # Process the request response = app(request) # Ensure request was blocked self.assertEqual(503, response.status_code)
def test_process_request_enabled_bypass(self): self.config(maintenance_mode=True, maintenance_mode_role='admin', group='service:api') request = FakeRequest() request.environ['context'] = context.DesignateContext(roles=['admin']) app = middleware.MaintenanceMiddleware({}) # Process the request response = app(request) # Ensure request was not blocked self.assertEqual('FakeResponse', response)