diff --git a/cassandra/search/src/test/java/org/opennms/newts/cassandra/search/CassandraIndexerStressITCase.java b/cassandra/search/src/test/java/org/opennms/newts/cassandra/search/CassandraIndexerStressITCase.java index 70d21cb2..adff81b7 100644 --- a/cassandra/search/src/test/java/org/opennms/newts/cassandra/search/CassandraIndexerStressITCase.java +++ b/cassandra/search/src/test/java/org/opennms/newts/cassandra/search/CassandraIndexerStressITCase.java @@ -31,6 +31,9 @@ import org.opennms.newts.cassandra.ContextConfigurations; import com.codahale.metrics.MetricRegistry; +import com.datastax.driver.core.BoundStatement; +import com.datastax.driver.core.PreparedStatement; +import com.datastax.driver.core.RegularStatement; import com.datastax.driver.core.ResultSetFuture; import com.datastax.driver.core.Statement; import com.google.common.collect.Lists; @@ -47,6 +50,12 @@ public void canIndexManyResources() { CassandraSession session = mock(CassandraSession.class); when(session.executeAsync(any(Statement.class))).thenReturn(future); + PreparedStatement preparedStatement = mock(PreparedStatement.class); + BoundStatement boundStatement = mock(BoundStatement.class); + when(session.prepare(any(RegularStatement.class))).thenReturn(preparedStatement); + when(preparedStatement.bind()).thenReturn(boundStatement); + when(boundStatement.setString(any(String.class), any(String.class))).thenReturn(boundStatement); + ContextConfigurations contexts = new ContextConfigurations(); MetricRegistry metrics = new MetricRegistry();