cli: fix \demo shutdown with --global

Previously, it crashed due to not setting up the demo's rpc context to include the artificial latency map. Release note: None
cockroachdb · Mar 28, 2021 · 6b1b69a · 6b1b69a
1 parent 3cfe2a3
commit 6b1b69a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkg/cli/start.go b/pkg/cli/start.go
@@ -1075,6 +1075,9 @@ func getClientGRPCConn(
 		Stopper:    stopper,
 		Settings:   cfg.Settings,
 	})
+	if cfg.TestingKnobs.Server != nil {
+		rpcContext.Knobs = cfg.TestingKnobs.Server.(*server.TestingKnobs).ContextTestingKnobs
+	}
 	addr, err := addrWithDefaultHost(cfg.AdvertiseAddr)
 	if err != nil {
 		stopper.Stop(ctx)