diff options
Diffstat (limited to 'spec/lib/gitlab/instrumentation/rate_limiting_gates_spec.rb')
-rw-r--r-- | spec/lib/gitlab/instrumentation/rate_limiting_gates_spec.rb | 39 |
1 files changed, 39 insertions, 0 deletions
diff --git a/spec/lib/gitlab/instrumentation/rate_limiting_gates_spec.rb b/spec/lib/gitlab/instrumentation/rate_limiting_gates_spec.rb new file mode 100644 index 00000000000..ac308eb7c80 --- /dev/null +++ b/spec/lib/gitlab/instrumentation/rate_limiting_gates_spec.rb @@ -0,0 +1,39 @@ +# 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 |