From efdb80cc48d302c0c8232ccece60bff485d1cc19 Mon Sep 17 00:00:00 2001 From: Matthias Hanel Date: Fri, 12 Mar 2021 01:51:28 -0500 Subject: [PATCH] More unit test fixes Signed-off-by: Matthias Hanel --- server/jetstream_cluster_test.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/server/jetstream_cluster_test.go b/server/jetstream_cluster_test.go index 535fb8e5..34c29e4b 100644 --- a/server/jetstream_cluster_test.go +++ b/server/jetstream_cluster_test.go @@ -1721,6 +1721,7 @@ func TestJetStreamClusterExtendedStreamInfo(t *testing.T) { } oldLeader = c.restartServer(oldLeader) + c.waitOnStreamLeader("$G", "TEST") c.waitOnStreamCurrent(oldLeader, "$G", "TEST") // Re-request. @@ -3717,6 +3718,7 @@ func TestJetStreamClusterRemoveServer(t *testing.T) { sl := c.streamLeader("$G", "TEST") c.removeJetStream(sl) + c.waitOnLeader() c.waitOnStreamLeader("$G", "TEST") // Faster timeout since we loop below checking for condition. @@ -5112,7 +5114,7 @@ func (c *cluster) expectNoLeader() { func (c *cluster) waitOnLeader() { c.t.Helper() - expires := time.Now().Add(20 * time.Second) + expires := time.Now().Add(40 * time.Second) for time.Now().Before(expires) { if leader := c.leader(); leader != nil { time.Sleep(100 * time.Millisecond)