Skip to content

Commit e98024c

Browse files
MONGOID-5177 Fix flaky contextual spec (#5064)
1 parent aacc323 commit e98024c

File tree

3 files changed

+16
-2
lines changed

3 files changed

+16
-2
lines changed

spec/mongoid/contextual/mongo_spec.rb

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -640,8 +640,21 @@
640640
end
641641

642642
it "does not load all documents" do
643-
expect(Mongo::Logger.logger).to receive(:debug?).exactly(2).times.and_call_original
643+
subscriber = Mrss::EventSubscriber.new
644+
context.view.client.subscribe(Mongo::Monitoring::COMMAND, subscriber)
645+
644646
enum.next
647+
648+
find_events = subscriber.all_events.select do |evt|
649+
evt.command_name == 'find'
650+
end
651+
expect(find_events.length).to be(2)
652+
get_more_events = subscriber.all_events.select do |evt|
653+
evt.command_name == 'getMore'
654+
end
655+
expect(get_more_events.length).to be(0)
656+
ensure
657+
context.view.client.unsubscribe(Mongo::Monitoring::COMMAND, subscriber)
645658
end
646659
end
647660
end

spec/spec_helper.rb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ def database_id_alt
3131
require 'mrss/cluster_config'
3232
require 'support/client_registry'
3333
require 'mrss/constraints'
34+
require 'mrss/event_subscriber'
3435

3536
ClusterConfig = Mrss::ClusterConfig
3637

0 commit comments

Comments
 (0)