Allow throttling code in test environment

This code should not break other tests because the rate limits are off by default.

config/initializers/rack_attack_global.rb

@@ -21,27 +21,23 @@ class Rack::Attack
     { limit: limit_proc, period: period_proc }
   end
 
-  def self.define_throttles
+  throttle('throttle_unauthenticated', throttle_unauthenticated_options) do |req|
-    throttle('throttle_unauthenticated', throttle_unauthenticated_options) do |req|
+    settings.throttle_unauthenticated_enabled &&
-      settings.throttle_unauthenticated_enabled &&
+      req.unauthenticated? &&
-        req.unauthenticated? &&
+      req.ip
-        req.ip
-    end
-
-    throttle('throttle_authenticated_api', throttle_authenticated_api_options) do |req|
-      settings.throttle_authenticated_api_enabled &&
-        req.api_request? &&
-        req.authenticated_user_id
-    end
-
-    throttle('throttle_authenticated_web', throttle_authenticated_web_options) do |req|
-      settings.throttle_authenticated_web_enabled &&
-        req.web_request? &&
-        req.authenticated_user_id
-    end
   end
 
-  define_throttles unless Rails.env.test?
+  throttle('throttle_authenticated_api', throttle_authenticated_api_options) do |req|
+    settings.throttle_authenticated_api_enabled &&
+      req.api_request? &&
+      req.authenticated_user_id
+  end
+
+  throttle('throttle_authenticated_web', throttle_authenticated_web_options) do |req|
+    settings.throttle_authenticated_web_enabled &&
+      req.web_request? &&
+      req.authenticated_user_id
+  end
 
   class Request
     def unauthenticated?

spec/requests/rack_attack_spec.rb

@@ -4,9 +4,6 @@ describe Rack::Attack do
   let(:settings) { Gitlab::CurrentSettings.current_application_settings }
 
   before do
-    # Ensure throttles are defined, because this is normally skipped for tests
-    described_class.define_throttles
-
     # Instead of test environment's :null_store
     Rack::Attack.cache.store = ActiveSupport::Cache::MemoryStore.new