2843

Author: p

lib/mongo/collection.rb

@@ -493,6 +493,7 @@ def count(filter = nil, options = {})
     #   command to run.
     # @option options :read [ Hash ] The read preference options.
     # @option options :collation [ Hash ] The collation to use.
+    # @option options [ Session ] :session The session to use.
     #
     # @return [ Integer ] The document count.
     #

lib/mongo/collection/view/readable.rb

@@ -207,6 +207,7 @@ def count(opts = {})
         #   command to run.
         # @option opts [ Hash ] :read The read preference options.
         # @option opts [ Hash ] :collation The collation to use.
+        # @option opts [ Mongo::Session ] :session The session to use for the operation.
         #
         # @return [ Integer ] The document count.
         #

spec/mongo/collection/view/readable_spec.rb

@@ -671,6 +671,32 @@
     end
   end
 
+  describe '#count_documents' do
+
+    let(:result) do
+      new_view.count_documents(options)
+    end
+
+    context 'when session is given' do
+      let(:subscriber) { Mrss::EventSubscriber.new }
+
+      before do
+        authorized_collection.client.subscribe(Mongo::Monitoring::COMMAND, subscriber)
+      end
+
+      it 'passes the session' do
+        authorized_collection.client.with_session do |session|
+          session_id = session.session_id
+
+          authorized_collection.count_documents({}, session: session)
+
+          event = subscriber.single_command_started_event('aggregate')
+          event.command['lsid'].should == session_id
+        end
+      end
+    end
+  end
+
   describe '#distinct' do
 
     context 'when incorporating read concern' do