diff options
Diffstat (limited to 'spec/lib/gitlab/application_rate_limiter/increment_per_actioned_resource_spec.rb')
-rw-r--r-- | spec/lib/gitlab/application_rate_limiter/increment_per_actioned_resource_spec.rb | 52 |
1 files changed, 52 insertions, 0 deletions
diff --git a/spec/lib/gitlab/application_rate_limiter/increment_per_actioned_resource_spec.rb b/spec/lib/gitlab/application_rate_limiter/increment_per_actioned_resource_spec.rb new file mode 100644 index 00000000000..1f3ae2d749a --- /dev/null +++ b/spec/lib/gitlab/application_rate_limiter/increment_per_actioned_resource_spec.rb @@ -0,0 +1,52 @@ +# frozen_string_literal: true + +require 'spec_helper' + +RSpec.describe Gitlab::ApplicationRateLimiter::IncrementPerActionedResource, + :freeze_time, :clean_gitlab_redis_rate_limiting do + let(:cache_key) { 'test' } + let(:expiry) { 60 } + + def increment(resource_key) + described_class.new(resource_key).increment(cache_key, expiry) + end + + describe '#increment' do + it 'increments per resource', :aggregate_failures do + expect(increment('resource_1')).to eq(1) + expect(increment('resource_1')).to eq(1) + expect(increment('resource_2')).to eq(2) + expect(increment('resource_2')).to eq(2) + expect(increment('resource_3')).to eq(3) + end + + it 'sets time to live (TTL) for the key' do + def ttl + Gitlab::Redis::RateLimiting.with { |r| r.ttl(cache_key) } + end + + key_does_not_exist = -2 + + expect(ttl).to eq key_does_not_exist + expect { increment('resource_1') }.to change { ttl }.by(a_value > 0) + end + end + + describe '#read' do + def read + described_class.new(nil).read(cache_key) + end + + it 'returns 0 when there is no data' do + expect(read).to eq 0 + end + + it 'returns the correct value', :aggregate_failures do + increment 'r1' + expect(read).to eq 1 + + increment 'r2' + expect(read).to eq 2 + end + end +end |