debian-mirror-gitlab/spec/lib/gitlab/metrics/sidekiq_middleware_spec.rb

57 lines
1.8 KiB
Ruby
Raw Normal View History

2019-12-04 20:38:33 +05:30
# frozen_string_literal: true
require 'spec_helper'
describe Gitlab::Metrics::SidekiqMiddleware do
let(:middleware) { described_class.new }
2016-08-24 12:49:21 +05:30
let(:message) { { 'args' => ['test'], 'enqueued_at' => Time.new(2016, 6, 23, 6, 59).to_f } }
describe '#call' do
it 'tracks the transaction' do
worker = double(:worker, class: double(:class, name: 'TestWorker'))
2018-03-17 18:26:18 +05:30
expect(Gitlab::Metrics::BackgroundTransaction).to receive(:new)
.with(worker.class)
2017-09-10 17:25:29 +05:30
.and_call_original
2017-09-10 17:25:29 +05:30
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:set)
.with(:sidekiq_queue_duration, instance_of(Float))
2016-09-13 17:45:13 +05:30
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:finish)
2016-08-24 12:49:21 +05:30
middleware.call(worker, message, :test) { nil }
end
it 'tracks the transaction (for messages without `enqueued_at`)' do
worker = double(:worker, class: double(:class, name: 'TestWorker'))
2018-03-17 18:26:18 +05:30
expect(Gitlab::Metrics::BackgroundTransaction).to receive(:new)
.with(worker.class)
2017-09-10 17:25:29 +05:30
.and_call_original
2016-08-24 12:49:21 +05:30
2017-09-10 17:25:29 +05:30
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:set)
.with(:sidekiq_queue_duration, instance_of(Float))
2016-09-13 17:45:13 +05:30
2016-08-24 12:49:21 +05:30
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:finish)
middleware.call(worker, {}, :test) { nil }
end
2016-09-13 17:45:13 +05:30
it 'tracks any raised exceptions' do
worker = double(:worker, class: double(:class, name: 'TestWorker'))
2017-09-10 17:25:29 +05:30
expect_any_instance_of(Gitlab::Metrics::Transaction)
.to receive(:run).and_raise(RuntimeError)
2016-09-13 17:45:13 +05:30
2017-09-10 17:25:29 +05:30
expect_any_instance_of(Gitlab::Metrics::Transaction)
.to receive(:add_event).with(:sidekiq_exception)
2016-09-13 17:45:13 +05:30
2017-09-10 17:25:29 +05:30
expect_any_instance_of(Gitlab::Metrics::Transaction)
.to receive(:finish)
2016-09-13 17:45:13 +05:30
2017-09-10 17:25:29 +05:30
expect { middleware.call(worker, message, :test) }
.to raise_error(RuntimeError)
2016-09-13 17:45:13 +05:30
end
end
end