diff --git a/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/AsyncCqlTemplateUnitTests.java b/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/AsyncCqlTemplateUnitTests.java index 157267547..a63e49df9 100644 --- a/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/AsyncCqlTemplateUnitTests.java +++ b/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/AsyncCqlTemplateUnitTests.java @@ -43,16 +43,7 @@ import org.springframework.data.cassandra.CassandraInvalidQueryException; import org.springframework.scheduling.annotation.AsyncResult; import org.springframework.util.concurrent.ListenableFuture; -import com.datastax.driver.core.BoundStatement; -import com.datastax.driver.core.ColumnDefinitions; -import com.datastax.driver.core.ConsistencyLevel; -import com.datastax.driver.core.PreparedStatement; -import com.datastax.driver.core.ResultSet; -import com.datastax.driver.core.ResultSetFuture; -import com.datastax.driver.core.Row; -import com.datastax.driver.core.Session; -import com.datastax.driver.core.SimpleStatement; -import com.datastax.driver.core.Statement; +import com.datastax.driver.core.*; import com.datastax.driver.core.exceptions.InvalidQueryException; import com.datastax.driver.core.exceptions.NoHostAvailableException; import com.datastax.driver.core.policies.DowngradingConsistencyRetryPolicy; @@ -578,6 +569,7 @@ public class AsyncCqlTemplateUnitTests { when(session.executeAsync(boundStatement)).thenReturn(new TestResultSetFuture(resultSet)); when(resultSet.iterator()).thenReturn(Collections.singleton(row).iterator()); + when(resultSet.spliterator()).thenCallRealMethod(); ListenableFuture future = template .query(session -> new AsyncResult(preparedStatement), ps -> { @@ -773,6 +765,7 @@ public class AsyncCqlTemplateUnitTests { when(this.session.executeAsync((Statement) any())).thenReturn(new TestResultSetFuture(resultSet)); when(this.resultSet.iterator()).thenReturn(Arrays.asList(row, row, row).iterator()); + when(this.resultSet.spliterator()).thenCallRealMethod(); when(this.row.getString(0)).thenReturn(results[0], results[1], results[2]); diff --git a/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/CqlTemplateUnitTests.java b/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/CqlTemplateUnitTests.java index 4cbc3dfff..feaed66e6 100644 --- a/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/CqlTemplateUnitTests.java +++ b/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/cql/CqlTemplateUnitTests.java @@ -513,6 +513,7 @@ public class CqlTemplateUnitTests { when(session.execute(boundStatement)).thenReturn(resultSet); when(resultSet.iterator()).thenReturn(Collections.singleton(row).iterator()); + when(resultSet.spliterator()).thenCallRealMethod(); ResultSet resultSet = template.query(session -> preparedStatement, ps -> { ps.bind("a", "b"); @@ -684,6 +685,7 @@ public class CqlTemplateUnitTests { when(this.session.execute((Statement) any())).thenReturn(resultSet); when(this.resultSet.iterator()).thenReturn(Arrays.asList(row, row, row).iterator()); + when(this.resultSet.spliterator()).thenCallRealMethod(); when(this.row.getString(0)).thenReturn(results[0], results[1], results[2]); when(this.session.prepare(anyString())).thenReturn(preparedStatement);