Spring/spring-data-cassandra
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

DATACASS-56 - Polishing.

Browse Source 
Improve test synchronization.
This commit is contained in:
Mark Paluch
2018-02-19 11:25:59 +01:00
parent 1c53bd09d4
commit 819789086b
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/AsyncCassandraTemplateIntegrationTests.java
View File

@@ -15,16 +15,17 @@
*/
package org.springframework.data.cassandra.core;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.*;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;
import java.util.concurrent.Future;
import org.junit.Before;
import org.junit.Test;
import org.springframework.data.cassandra.core.convert.MappingCassandraConverter;
import org.springframework.data.cassandra.core.cql.AsyncCqlTemplate;
import org.springframework.data.cassandra.core.query.CassandraPageRequest;
@@ -52,7 +53,7 @@ public class AsyncCassandraTemplateIntegrationTests extends AbstractKeyspaceCrea
private AsyncCassandraTemplate template;
@Before
public void setUp() throws Exception {
public void setUp() {
MappingCassandraConverter converter = new MappingCassandraConverter();
CassandraTemplate cassandraTemplate = new CassandraTemplate(session, converter);
@@ -265,13 +266,16 @@ public class AsyncCassandraTemplateIntegrationTests extends AbstractKeyspaceCrea
public void shouldPageRequests() {
Set<String> expectedIds = new LinkedHashSet<>();
List<ListenableFuture<User>> futures = new ArrayList<>();
for (int count = 0; count < 100; count++) {
User user = new User("heisenberg" + count, "Walter", "White");
expectedIds.add(user.getId());
template.insert(user);
futures.add(template.insert(user));
}
futures.forEach(AsyncCassandraTemplateIntegrationTests::getUninterruptibly);
Set<String> ids = new HashSet<>();
Query query = Query.empty();