From 569ecec7c7e36e3dec319925b49eae5bc706142c Mon Sep 17 00:00:00 2001 From: Gonzalo Rodriguez Date: Tue, 13 Mar 2018 14:19:22 -0300 Subject: [PATCH] Acceptance test Rack::Attack#track --- spec/acceptance/track_spec.rb | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 spec/acceptance/track_spec.rb diff --git a/spec/acceptance/track_spec.rb b/spec/acceptance/track_spec.rb new file mode 100644 index 0000000..7f4b393 --- /dev/null +++ b/spec/acceptance/track_spec.rb @@ -0,0 +1,27 @@ +require_relative "../spec_helper" + +describe "#track" do + it "notifies when track block returns true" do + Rack::Attack.track("ip 1.2.3.4") do |request| + request.ip == "1.2.3.4" + end + + 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 "ip 1.2.3.4", notification_matched + assert_equal :track, notification_type + end +end