Fix should_wait_for_each_contact_point_at_most_once for 2.1+.

tolbertam · tolbertam · commit 3ffa863dace3 · 2015-05-27T18:19:34.000-05:00
diff --git a/driver-core/src/test/java/com/datastax/driver/core/ClusterInitTest.java b/driver-core/src/test/java/com/datastax/driver/core/ClusterInitTest.java
@@ -92,10 +92,7 @@ public void should_wait_for_each_contact_point_at_most_once() {
 
             // We have one live host so 2 successful connections (1 control connection and 1 core connection in the pool).
             // The other 5 hosts are unreachable, we should attempt to connect to each of them only once.
-            int coreConnections = cluster.getConfiguration()
-                    .getPoolingOptions()
-                    .getCoreConnectionsPerHost(HostDistance.LOCAL);
-            verify(socketOptions, times(1 + coreConnections + 5)).getKeepAlive();
+            verify(socketOptions, times(1 + TestUtils.numberOfLocalCoreConnections(cluster) + 5)).getKeepAlive();
         } finally {
             if (cluster != null)
                 cluster.close();
