sjs/rack-attack
1
0
Fork 0
mirror of https://github.com/samsonjs/rack-attack.git synced 2026-03-25 09:25:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Acceptance test ability to access match data in #blocklisted_response

Browse source
This commit is contained in:
Gonzalo Rodriguez 2018-03-22 11:44:41 -03:00
parent 6614d0b6ee
commit da1f54b6fc
No known key found for this signature in database
GPG key ID: 5DB8B81B049B8AB1
1 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
spec/acceptance/customizing_blocked_response_spec.rb
View file

@ -1,11 +1,13 @@
require_relative "../spec_helper"
describe "Customizing block responses" do
it "can be customized" do
before do
Rack::Attack.blocklist("block 1.2.3.4") do |request|
request.ip == "1.2.3.4"
end
end
it "can be customized" do
get "/", {}, "REMOTE_ADDR" => "1.2.3.4"
assert_equal 403, last_response.status
@ -19,4 +21,21 @@ describe "Customizing block responses" do
assert_equal 503, last_response.status
assert_equal "Blocked", last_response.body
end
it "exposes match data" do
matched = nil
match_type = nil
Rack::Attack.blocklisted_response = lambda do |env|
matched = env['rack.attack.matched']
match_type = env['rack.attack.match_type']
[503, {}, ["Blocked"]]
end
get "/", {}, "REMOTE_ADDR" => "1.2.3.4"
assert_equal "block 1.2.3.4", matched
assert_equal :blocklist, match_type
end
end