spec/workers/gitlab_performance_bar_stats_worker_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

# frozen_string_literal: true

require 'spec_helper'

RSpec.describe GitlabPerformanceBarStatsWorker do
  include ExclusiveLeaseHelpers

  subject(:worker) { described_class.new }

  describe '#perform' do
    let(:redis) { double(Gitlab::Redis::SharedState) }
    let(:uuid) { 1 }

    before do
      expect(Gitlab::Redis::SharedState).to receive(:with).and_yield(redis)
      expect_to_cancel_exclusive_lease(GitlabPerformanceBarStatsWorker::LEASE_KEY, uuid)
    end

    it 'fetches list of request ids and processes them' do
      expect(redis).to receive(:smembers).with(GitlabPerformanceBarStatsWorker::STATS_KEY).and_return([1, 2])
      expect(redis).to receive(:del).with(GitlabPerformanceBarStatsWorker::STATS_KEY)
      expect_next_instance_of(Gitlab::PerformanceBar::Stats) do |stats|
        expect(stats).to receive(:process).with(1)
        expect(stats).to receive(:process).with(2)
      end

      worker.perform(uuid)
    end
  end
end