From c369a26c031cea79ee0142d211d2b72f449251bc Mon Sep 17 00:00:00 2001 From: Ivan Kozlovic Date: Mon, 12 Apr 2021 09:31:33 -0600 Subject: [PATCH] Fixed leafnode flapper Signed-off-by: Ivan Kozlovic --- server/leafnode_test.go | 22 ++++++++++++---------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/server/leafnode_test.go b/server/leafnode_test.go index f3c6aaa3..9d4bf3f4 100644 --- a/server/leafnode_test.go +++ b/server/leafnode_test.go @@ -3307,16 +3307,18 @@ func TestLeafNodeNoPingBeforeConnect(t *testing.T) { } var leaf *client - s.grMu.Lock() - for _, l := range s.grTmpClients { - leaf = l - break - } - s.grMu.Unlock() - - if leaf == nil { - t.Fatal("No leaf connection found") - } + checkFor(t, time.Second, 15*time.Millisecond, func() error { + s.grMu.Lock() + for _, l := range s.grTmpClients { + leaf = l + break + } + s.grMu.Unlock() + if leaf == nil { + return fmt.Errorf("No leaf connection found") + } + return nil + }) // Make sure that ping timer is not set leaf.mu.Lock()