Update pkg/lib/watcher/manager_test.go

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
bacalhau-project · Nov 21, 2024 · c8735e1 · c8735e1
1 parent 2306540
commit c8735e1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/lib/watcher/manager_test.go b/pkg/lib/watcher/manager_test.go
@@ -152,8 +152,9 @@ func (s *ManagerTestSuite) TestStopWithTimeout() {
 	}, 200*time.Millisecond, 10*time.Millisecond)
 
 	// sleep and verify that the watcher is still stopping
-	time.Sleep(100 * time.Millisecond)
-	s.Require().Equal(watcher.StateStopping, w.Stats().State)
+	s.Require().Eventually(func() bool {
+		return w.Stats().State == watcher.StateStopping
+	}, 200*time.Millisecond, 10*time.Millisecond)
 
 	// Unblock GetEvents
 	close(getEventsCh)