retry assertions

apache · Jun 27, 2023 · 7255d9d · 7255d9d
1 parent 2da9a0d
commit 7255d9d
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/cassandra/src/it/scala/org/apache/pekko/projection/cassandra/CassandraProjectionSpec.scala b/cassandra/src/it/scala/org/apache/pekko/projection/cassandra/CassandraProjectionSpec.scala
@@ -368,13 +368,17 @@ class CassandraProjectionSpec
           sourceProbe.get.sendNext(Envelope(entityId, n, s"elem-$n"))
         }
         offsetShouldBe(projectionId, 10L)
-        repository.findById(entityId).futureValue.get.text should include("elem-15")
+        eventually {
+          repository.findById(entityId).futureValue.get.text should include("elem-15")
+        }
 
         (16 to 22).foreach { n =>
           sourceProbe.get.sendNext(Envelope(entityId, n, s"elem-$n"))
         }
         offsetShouldBe(projectionId, 20L)
-        repository.findById(entityId).futureValue.get.text should include("elem-22")
+        eventually {
+          repository.findById(entityId).futureValue.get.text should include("elem-22")
+        }
       }
     }
 
@@ -404,13 +408,17 @@ class CassandraProjectionSpec
           sourceProbe.get.sendNext(Envelope(entityId, n, s"elem-$n"))
         }
         offsetShouldBe(projectionId, 10L)
-        repository.findById(entityId).futureValue.get.text should include("elem-15")
+        eventually {
+          repository.findById(entityId).futureValue.get.text should include("elem-15")
+        }
 
         (16 to 17).foreach { n =>
           sourceProbe.get.sendNext(Envelope(entityId, n, s"elem-$n"))
         }
         offsetShouldBe(projectionId, 17L)
-        repository.findById(entityId).futureValue.get.text should include("elem-17")
+        eventually {
+          repository.findById(entityId).futureValue.get.text should include("elem-17")
+        }
 
       }
     }