test(hare): add cleanup

spacemeshos · Aug 7, 2024 · ddee087 · ddee087
1 parent 6b39ce4
commit ddee087
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hare4/hare_test.go b/hare4/hare_test.go
@@ -598,12 +598,15 @@ func (cl *lockstepCluster) waitStopped() {
 // drainInteractiveMessages will make sure that the channels that signal
 // that interactive messages came in on the tracer are read from.
 func (cl *lockstepCluster) drainInteractiveMessages() {
+	done := make(chan struct{})
+	ct.t.Cleanup(func() { close(done) })
 	for _, n := range cl.nodes {
 		go func() {
 			for {
 				select {
 				case <-n.tracer.compactReq:
 				case <-n.tracer.compactResp:
+				case <-done:
 				}
 			}
 		}()