rack-attack/spec/acceptance/rails_middleware_spec.rb

# frozen_string_literal: true

require_relative "../spec_helper"

if defined?(Rails) && Gem::Version.new(Rails::VERSION::STRING) >= Gem::Version.new("5")
  describe "Middleware for Rails" do
    before do
      @app = Class.new(Rails::Application) do
        config.eager_load = false
        config.logger = Logger.new(nil) # avoid creating the log/ directory automatically
        config.cache_store = :null_store # avoid creating tmp/ directory for cache
      end
    end

    it "is enabled by default" do
      @app.initialize!
      assert_equal 1, @app.middleware.count(Rack::Attack)
    end

    it "is not added when it was added explicitly" do
      @app.config.middleware.use(Rack::Attack)
      @app.initialize!
      assert_equal 1, @app.middleware.count(Rack::Attack)
    end

    it "is not added when it was explicitly deleted" do
      @app.config.middleware.delete(Rack::Attack)
      @app.initialize!
      refute @app.middleware.include?(Rack::Attack)
    end
  end
end