Acceptance test ActiveSupport::Cache::MemoryStore (via activesupport) as cache store backend

This commit is contained in:
Gonzalo Rodriguez 2018-06-22 14:11:34 -03:00
parent 1f05ff30d5
commit bcc1f5857e
No known key found for this signature in database
GPG key ID: 5DB8B81B049B8AB1
2 changed files with 103 additions and 72 deletions

View file

@ -0,0 +1,38 @@
require_relative "../../spec_helper"
require_relative "../../support/cache_store_helper"
require "timecop"
describe "ActiveSupport::Cache::MemoryStore as a cache backend" do
before do
Rack::Attack.cache.store = ActiveSupport::Cache::MemoryStore.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

View file

@ -17,13 +17,7 @@ describe Rack::Attack::Cache do
require 'connection_pool' require 'connection_pool'
cache_stores = [ store = Rack::Attack::StoreProxy.build(ConnectionPool.new { Dalli::Client.new })
ActiveSupport::Cache::MemoryStore.new,
ConnectionPool.new { Dalli::Client.new }
]
cache_stores.each do |store|
store = Rack::Attack::StoreProxy.build(store)
describe "with #{store.class}" do describe "with #{store.class}" do
before do before do
@ -109,5 +103,4 @@ describe Rack::Attack::Cache do
end end
end end
end end
end
end end