From 76befecee344c883539d961733119a1579b9dc00 Mon Sep 17 00:00:00 2001
From: jruaux <jruaux@gmail.com>
Date: Mon, 22 Apr 2024 14:36:46 -0700
Subject: [PATCH] test: Added wait for live test

---
 .../test/java/com/redis/riot/cli/AbstractReplicationTests.java  | 2 ++
 1 file changed, 2 insertions(+)

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
index ae440fb49..6f18e2060 100644
--- 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());