Skip to content

Commit 6b5faa3

Browse files
committed
RUBY-932 Add test for Cluster#remove
1 parent 37994f5 commit 6b5faa3

File tree

1 file changed

+41
-0
lines changed

1 file changed

+41
-0
lines changed

spec/mongo/cluster_spec.rb

Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -210,6 +210,47 @@
210210
end
211211
end
212212

213+
describe '#remove' do
214+
215+
let(:address_a) do
216+
Mongo::Address.new('127.0.0.1:27017')
217+
end
218+
219+
let(:address_b) do
220+
Mongo::Address.new('127.0.0.1:27018')
221+
end
222+
223+
let(:server_a) do
224+
Mongo::Server.new(address_a, Mongo::Event::Listeners.new)
225+
end
226+
227+
let(:server_b) do
228+
Mongo::Server.new(address_b, Mongo::Event::Listeners.new)
229+
end
230+
231+
let(:servers) do
232+
[ server_a, server_b ]
233+
end
234+
235+
let(:addresses) do
236+
[ address_a, address_b ]
237+
end
238+
239+
before do
240+
cluster.instance_variable_set(:@servers, servers)
241+
cluster.instance_variable_set(:@addresses, addresses)
242+
cluster.remove('127.0.0.1:27017')
243+
end
244+
245+
it 'removes the host from the list of servers' do
246+
expect(cluster.instance_variable_get(:@servers)).to eq([server_b])
247+
end
248+
249+
it 'removes the host from the list of addresses' do
250+
expect(cluster.instance_variable_get(:@addresses)).to eq([address_b])
251+
end
252+
end
253+
213254
describe '#add_hosts' do
214255

215256
let(:servers) do

0 commit comments

Comments
 (0)