Add equals() and hashCode() to MongoClientURI.

smola · rozza · commit 2b3e75433aaa · 2014-10-07T15:45:38.000+01:00
diff --git a/src/main/com/mongodb/MongoClientURI.java b/src/main/com/mongodb/MongoClientURI.java
@@ -653,4 +653,48 @@ public String toString() {
         return uri;
     }
 
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) {
+            return true;
+        }
+        if (o == null || getClass() != o.getClass()) {
+            return false;
+        }
+
+        MongoClientURI that = (MongoClientURI) o;
+
+        if (collection != null ? !collection.equals(that.collection) : that.collection != null) {
+            return false;
+        }
+        if (credentials != null ? !credentials.equals(that.credentials) : that.credentials != null) {
+            return false;
+        }
+        if (database != null ? !database.equals(that.database) : that.database != null) {
+            return false;
+        }
+        if (hosts != null ? !hosts.equals(that.hosts) : that.hosts != null) {
+            return false;
+        }
+        if (options != null ? !options.equals(that.options) : that.options != null) {
+            return false;
+        }
+        if (uri != null ? !uri.equals(that.uri) : that.uri != null) {
+            return false;
+        }
+
+        return true;
+  }
+
+  @Override
+  public int hashCode() {
+      int result = options != null ? options.hashCode() : 0;
+      result = 31 * result + (credentials != null ? credentials.hashCode() : 0);
+      result = 31 * result + (hosts != null ? hosts.hashCode() : 0);
+      result = 31 * result + (database != null ? database.hashCode() : 0);
+      result = 31 * result + (collection != null ? collection.hashCode() : 0);
+      result = 31 * result + (uri != null ? uri.hashCode() : 0);
+      return result;
+  }
+
 }
diff --git a/src/test/com/mongodb/MongoClientURITest.java b/src/test/com/mongodb/MongoClientURITest.java
@@ -25,6 +25,7 @@
 import static org.junit.Assert.assertArrayEquals;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNotEquals;
 import static org.junit.Assert.assertNull;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
@@ -327,6 +328,31 @@ public void testMultipleIPV6ServersWithPorts() {
         assertEquals("[2010:836B:4179::836B:4179]:2000", u.getHosts().get(1));
     }
 
+    @Test
+    public void testEqualsAndHashCode() {
+      MongoClientURI uris[] = new MongoClientURI[] {
+          new MongoClientURI("mongodb://user:pass@[2010:836B:4179::836B:4179]"),
+          new MongoClientURI("mongodb://localhost/?readPreference=secondaryPreferred"),
+          new MongoClientURI("mongodb://[::1]:1000,[2010:836B:4179::836B:4179]:2000"),
+          new MongoClientURI("mongodb://localhost/?" +
+              "maxPoolSize=10;waitQueueMultiple=5;waitQueueTimeoutMS=150;" +
+              "minPoolSize=7;maxIdleTimeMS=1000;maxLifeTimeMS=2000;" +
+              "replicaSet=test;" +
+              "connectTimeoutMS=2500;socketTimeoutMS=5500;autoConnectRetry=true;" +
+              "slaveOk=true;safe=false;w=1;wtimeout=2500;fsync=true")
+      };
+      for (MongoClientURI uri : uris) {
+        assertEquals(uri, uri);
+        MongoClientURI reinstantiatedUri = new MongoClientURI(uri.getURI());
+        assertEquals(uri, reinstantiatedUri);
+        assertEquals(uri.hashCode(), reinstantiatedUri.hashCode());
+        for (MongoClientURI anotherURI : uris) {
+          if (uri == anotherURI) continue;
+          assertNotEquals(uri, anotherURI);
+        }
+      }
+    }
+
     @SuppressWarnings("deprecation")
     private void assertOnOptions(MongoClientOptions options) {
         assertEquals(10, options.getConnectionsPerHost(), 10);
