fix(dot/sync): fix Test_lockQueue_threadSafety

ChainSafe · Jun 15, 2022 · 4453181 · 4453181
1 parent 7f55bec
commit 4453181
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/dot/sync/block_queue_test.go b/dot/sync/block_queue_test.go
@@ -219,6 +219,17 @@ func Test_lockQueue_threadSafety(t *testing.T) {
 	}
 	blockHash := common.Hash{1}
 
+	endWg.Add(1)
+	go func() {
+		defer endWg.Done()
+		<-ctx.Done()
+		// Empty queue channel to make sure `push` does not block
+		// when the context is cancelled.
+		for len(blockQueue.queue) > 0 {
+			<-blockQueue.queue
+		}
+	}()
+
 	for i := 0; i < parallelism; i++ {
 		go runInLoop(func() {
 			blockQueue.push(blockData)