SERVER-41134 Add RSM::findHostsWithMaxWait

Add support for a new method on the ReplicaSetMonitor, which finds all
acceptable hosts for a read preference (rather than just a single host).

This should use existing selection logic to return the host that would
be returned by findHostWithMaxWait in the [0]th element

1 files changed, 6 insertions, 1 deletions

diff --git a/src/mongo/client/remote_command_targeter_mock.h b/src/mongo/client/remote_command_targeter_mock.h
index 89c7245d62b..3ea39b80d22 100644
--- a/src/mongo/client/remote_command_targeter_mock.h
+++ b/src/mongo/client/remote_command_targeter_mock.h
@@ -58,6 +58,9 @@ public:
     SemiFuture<HostAndPort> findHostWithMaxWait(const ReadPreferenceSetting& readPref,
                                                 Milliseconds maxWait) override;
 
+    SemiFuture<std::vector<HostAndPort>> findHostsWithMaxWait(const ReadPreferenceSetting& readPref,
+                                                              Milliseconds maxWait) override;
+
     StatusWith<HostAndPort> findHost(OperationContext* opCtx,
                                      const ReadPreferenceSetting& readPref) override;
 
@@ -81,6 +84,8 @@ public:
      */
     void setFindHostReturnValue(StatusWith<HostAndPort> returnValue);
 
+    void setFindHostsReturnValue(StatusWith<std::vector<HostAndPort>> returnValue);
+
     /**
      * Returns the current set of hosts marked down and resets the mock's internal list of marked
      * down hosts.
@@ -89,7 +94,7 @@ public:
 
 private:
     ConnectionString _connectionStringReturnValue;
-    StatusWith<HostAndPort> _findHostReturnValue;
+    StatusWith<std::vector<HostAndPort>> _findHostReturnValue;
 
     // Protects _hostsMarkedDown.
     mutable stdx::mutex _mutex;