fix: target manager is now properly checked for during shutdown

Signed-off-by: Bruno Bressi <[email protected]>

pkg/sparrow/run.go

@@ -105,7 +105,7 @@ func (s *Sparrow) Run(ctx context.Context) error {
 		s.cErr <- s.loader.Run(ctx)
 	}()
 	go func() {
-		if s.cfg.HasTargetManager() {
+		if s.tarMan != nil {
 			s.cErr <- s.tarMan.Reconcile(ctx)
 		}
 	}()
@@ -316,7 +316,7 @@ func (s *Sparrow) shutdown(ctx context.Context) {
 	s.shutOnce.Do(func() {
 		log.Info("Shutting down sparrow gracefully")
 		var errS error
-		if s.cfg.HasTargetManager() {
+		if s.tarMan != nil {
 			errS = s.tarMan.Shutdown(ctx)
 		}
 		errA := s.shutdownAPI(ctx)

pkg/sparrow/run_test.go

@@ -243,20 +243,11 @@ func TestSparrow_Run_ContextCancel(t *testing.T) {
 			File:     config.FileLoaderConfig{Path: "../config/testdata/config.yaml"},
 			Interval: time.Second * 1,
 		},
-		TargetManager: config.TargetManagerConfig{
-			CheckInterval:        time.Second * 1,
-			RegistrationInterval: time.Second * 1,
-			UnhealthyThreshold:   time.Second * 1,
-			Gitlab: config.GitlabTargetManagerConfig{
-				BaseURL:   "https://gitlab.com",
-				Token:     "my-cool-token",
-				ProjectID: 42,
-			},
-		},
 	}
 
 	// start sparrow
 	s := New(c)
+	s.tarMan = &gitlabmock.MockTargetManager{}
 	ctx, cancel := context.WithCancel(context.Background())
 	go func() {
 		err := s.Run(ctx)