address coderabbit comments

bacalhau-project · Nov 21, 2024 · b9ffe3d · b9ffe3d
1 parent 383a580
commit b9ffe3d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/lib/watcher/manager_test.go b/pkg/lib/watcher/manager_test.go
@@ -80,15 +80,15 @@ func (s *ManagerTestSuite) TestLookup() {
 	s.Require().NoError(err)
 
 	// Lookup existing watcher
-	retrievedWatcher, err := s.manager.Lookup(watcherID)
+	retrievedWatcher, err := s.manager.Lookup(context.Background(), watcherID)
 	s.Require().NoError(err)
 	s.Require().NotNil(retrievedWatcher)
 	s.Equal(watcherID, retrievedWatcher.ID())
 	s.Equal(w, retrievedWatcher)
 }
 
 func (s *ManagerTestSuite) TestLookupNonExistentWatcher() {
-	_, err := s.manager.Lookup("non-existent")
+	_, err := s.manager.Lookup(context.Background(), "non-existent")
 	s.Require().Error(err)
 	s.Contains(err.Error(), "watcher not found")
 }