40 lines
985 B
Ruby
40 lines
985 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
RSpec.describe Gitlab::Instrumentation::RateLimitingGates, :request_store do
|
||
|
|
describe '.gates' do
|
||
|
|
it 'returns an empty array when no gates are tracked' do
|
||
|
|
expect(described_class.gates).to eq([])
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'returns all gates used in the request' do
|
||
|
|
described_class.track(:foo)
|
||
|
|
|
||
|
|
RequestStore.clear!
|
||
|
|
|
||
|
|
described_class.track(:bar)
|
||
|
|
described_class.track(:baz)
|
||
|
|
|
||
|
|
expect(described_class.gates).to contain_exactly(:bar, :baz)
|
||
|
|
end
|
||
|
|
|
||
|
|
it 'deduplicates its results' do
|
||
|
|
described_class.track(:foo)
|
||
|
|
described_class.track(:bar)
|
||
|
|
described_class.track(:foo)
|
||
|
|
|
||
|
|
expect(described_class.gates).to contain_exactly(:foo, :bar)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
describe '.payload' do
|
||
|
|
it 'returns the gates in a hash' do
|
||
|
|
described_class.track(:foo)
|
||
|
|
described_class.track(:bar)
|
||
|
|
|
||
|
|
expect(described_class.payload).to eq(described_class::GATES => [:foo, :bar])
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|