From a7a104e1fdf1ca4ee0f0e629e7918757ced145a4 Mon Sep 17 00:00:00 2001
From: Mark Paluch <mpaluch@pivotal.io>
Date: Fri, 15 Feb 2019 11:28:21 +0100
Subject: [PATCH] DATACASS-628 - Polishing.

Add synchronization to inserts to properly await completion before querying data.
---
 .../cassandra/core/AsyncCassandraTemplateIntegrationTests.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/AsyncCassandraTemplateIntegrationTests.java b/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/AsyncCassandraTemplateIntegrationTests.java
index b5f93194d..5fddad0a7 100644
--- a/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/AsyncCassandraTemplateIntegrationTests.java
+++ b/spring-data-cassandra/src/test/java/org/springframework/data/cassandra/core/AsyncCassandraTemplateIntegrationTests.java
@@ -317,7 +317,7 @@ public class AsyncCassandraTemplateIntegrationTests extends AbstractKeyspaceCrea
 		for (int count = 0; count < 100; count++) {
 			User user = new User("heisenberg" + count, "Walter", "White");
 			expectedIds.add(user.getId());
-			template.insert(user);
+			getUninterruptibly(template.insert(user));
 		}
 
 		Set<String> ids = new HashSet<>();