rack-attack/spec/integration/rack_attack_cache_spec.rb

require_relative '../spec_helper'

describe Rack::Attack::Cache do
  # A convenience method for deleting a key from cache.
  # Slightly different than @cache.delete, which adds a prefix.
  def delete(key)
    if @cache.store.respond_to?(:delete)
      @cache.store.delete(key)
    else
      @cache.store.del(key)
    end
  end

  def sleep_until_expired
    sleep(@expires_in * 1.1) # Add 10% to reduce errors
  end

  require 'active_support/cache/dalli_store'
  require 'active_support/cache/mem_cache_store'
  require 'active_support/cache/redis_store'
  require 'active_support/cache/redis_cache_store' if ActiveSupport.version.to_s.to_f >= 5.2
  require 'connection_pool'

  cache_stores = [
    ActiveSupport::Cache::MemoryStore.new,
    ActiveSupport::Cache::DalliStore.new("127.0.0.1"),
    ActiveSupport::Cache::RedisStore.new("127.0.0.1"),
    ActiveSupport::Cache::MemCacheStore.new("127.0.0.1"),
    Dalli::Client.new,
    ConnectionPool.new { Dalli::Client.new },
    Redis::Store.new
  ]

  cache_stores << ActiveSupport::Cache::RedisCacheStore.new if defined?(ActiveSupport::Cache::RedisCacheStore)

  cache_stores.each do |store|
    store = Rack::Attack::StoreProxy.build(store)

    describe "with #{store.class}" do
      before {
        @cache = Rack::Attack::Cache.new
        @key = "rack::attack:cache-test-key"
        @expires_in = 1
        @cache.store = store
        delete(@key)
      }

      after { delete(@key) }

      describe "do_count once" do
        it "should be 1" do
          @cache.send(:do_count, @key, @expires_in).must_equal 1
        end
      end

      describe "do_count twice" do
        it "must be 2" do
          @cache.send(:do_count, @key, @expires_in)
          @cache.send(:do_count, @key, @expires_in).must_equal 2
        end
      end

      describe "do_count after expires_in" do
        it "must be 1" do
          @cache.send(:do_count, @key, @expires_in)
          sleep_until_expired
          @cache.send(:do_count, @key, @expires_in).must_equal 1
        end
      end

      describe "write" do
        it "should write a value to the store with prefix" do
          @cache.write("cache-test-key", "foobar", 1)
          store.read(@key).must_equal "foobar"
        end
      end

      describe "write after expiry" do
        it "must not have a value" do
          @cache.write("cache-test-key", "foobar", @expires_in)
          sleep_until_expired
          store.read(@key).must_be :nil?
        end
      end

      describe "read" do
        it "must read the value with a prefix" do
          store.write(@key, "foobar", :expires_in => @expires_in)
          @cache.read("cache-test-key").must_equal "foobar"
        end
      end

      describe "delete" do
        it "must delete the value" do
          store.write(@key, "foobar", :expires_in => @expires_in)
          @cache.read('cache-test-key').must_equal "foobar"
          store.delete(@key)
          assert_nil @cache.read('cache-test-key')
        end
      end

      describe "cache#delete" do
        it "must delete the value" do
          @cache.write("cache-test-key", "foobar", 1)
          store.read(@key).must_equal "foobar"
          @cache.delete('cache-test-key')
          store.read(@key).must_be :nil?
        end
      end

      describe "reset_count" do
        it "must delete the value" do
          period = 1.minute
          unprefixed_key = 'cache-test-key'
          @cache.count(unprefixed_key, period)
          period_key, _ = @cache.send(:key_and_expiry, 'cache-test-key', period)
          store.read(period_key).to_i.must_equal 1
          @cache.reset_count(unprefixed_key, period)
          assert_nil store.read(period_key)
        end
      end
    end
  end
end