mirror of
https://github.com/samsonjs/rack-attack.git
synced 2026-04-09 11:55:51 +00:00
5004b04ac7
https://github.com/kickstarter/rack-attack/issues/255 Change the object type from instances of type Rack::Attack::Request to instances of type Hash. (`req` becomes `request: req`).
31 lines
1,005 B
Ruby
31 lines
1,005 B
Ruby
# ActiveSupport::Subscribers added in ~> 4.0.2.0
|
|
if ActiveSupport::VERSION::MAJOR > 3
|
|
require_relative 'spec_helper'
|
|
require 'active_support/subscriber'
|
|
class CustomSubscriber < ActiveSupport::Subscriber
|
|
def rack(event)
|
|
# Do virtually (but not) nothing.
|
|
event.inspect
|
|
end
|
|
end
|
|
|
|
describe 'Rack::Attack.instrument' do
|
|
before do
|
|
@period = 60 # Use a long period; failures due to cache key rotation less likely
|
|
Rack::Attack.cache.store = ActiveSupport::Cache::MemoryStore.new
|
|
Rack::Attack.throttle('ip/sec', limit: 1, period: @period) { |req| req.ip }
|
|
end
|
|
|
|
describe "with throttling" do
|
|
before do
|
|
ActiveSupport::Notifications.stub(:notifier, ActiveSupport::Notifications::Fanout.new) do
|
|
CustomSubscriber.attach_to("attack")
|
|
2.times { get '/', {}, 'REMOTE_ADDR' => '1.2.3.4' }
|
|
end
|
|
end
|
|
it 'should instrument without error' do
|
|
last_response.status.must_equal 429
|
|
end
|
|
end
|
|
end
|
|
end
|