test: Added wait for live test

redis · Apr 22, 2024 · 76befec · 76befec
1 parent e7e5e57
commit 76befec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/plugins/riot/src/test/java/com/redis/riot/cli/AbstractReplicationTests.java b/plugins/riot/src/test/java/com/redis/riot/cli/AbstractReplicationTests.java
@@ -14,6 +14,7 @@
 import com.redis.spring.batch.gen.GeneratorItemReader;
 import com.redis.spring.batch.gen.Item.Type;
 import com.redis.spring.batch.test.KeyspaceComparison;
+import com.redis.spring.batch.util.Await;
 
 import io.lettuce.core.cluster.SlotHash;
 
@@ -89,6 +90,7 @@ void liveKeyExclude(TestInfo info) throws Throwable {
 		generator2.setKeyspace("bad");
 		generateAsync(testInfo(info, "gen-2"), generator2);
 		execute(info, "replicate-live-key-exclude");
+		Await.await().until(() -> redisCommands.pubsubNumpat() == 0);
 		Assertions.assertEquals(badCount, keyCount("bad:*"));
 		Assertions.assertEquals(0, targetRedisCommands.keys("bad:*").size());
 		Assertions.assertEquals(goodCount, targetRedisCommands.keys("gen:*").size());