From 98757253f990b4ff1bfdf2d1213f1dd477eef4c3 Mon Sep 17 00:00:00 2001 From: Derek Collison Date: Thu, 18 Nov 2021 14:50:22 -0800 Subject: [PATCH] Recreate client in case shutdown server was the one we were connected to Signed-off-by: Derek Collison --- server/norace_test.go | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) diff --git a/server/norace_test.go b/server/norace_test.go index f30297f8..c8e07acf 100644 --- a/server/norace_test.go +++ b/server/norace_test.go @@ -3770,11 +3770,19 @@ func TestNoRaceJetStreamClusterStreamSeqMismatchIssue(t *testing.T) { t.Fatalf("Unexpected error: %v", err) } } + // Close in case we are connected here. Will recreate. + nc.Close() // Shutdown a non-leader. s := c.randomNonStreamLeader("$G", "KV_MM") s.Shutdown() + nc, js = jsClientConnect(t, c.randomServer()) + defer nc.Close() + + kv, err = js.KeyValue("MM") + require_NoError(t, err) + // Now change the state of the stream such that we have to do a compact upon restart // of the downed server. for i := 1; i <= 10; i++ { @@ -3856,10 +3864,19 @@ func TestNoRaceJetStreamClusterStreamDropCLFS(t *testing.T) { _, err = kv.Create("k.2", []byte("Z")) require_NoError(t, err) + // Close in case we are connected here. Will recreate. + nc.Close() + // Shutdown, which will also clear clfs. s := c.randomNonStreamLeader("$G", "KV_CLFS") s.Shutdown() + nc, js = jsClientConnect(t, c.randomServer()) + defer nc.Close() + + kv, err = js.KeyValue("CLFS") + require_NoError(t, err) + // Drive up CLFS state on leader. for i := 0; i < 10; i++ { _, err = kv.Create("k.1", []byte("X"))