mirror of
https://github.com/samsonjs/rack-attack.git
synced 2026-03-25 09:25:49 +00:00
Acceptance test ActiveSupport::Cache::DalliStore (via dalli) as cache store backend
This commit is contained in:
parent
93b1bf4c86
commit
bca253c674
2 changed files with 39 additions and 2 deletions
39
spec/acceptance/stores/active_support_dalli_store_spec.rb
Normal file
39
spec/acceptance/stores/active_support_dalli_store_spec.rb
Normal file
|
|
@ -0,0 +1,39 @@
|
|||
require_relative "../../spec_helper"
|
||||
require_relative "../../support/cache_store_helper"
|
||||
|
||||
require "active_support/cache/dalli_store"
|
||||
require "timecop"
|
||||
|
||||
describe "ActiveSupport::Cache::DalliStore as a cache backend" do
|
||||
before do
|
||||
Rack::Attack.cache.store = ActiveSupport::Cache::DalliStore.new
|
||||
end
|
||||
|
||||
after do
|
||||
Rack::Attack.cache.store.clear
|
||||
end
|
||||
|
||||
it_works_for_cache_backed_features
|
||||
|
||||
it "doesn't leak keys" do
|
||||
Rack::Attack.throttle("by ip", limit: 1, period: 1) do |request|
|
||||
request.ip
|
||||
end
|
||||
|
||||
key = nil
|
||||
|
||||
# Freeze time during these statement to be sure that the key used by rack attack is the same
|
||||
# we pre-calculate in local variable `key`
|
||||
Timecop.freeze do
|
||||
key = "rack::attack:#{Time.now.to_i}:by ip:1.2.3.4"
|
||||
|
||||
get "/", {}, "REMOTE_ADDR" => "1.2.3.4"
|
||||
end
|
||||
|
||||
assert Rack::Attack.cache.store.fetch(key)
|
||||
|
||||
sleep 2.1
|
||||
|
||||
assert_nil Rack::Attack.cache.store.fetch(key)
|
||||
end
|
||||
end
|
||||
|
|
@ -15,12 +15,10 @@ describe Rack::Attack::Cache do
|
|||
sleep(@expires_in * 1.1) # Add 10% to reduce errors
|
||||
end
|
||||
|
||||
require 'active_support/cache/dalli_store'
|
||||
require 'connection_pool'
|
||||
|
||||
cache_stores = [
|
||||
ActiveSupport::Cache::MemoryStore.new,
|
||||
ActiveSupport::Cache::DalliStore.new("127.0.0.1"),
|
||||
Dalli::Client.new,
|
||||
ConnectionPool.new { Dalli::Client.new }
|
||||
]
|
||||
|
|
|
|||
Loading…
Reference in a new issue