Fix scavenger test suite leaky logger (#5490)

cadence-workflow · Dec 18, 2023 · 59f540d · 59f540d
1 parent deed4f0
commit 59f540d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/service/worker/scanner/tasklist/scavenger_test.go b/service/worker/scanner/tasklist/scavenger_test.go
@@ -193,8 +193,16 @@ func (s *ScavengerTestSuite) TestAllExpiredTasksWithErrors() {
 func (s *ScavengerTestSuite) runScavenger() {
 	s.scvgr.Start()
 	timer := time.NewTimer(scavengerTestTimeout)
+
+	waiter := make(chan struct{})
+
+	go func() {
+		s.scvgr.stopWG.Wait()
+		close(waiter)
+	}()
+
 	select {
-	case <-s.scvgr.stopC:
+	case <-waiter:
 		timer.Stop()
 		return
 	case <-timer.C: