From ae4374fcd0f1f7a87aa2c09a30a90731d65cb2fa Mon Sep 17 00:00:00 2001
From: Matthias Hanel <mh@synadia.com>
Date: Thu, 22 Apr 2021 20:35:00 -0400
Subject: [PATCH] fix unit test crashing when leader was nil caused by quorum
 reformation

Signed-off-by: Matthias Hanel <mh@synadia.com>
---
 server/jetstream_cluster_test.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/server/jetstream_cluster_test.go b/server/jetstream_cluster_test.go
index 1524ee23..3275cb7c 100644
--- a/server/jetstream_cluster_test.go
+++ b/server/jetstream_cluster_test.go
@@ -6456,6 +6456,10 @@ func (c *cluster) waitOnPeerCount(n int) {
 	c.t.Helper()
 	c.waitOnLeader()
 	leader := c.leader()
+	for leader == nil {
+		c.waitOnLeader()
+		leader = c.leader()
+	}
 	expires := time.Now().Add(10 * time.Second)
 	for time.Now().Before(expires) {
 		peers := leader.JetStreamClusterPeers()