54 lines
1.8 KiB
Ruby
54 lines
1.8 KiB
Ruby
require 'spec_helper'
|
|
|
|
describe Gitlab::Metrics::SidekiqMiddleware do
|
|
let(:middleware) { described_class.new }
|
|
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'))
|
|
|
|
expect(Gitlab::Metrics::Transaction).to receive(:new).
|
|
with('TestWorker#perform').
|
|
and_call_original
|
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:set).
|
|
with(:sidekiq_queue_duration, instance_of(Float))
|
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:finish)
|
|
|
|
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'))
|
|
|
|
expect(Gitlab::Metrics::Transaction).to receive(:new).
|
|
with('TestWorker#perform').
|
|
and_call_original
|
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:set).
|
|
with(:sidekiq_queue_duration, instance_of(Float))
|
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction).to receive(:finish)
|
|
|
|
middleware.call(worker, {}, :test) { nil }
|
|
end
|
|
|
|
it 'tracks any raised exceptions' do
|
|
worker = double(:worker, class: double(:class, name: 'TestWorker'))
|
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction).
|
|
to receive(:run).and_raise(RuntimeError)
|
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction).
|
|
to receive(:add_event).with(:sidekiq_exception)
|
|
|
|
expect_any_instance_of(Gitlab::Metrics::Transaction).
|
|
to receive(:finish)
|
|
|
|
expect { middleware.call(worker, message, :test) }.
|
|
to raise_error(RuntimeError)
|
|
end
|
|
end
|
|
end
|