Need to wait on cluster ready

Signed-off-by: Derek Collison <derek@nats.io>
dmpriso · Sep 7, 2022 · 5345393 · 5345393
1 parent c3203a3
commit 5345393
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/server/jetstream_cluster_test.go b/server/jetstream_cluster_test.go
@@ -3737,12 +3737,10 @@ func TestJetStreamClusterAccountPurge(t *testing.T) {
 	updateJwt(t, c.randomServer().ClientURL(), sysCreds, accJwt, 3)
 
 	createTestData := func(t *testing.T) {
-		nc := natsConnect(t, c.randomNonLeader().ClientURL(), nats.UserCredentials(accCreds))
+		nc, js := jsClientConnect(t, c.randomNonLeader(), nats.UserCredentials(accCreds))
 		defer nc.Close()
 
-		js, err := nc.JetStream()
-		require_NoError(t, err)
-		_, err = js.AddStream(&nats.StreamConfig{
+		_, err := js.AddStream(&nats.StreamConfig{
 			Name:     "TEST1",
 			Subjects: []string{"foo"},
 			Replicas: 3,
@@ -3866,6 +3864,7 @@ func TestJetStreamClusterAccountPurge(t *testing.T) {
 	})
 
 	t.Run("purge-with-restart", func(t *testing.T) {
+		c.waitOnStreamLeader(accpub, "TEST1")
 		createTestData(t)
 		checkForDirs(t, 6, 4)
 		purge(t)