spec/lib/gitlab/instrumentation/rate_limiting_gates_spec.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
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