From 21fe32b895c61daf03aebde3a4244bb43720ff03 Mon Sep 17 00:00:00 2001 From: Gonzalo Rodriguez Date: Mon, 26 Mar 2018 16:00:32 -0300 Subject: [PATCH] Acceptance test ability to subscribe to blocking events --- spec/acceptance/blocking_spec.rb | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/spec/acceptance/blocking_spec.rb b/spec/acceptance/blocking_spec.rb index 2e9a562..f245a6b 100644 --- a/spec/acceptance/blocking_spec.rb +++ b/spec/acceptance/blocking_spec.rb @@ -18,4 +18,24 @@ describe "#blocklist" do assert_equal 200, last_response.status end + + it "notifies when the request is blocked" do + notification_matched = nil + notification_type = nil + + ActiveSupport::Notifications.subscribe("rack.attack") do |_name, _start, _finish, _id, request| + notification_matched = request.env["rack.attack.matched"] + notification_type = request.env["rack.attack.match_type"] + end + + get "/", {}, "REMOTE_ADDR" => "5.6.7.8" + + assert_nil notification_matched + assert_nil notification_type + + get "/", {}, "REMOTE_ADDR" => "1.2.3.4" + + assert_equal "block 1.2.3.4", notification_matched + assert_equal :blocklist, notification_type + end end