test(cli/clitest): Fix panic in Wait

mafredri · mafredri · commit fddeb0010348 · 2023-04-20T18:29:49.000Z
diff --git a/cli/clitest/clitest.go b/cli/clitest/clitest.go
@@ -231,11 +231,14 @@ func StartWithWaiter(t *testing.T, inv *clibase.Invocation) *ErrorWaiter {
 		errCh <- err
 	}()
 
+	ew := &ErrorWaiter{c: errCh, t: t}
+
 	// Don't exit test routine until server is done.
 	t.Cleanup(func() {
 		cancel()
 		cleaningUp.Store(true)
+		_ = ew.Wait() // We must allow Wait to deplete errCh because if we deplete it here Wait will panic.
 		<-errCh
 	})
-	return &ErrorWaiter{c: errCh, t: t}
+	return ew
 }
