From fa2c0ec5835825be60802fb67ddf12d283c4302d Mon Sep 17 00:00:00 2001
From: David Turanski <dturanski@vmware.com>
Date: Thu, 13 Aug 2020 15:01:31 -0400
Subject: [PATCH] Wait for data

---
 consumer/mongodb-consumer/pom.xml                          | 5 +++++
 .../fn/consumer/mongo/MongoDbConsumerApplicationTests.java | 7 ++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/consumer/mongodb-consumer/pom.xml b/consumer/mongodb-consumer/pom.xml
index 2799602e..0ab63380 100644
--- a/consumer/mongodb-consumer/pom.xml
+++ b/consumer/mongodb-consumer/pom.xml
@@ -48,6 +48,11 @@
             <artifactId>de.flapdoodle.embed.mongo</artifactId>
             <scope>test</scope>
         </dependency>
+        <dependency>
+            <groupId>org.awaitility</groupId>
+            <artifactId>awaitility</artifactId>
+            <scope>test</scope>
+        </dependency>
         <dependency>
             <groupId>org.springframework.boot</groupId>
             <artifactId>spring-boot-configuration-processor</artifactId>
diff --git a/consumer/mongodb-consumer/src/test/java/org/springframework/cloud/fn/consumer/mongo/MongoDbConsumerApplicationTests.java b/consumer/mongodb-consumer/src/test/java/org/springframework/cloud/fn/consumer/mongo/MongoDbConsumerApplicationTests.java
index 421ec335..b667203b 100644
--- a/consumer/mongodb-consumer/src/test/java/org/springframework/cloud/fn/consumer/mongo/MongoDbConsumerApplicationTests.java
+++ b/consumer/mongodb-consumer/src/test/java/org/springframework/cloud/fn/consumer/mongo/MongoDbConsumerApplicationTests.java
@@ -35,6 +35,7 @@ import org.springframework.messaging.Message;
 import org.springframework.messaging.support.GenericMessage;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.awaitility.Awaitility.await;
 
 /**
  * @author David Turanski
@@ -70,7 +71,11 @@ class MongoDbConsumerApplicationTests {
 		messages.map(message -> {
 			mongodbConsumer.accept(message);
 			return message;
-		}).blockLast(Duration.ofSeconds(30));
+
+		}).subscribe();
+
+		await().timeout(Duration.ofSeconds(10))
+				.until(() -> mongoTemplate.findAll(Document.class, properties.getCollection()).count().block() == 3L);
 
 		StepVerifier.create(this.mongoTemplate.findAll(Document.class, properties.getCollection())
 				.sort(Comparator.comparing(d -> d.get("_id").toString())))