spec/lib/gitlab/sidekiq_throttler_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
40
41
42
43
44

require 'spec_helper'

describe Gitlab::SidekiqThrottler do
  describe '#execute!' do
    context 'when job throttling is enabled' do
      before do
        Sidekiq.options[:concurrency] = 35

        stub_application_setting(
          sidekiq_throttling_enabled: true,
          sidekiq_throttling_factor: 0.1,
          sidekiq_throttling_queues: %w[build project_cache]
        )
      end

      it 'requires sidekiq-limit_fetch' do
        expect(described_class).to receive(:require).with('sidekiq-limit_fetch').and_call_original

        described_class.execute!
      end

      it 'sets limits on the selected queues' do
        described_class.execute!

        expect(Sidekiq::Queue['build'].limit).to eq 4
        expect(Sidekiq::Queue['project_cache'].limit).to eq 4
      end

      it 'does not set limits on other queues' do
        described_class.execute!

        expect(Sidekiq::Queue['merge'].limit).to be_nil
      end
    end

    context 'when job throttling is disabled' do
      it 'does not require sidekiq-limit_fetch' do
        expect(described_class).not_to receive(:require).with('sidekiq-limit_fetch')

        described_class.execute!
      end
    end
  end
end