blob: 48595c3f172376d42f0cf685a22843f6749f85d3 (
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
# frozen_string_literal: true
require 'spec_helper'
RSpec.describe Gitlab::Memory::Watchdog::SidekiqEventReporter, feature_category: :application_performance do
let(:counter) { instance_double(::Prometheus::Client::Counter) }
before do
allow(Gitlab::Metrics).to receive(:counter).and_return(counter)
allow(counter).to receive(:increment)
end
describe 'delegations' do
it { is_expected.to delegate_method(:started).to(:event_reporter) }
it { is_expected.to delegate_method(:stopped).to(:event_reporter) }
it { is_expected.to delegate_method(:threshold_violated).to(:event_reporter) }
it { is_expected.to delegate_method(:logger).to(:event_reporter) }
end
describe '#strikes_exceeded' do
let(:sidekiq_event_reporter) { described_class.new(logger: logger) }
let(:sidekiq_watchdog_running_jobs_counter) { instance_double(::Prometheus::Client::Counter) }
let(:logger) { instance_double(::Logger) }
let(:queue) { 'default' }
let(:jid) { SecureRandom.hex }
let(:running_jobs) { { jid => { worker_class: DummyWorker } } }
let(:sidekiq_daemon_monitor) { instance_double(Gitlab::SidekiqDaemon::Monitor) }
let(:worker) do
Class.new do
def self.name
'DummyWorker'
end
end
end
before do
stub_const('DummyWorker', worker)
allow(Gitlab::SidekiqDaemon::Monitor).to receive(:instance).and_return(sidekiq_daemon_monitor)
allow(::Gitlab::Metrics).to receive(:counter)
.with(:sidekiq_watchdog_running_jobs_total, anything)
.and_return(sidekiq_watchdog_running_jobs_counter)
allow(sidekiq_watchdog_running_jobs_counter).to receive(:increment)
allow(logger).to receive(:warn)
allow(sidekiq_daemon_monitor).to receive(:jobs).and_return(running_jobs)
end
it 'delegates #strikes_exceeded with correct arguments' do
is_expected.to delegate_method(:strikes_exceeded).to(:event_reporter)
.with_arguments(
:monitor_name,
{
message: 'dummy_text',
running_jobs: [jid: jid, worker_class: 'DummyWorker']
}
)
end
it 'increment running jobs counter' do
expect(sidekiq_watchdog_running_jobs_counter).to receive(:increment)
.with({ worker_class: "DummyWorker" })
sidekiq_event_reporter.strikes_exceeded(:monitor_name, { message: 'dummy_text' })
end
end
end
|