JAVA-2848: Make it easier to run tests with a replica set

If no org.mongodb.test.uri system property is provided,
try to figure out what it should be by interrogating the server at the
default host and port.  If a setName is found in the response to an
isMaster command, make the uri "mongodb://localhost?/replicaSet=<setName>"

Author: jyemin

driver-async/src/test/functional/com/mongodb/async/client/Fixture.java

@@ -35,11 +35,8 @@
  * Helper class for asynchronous tests.
  */
 public final class Fixture {
-    public static final String DEFAULT_URI = "mongodb://localhost:27017";
-    public static final String MONGODB_URI_SYSTEM_PROPERTY_NAME = "org.mongodb.test.uri";
     private static final String DEFAULT_DATABASE_NAME = "JavaDriverTest";
 
-    private static ConnectionString connectionString;
     private static MongoClientImpl mongoClient;
 
 
@@ -69,13 +66,7 @@ public void apply(final SslSettings.Builder builder) {
     }
 
     public static synchronized ConnectionString getConnectionString() {
-        if (connectionString == null) {
-            String mongoURIProperty = System.getProperty(MONGODB_URI_SYSTEM_PROPERTY_NAME);
-            String mongoURIString = mongoURIProperty == null || mongoURIProperty.isEmpty()
-                                    ? DEFAULT_URI : mongoURIProperty;
-            connectionString = new ConnectionString(mongoURIString);
-        }
-        return connectionString;
+        return ClusterFixture.getConnectionString();
     }
 
     public static String getDefaultDatabaseName() {

driver-core/src/test/functional/com/mongodb/ClusterFixture.java

@@ -50,6 +50,7 @@
 import com.mongodb.operation.AsyncReadOperation;
 import com.mongodb.operation.AsyncWriteOperation;
 import com.mongodb.operation.BatchCursor;
+import com.mongodb.operation.CommandReadOperation;
 import com.mongodb.operation.CommandWriteOperation;
 import com.mongodb.operation.DropDatabaseOperation;
 import com.mongodb.operation.ReadOperation;
@@ -98,10 +99,6 @@ public final class ClusterFixture {
     private static Map<ReadPreference, AsyncReadWriteBinding> asyncBindingMap = new HashMap<ReadPreference, AsyncReadWriteBinding>();
 
     static {
-        String mongoURIProperty = System.getProperty(MONGODB_URI_SYSTEM_PROPERTY_NAME);
-        String mongoURIString = mongoURIProperty == null || mongoURIProperty.isEmpty()
-                ? DEFAULT_URI : mongoURIProperty;
-        connectionString = new ConnectionString(mongoURIString);
         Runtime.getRuntime().addShutdownHook(new ShutdownHook());
     }
 
@@ -188,7 +185,37 @@ public void run() {
     }
 
     public static synchronized ConnectionString getConnectionString() {
-        return connectionString;
+        if (connectionString != null) {
+            return connectionString;
+        }
+
+        String mongoURIProperty = System.getProperty(MONGODB_URI_SYSTEM_PROPERTY_NAME);
+        if (mongoURIProperty != null && !mongoURIProperty.isEmpty()) {
+            connectionString = new ConnectionString(mongoURIProperty);
+            return connectionString;
+        }
+
+        // Figure out what the connection string should be
+        Cluster cluster = createCluster(new ConnectionString(DEFAULT_URI),
+                new SocketStreamFactory(SocketSettings.builder().build(), SslSettings.builder().build()));
+        try {
+            BsonDocument isMasterResult = new CommandReadOperation<BsonDocument>("admin",
+                    new BsonDocument("ismaster", new BsonInt32(1)), new BsonDocumentCodec()).execute(new ClusterBinding(cluster,
+                    ReadPreference.nearest(), ReadConcern.DEFAULT));
+            if (isMasterResult.containsKey("setName")) {
+                connectionString = new ConnectionString(DEFAULT_URI + "/?replicaSet="
+                        + isMasterResult.getString("setName").getValue());
+            } else {
+                connectionString = new ConnectionString(DEFAULT_URI);
+                ClusterFixture.cluster = cluster;
+            }
+
+            return connectionString;
+        } finally {
+            if (ClusterFixture.cluster == null) {
+                cluster.close();
+            }
+        }
     }
 
     public static ReadWriteBinding getBinding(final Cluster cluster) {
@@ -263,15 +290,19 @@ public static synchronized Cluster getAsyncCluster() {
         return asyncCluster;
     }
 
-    @SuppressWarnings("deprecation")
     public static Cluster createCluster(final StreamFactory streamFactory) {
-        return new DefaultClusterFactory().createCluster(ClusterSettings.builder().applyConnectionString(getConnectionString()).build(),
+        return createCluster(getConnectionString(), streamFactory);
+    }
+
+    @SuppressWarnings("deprecation")
+    private static Cluster createCluster(final ConnectionString connectionString, final StreamFactory streamFactory) {
+        return new DefaultClusterFactory().createCluster(ClusterSettings.builder().applyConnectionString(connectionString).build(),
                 ServerSettings.builder().build(),
-                ConnectionPoolSettings.builder().applyConnectionString(getConnectionString()).build(),
+                ConnectionPoolSettings.builder().applyConnectionString(connectionString).build(),
                 streamFactory,
-                new SocketStreamFactory(SocketSettings.builder().build(), getSslSettings()),
-                getConnectionString().getCredentialList(), null, null, null,
-                getConnectionString().getCompressorList());
+                new SocketStreamFactory(SocketSettings.builder().build(), getSslSettings(connectionString)),
+                connectionString.getCredentialList(), null, null, null,
+                connectionString.getCompressorList());
     }
 
     public static StreamFactory getAsyncStreamFactory() {

driver-legacy/src/test/functional/com/mongodb/Fixture.java

@@ -24,8 +24,6 @@
  * Helper class for the acceptance tests.
  */
 public final class Fixture {
-    private static final String DEFAULT_URI = "mongodb://localhost:27017";
-    private static final String MONGODB_URI_SYSTEM_PROPERTY_NAME = "org.mongodb.test.uri";
     private static final String DEFAULT_DATABASE_NAME = "JavaDriverTest";
 
     private static MongoClient mongoClient;
@@ -85,9 +83,7 @@ public void run() {
     }
 
     public static synchronized String getMongoClientURIString() {
-        String mongoURIProperty = System.getProperty(MONGODB_URI_SYSTEM_PROPERTY_NAME);
-        return mongoURIProperty == null || mongoURIProperty.isEmpty()
-               ? DEFAULT_URI : mongoURIProperty;
+        return ClusterFixture.getConnectionString().getConnectionString();
     }
 
     public static synchronized MongoClientURI getMongoClientURI() {

driver-sync/src/test/functional/com/mongodb/client/Fixture.java

@@ -17,6 +17,7 @@
 package com.mongodb.client;
 
 import com.mongodb.Block;
+import com.mongodb.ClusterFixture;
 import com.mongodb.ConnectionString;
 import com.mongodb.MongoClientSettings;
 import com.mongodb.ServerAddress;
@@ -29,8 +30,6 @@
  * Helper class for the acceptance tests.
  */
 public final class Fixture {
-    private static final String DEFAULT_URI = "mongodb://localhost:27017";
-    private static final String MONGODB_URI_SYSTEM_PROPERTY_NAME = "org.mongodb.test.uri";
     private static final String DEFAULT_DATABASE_NAME = "JavaDriverTest";
 
     private static MongoClient mongoClient;
@@ -75,8 +74,7 @@ public void run() {
     }
 
     private static synchronized String getConnectionStringProperty() {
-        String mongoURIProperty = System.getProperty(MONGODB_URI_SYSTEM_PROPERTY_NAME);
-        return mongoURIProperty == null || mongoURIProperty.isEmpty() ? DEFAULT_URI : mongoURIProperty;
+        return ClusterFixture.getConnectionString().getConnectionString();
     }
 
     public static synchronized MongoClientSettings getMongoClientSettings() {