Add capability to retrieve preparations in ScassandraCluster.

Clint Ascencio · olim7t · commit ea89817ba1b7 · 2015-08-09T20:19:19.000+02:00
Also primes a query for retrieving peer by peer ip.
diff --git a/driver-core/src/test/java/com/datastax/driver/core/SCassandraCluster.java b/driver-core/src/test/java/com/datastax/driver/core/SCassandraCluster.java
@@ -98,6 +98,10 @@ public List<PreparedStatementExecution> retrievePreparedStatementExecutions(int
         return activityClients.get(node - 1).retrievePreparedStatementExecutions();
     }
 
+    public List<PreparedStatementPreparation> retrievePreparedStatementPreparations(int node) {
+        return activityClients.get(node - 1).retrievePreparedStatementPreparations();
+    }
+
     public void clearAllPrimes() {
         for (PrimingClient primingClient : primingClients)
             primingClient.clearAllPrimes();
@@ -121,14 +125,24 @@ private void primePeers(PrimingClient primingClient, Scassandra toIgnore) {
             if (scassandras.get(i).equals(toIgnore))
                 continue;
             InetAddress address = addresses.get(i);
-            rows.add(ImmutableMap.<String, Object>builder()
+            Map<String, ?> row = ImmutableMap.<String, Object>builder()
                 .put("peer", address)
                 .put("rpc_address", address)
                 .put("data_center", "datacenter1")
                 .put("rack", "rack1")
                 .put("release_version", "2.0.1")
                 .put("tokens", ImmutableSet.of(Long.toString(Long.MIN_VALUE + i)))
-                .build());
+                .build();
+
+            rows.add(row);
+
+            String query = "SELECT * FROM system.peers WHERE peer='" + address.toString().substring(1) + "'";
+            primingClient.prime(
+                PrimingRequest.queryBuilder()
+                    .withQuery(query)
+                    .withColumnTypes(SELECT_PEERS_COLUMN_TYPES)
+                    .withRows(row)
+                    .build());
         }
         primingClient.prime(
             PrimingRequest.queryBuilder()
diff --git a/pom.xml b/pom.xml
@@ -60,11 +60,7 @@
     <testng.version>6.8.8</testng.version>
     <assertj.version>1.7.0</assertj.version>
     <mockito.version>1.10.8</mockito.version>
-    <!--
-      There are more recent versions of Scassandra but they require JDK 7,
-      our build is still using JDK 6.
-    -->
-    <scassandra.version>0.4.1</scassandra.version>
+    <scassandra.version>0.9.1</scassandra.version>
     <main.basedir>${project.basedir}</main.basedir>
   </properties>
 
