From 623f2d528968f72024ec8d83acad9663ae17cb92 Mon Sep 17 00:00:00 2001 From: Derek Collison Date: Sun, 9 Apr 2023 08:13:20 -0700 Subject: [PATCH] Fix flapping test Signed-off-by: Derek Collison --- server/jetstream_cluster_2_test.go | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/server/jetstream_cluster_2_test.go b/server/jetstream_cluster_2_test.go index a13ada9c..775ea9c5 100644 --- a/server/jetstream_cluster_2_test.go +++ b/server/jetstream_cluster_2_test.go @@ -5238,12 +5238,14 @@ func TestJetStreamClusterDeleteAndRestoreAndRestart(t *testing.T) { nc, js = jsClientConnect(t, c.randomServer()) defer nc.Close() - si, err := js.StreamInfo("TEST") - require_NoError(t, err) - - if si.State.Msgs != 22 { - t.Fatalf("State is not correct after restart") - } + checkFor(t, 2*time.Second, 100*time.Millisecond, func() error { + si, err := js.StreamInfo("TEST") + require_NoError(t, err) + if si.State.Msgs != 22 { + return fmt.Errorf("State is not correct after restart, expected 22 msgs, got %d", si.State.Msgs) + } + return nil + }) ci, err := js.ConsumerInfo("TEST", "dlc") require_NoError(t, err)