Contents

module Mongoid
  module Expectations

    def connection_class
      if defined?(Mongo::Server::ConnectionBase)
        Mongo::Server::ConnectionBase
      else
        # Pre-2.8 drivers
        Mongo::Server::Connection
      end
    end

    def expect_query(number)
      RSpec::Mocks.with_temporary_scope do
        if number > 0
          expect_any_instance_of(connection_class).to receive(:command_started).exactly(number).times.and_call_original
        else
          expect_any_instance_of(connection_class).not_to receive(:command_started)
        end
        yield
      end
    end

    def expect_no_queries(&block)
      expect_query(0, &block)
    end
  end
end

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
mongoid-7.0.13	spec/support/expectations.rb
mongoid-7.0.12	spec/support/expectations.rb
mongoid-7.0.11	spec/support/expectations.rb
mongoid-7.0.10	spec/support/expectations.rb
mongoid-7.0.8	spec/support/expectations.rb
mongoid-7.0.7	spec/support/expectations.rb
mongoid-7.0.6	spec/support/expectations.rb