51 lines
1.4 KiB
Ruby
51 lines
1.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
describe Gitlab::RateLimitHelpers, :clean_gitlab_redis_shared_state do
|
|
let(:limiter_class) do
|
|
Class.new do
|
|
include ::Gitlab::RateLimitHelpers
|
|
|
|
attr_reader :request
|
|
|
|
def initialize(request)
|
|
@request = request
|
|
end
|
|
end
|
|
end
|
|
|
|
let(:request) { instance_double(ActionDispatch::Request, request_method: 'GET', ip: '127.0.0.1', fullpath: '/') }
|
|
let(:class_instance) { limiter_class.new(request) }
|
|
|
|
let_it_be(:user) { create(:user) }
|
|
let_it_be(:project) { create(:project) }
|
|
|
|
describe '#archive_rate_limit_reached?' do
|
|
context 'with a user' do
|
|
it 'rate limits the user properly' do
|
|
5.times do
|
|
expect(class_instance.archive_rate_limit_reached?(user, project)).to be_falsey
|
|
end
|
|
|
|
expect(class_instance.archive_rate_limit_reached?(user, project)).to be_truthy
|
|
end
|
|
end
|
|
|
|
context 'with an anonymous user' do
|
|
before do
|
|
stub_const('Gitlab::RateLimitHelpers::ARCHIVE_RATE_ANONYMOUS_THRESHOLD', 2)
|
|
end
|
|
|
|
it 'rate limits with higher limits' do
|
|
2.times do
|
|
expect(class_instance.archive_rate_limit_reached?(nil, project)).to be_falsey
|
|
end
|
|
|
|
expect(class_instance.archive_rate_limit_reached?(nil, project)).to be_truthy
|
|
expect(class_instance.archive_rate_limit_reached?(user, project)).to be_falsey
|
|
end
|
|
end
|
|
end
|
|
end
|