map_reduce.rb in mongo-2.12.0.rc0

- old
+ new

@@ -237,11 +237,11 @@
             msg = "Rerouting the MapReduce operation to the primary server - #{server.summary} is not suitable"
             log_warn(msg)
             server = cluster.next_primary(nil, session)
           end
           validate_collation!(server)
-          initial_query_op(session).execute(server)
+          initial_query_op(session).execute(server, client: client)
         end
 
         def fetch_query_spec
           Builder::MapReduce.new(map_function, reduce_function, view, options).query_specification
         end
@@ -257,10 +257,10 @@
             Operation::Find.new(fetch_query_spec)
           end
         end
 
         def send_fetch_query(server, session)
-          fetch_query_op(server, session).execute(server)
+          fetch_query_op(server, session).execute(server, client: client)
         end
 
         def validate_collation!(server)
           if (view.options[:collation] || options[:collation]) && !server.features.collation_enabled?
             raise Error::UnsupportedCollation.new