Fix failure tests in stream plugin

1 files changed, 3 insertions, 3 deletions

diff --git a/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/FailureTest.java b/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/FailureTest.java
index c82749f340..fe1df75488 100644
--- a/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/FailureTest.java
+++ b/deps/rabbitmq_stream/test/rabbit_stream_SUITE_data/src/test/java/com/rabbitmq/stream/FailureTest.java
@@ -142,7 +142,7 @@ public class FailureTest {
     assertThat(confirmLatch.get().await(10, TimeUnit.SECONDS)).isTrue();
     confirmLatch.set(null);
 
-    CountDownLatch consumeLatch = new CountDownLatch(2);
+    CountDownLatch consumeLatch = new CountDownLatch(messages.size());
     Set<String> bodies = ConcurrentHashMap.newKeySet();
     Client consumer =
         cf.get(
@@ -163,8 +163,8 @@ public class FailureTest {
         });
     assertThat(consumeLatch.await(10, TimeUnit.SECONDS)).isTrue();
     assertThat(bodies)
-        .hasSize(3)
-        .contains("all nodes available", "2 nodes available", "all nodes are back");
+        .hasSameSizeAs(messages)
+        .containsAll(messages);
   }
 
   @Test