diff --git a/spec/acceptance/allow2ban_spec.rb b/spec/acceptance/allow2ban_spec.rb new file mode 100644 index 00000000..5495c058 --- /dev/null +++ b/spec/acceptance/allow2ban_spec.rb @@ -0,0 +1,71 @@ +require_relative "../spec_helper" +require "timecop" + +describe "allow2ban" do + before do + Rack::Attack.cache.store = ActiveSupport::Cache::MemoryStore.new + + Rack::Attack.blocklist("allow2ban pentesters") do |request| + Rack::Attack::Allow2Ban.filter(request.ip, maxretry: 2, findtime: 30, bantime: 60) do + request.path.include?("scarce-resource") + end + end + end + + it "returns OK for many requests that doesn't match the filter" do + get "/" + assert_equal 200, last_response.status + + get "/" + assert_equal 200, last_response.status + end + + it "returns OK for first request that matches the filter" do + get "/scarce-resource" + assert_equal 200, last_response.status + end + + it "forbids all access after reaching maxretry limit" do + get "/scarce-resource" + assert_equal 200, last_response.status + + get "/scarce-resource" + assert_equal 200, last_response.status + + get "/scarce-resource" + assert_equal 403, last_response.status + + get "/" + assert_equal 403, last_response.status + end + + it "restores access after bantime elapsed" do + get "/scarce-resource" + assert_equal 200, last_response.status + + get "/scarce-resource" + assert_equal 200, last_response.status + + get "/" + assert_equal 403, last_response.status + + Timecop.travel(60) do + get "/" + + assert_equal 200, last_response.status + end + end + + it "does not forbid all access if maxrety condition is met but not within the findtime timespan" do + get "/scarce-resource" + assert_equal 200, last_response.status + + Timecop.travel(31) do + get "/scarce-resource" + assert_equal 200, last_response.status + + get "/" + assert_equal 200, last_response.status + end + end +end