summaryrefslogtreecommitdiff
path: root/spec/lib/gitlab/metrics/sidekiq_middleware_spec.rb
blob: 5882e7d81c7c55059a6d71836ae937c6ecbeaede (plain)
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
require 'spec_helper'

describe Gitlab::Metrics::SidekiqMiddleware do
  let(:middleware) { described_class.new }

  describe '#call' do
    it 'tracks the transaction' do
      worker = Class.new.new

      expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:finish)

      middleware.call(worker, 'test', :test) { nil }
    end
  end

  describe '#tag_worker' do
    it 'adds the worker class and action to the transaction' do
      trans  = Gitlab::Metrics::Transaction.new
      worker = double(:worker, class: double(:class, name: 'TestWorker'))

      expect(trans).to receive(:add_tag).with(:action, 'TestWorker#perform')

      middleware.tag_worker(trans, worker)
    end
  end
end