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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
# frozen_string_literal: true
# rubocop:todo all
require 'spec_helper'
describe 'Heartbeat events' do
class HeartbeatEventsSpecTestException < StandardError; end
# 4.4 has two monitors and thus issues heartbeats multiple times
max_server_version '4.2'
clean_slate_for_all
let(:subscriber) { Mrss::EventSubscriber.new }
before do
Mongo::Monitoring::Global.subscribe(Mongo::Monitoring::SERVER_HEARTBEAT, subscriber)
end
after do
Mongo::Monitoring::Global.unsubscribe(Mongo::Monitoring::SERVER_HEARTBEAT, subscriber)
end
let(:address_str) { ClusterConfig.instance.primary_address_str }
let(:client) { new_local_client([address_str],
SpecConfig.instance.all_test_options.merge(
server_selection_timeout: 0.1, connect: :direct)) }
it 'notifies on successful heartbeats' do
client.database.command(ping: 1)
started_event = subscriber.started_events.first
expect(started_event).not_to be nil
expect(started_event.address).to be_a(Mongo::Address)
expect(started_event.address.seed).to eq(address_str)
succeeded_event = subscriber.succeeded_events.first
expect(succeeded_event).not_to be nil
expect(succeeded_event.address).to be_a(Mongo::Address)
expect(succeeded_event.address.seed).to eq(address_str)
failed_event = subscriber.failed_events.first
expect(failed_event).to be nil
end
it 'notifies on failed heartbeats' do
exc = HeartbeatEventsSpecTestException.new
expect_any_instance_of(Mongo::Server::Monitor).to receive(:check).at_least(:once).and_raise(exc)
expect do
client.database.command(ping: 1)
end.to raise_error(Mongo::Error::NoServerAvailable)
started_event = subscriber.started_events.first
expect(started_event).not_to be nil
expect(started_event.address).to be_a(Mongo::Address)
expect(started_event.address.seed).to eq(address_str)
succeeded_event = subscriber.succeeded_events.first
expect(succeeded_event).to be nil
failed_event = subscriber.failed_events.first
expect(failed_event).not_to be nil
expect(failed_event.error).to be exc
expect(failed_event.failure).to be exc
expect(failed_event.address).to be_a(Mongo::Address)
expect(failed_event.address.seed).to eq(address_str)
end
context 'when monitoring option is false' do
let(:client) { new_local_client([address_str],
SpecConfig.instance.all_test_options.merge(
server_selection_timeout: 0.1, connect: :direct, monitoring: false)) }
shared_examples_for 'does not notify on heartbeats' do
it 'does not notify on heartbeats' do
client.database.command(ping: 1)
started_event = subscriber.started_events.first
expect(started_event).to be nil
end
end
it_behaves_like 'does not notify on heartbeats'
context 'when a subscriber is added manually' do
let(:client) do
sdam_proc = Proc.new do |client|
client.subscribe(Mongo::Monitoring::SERVER_HEARTBEAT, subscriber)
end
new_local_client([address_str],
SpecConfig.instance.all_test_options.merge(
server_selection_timeout: 0.1, connect: :direct,
monitoring: false, sdam_proc: sdam_proc))
end
it_behaves_like 'does not notify on heartbeats'
end
end
end
|
