From f0e1585490c5fa69cc0fa6d383a0f86a81432d8f Mon Sep 17 00:00:00 2001
From: Derek Collison <derek@nats.io>
Date: Fri, 17 Mar 2023 13:14:43 -0700
Subject: [PATCH] Fix flapping test

Signed-off-by: Derek Collison <derek@nats.io>
---
 server/jetstream_cluster_3_test.go | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/server/jetstream_cluster_3_test.go b/server/jetstream_cluster_3_test.go
index 84ba1e96..8c94ef47 100644
--- a/server/jetstream_cluster_3_test.go
+++ b/server/jetstream_cluster_3_test.go
@@ -3012,9 +3012,14 @@ func TestJetStreamClusterWorkQueueAfterScaleUp(t *testing.T) {
 	sendStreamMsg(t, nc, "WQ", "SOME WORK")
 	<-wch
 
-	si, err := js.StreamInfo("TEST")
-	require_NoError(t, err)
-	require_True(t, si.State.Msgs == 0)
+	checkFor(t, time.Second, 200*time.Millisecond, func() error {
+		si, err := js.StreamInfo("TEST")
+		require_NoError(t, err)
+		if si.State.Msgs == 0 {
+			return nil
+		}
+		return fmt.Errorf("Still have %d msgs left", si.State.Msgs)
+	})
 }
 
 func TestJetStreamClusterInterestBasedStreamAndConsumerSnapshots(t *testing.T) {