fix: controller manager polling (#146)

* fix controller manager polling * fix unit tests
pluralsh · Mar 18, 2024 · 8f9e335 · 8f9e335
1 parent efeca26
commit 8f9e335
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/pkg/controller/controller_manager.go b/pkg/controller/controller_manager.go
@@ -83,7 +83,8 @@ func (cm *ControllerManager) Start() error {
 		go func() {
 			defer controller.Do.ShutdownQueue()
 			defer controller.Do.WipeCache()
-			_ = wait.PollUntilContextTimeout(cm.ctx, cm.Refresh, time.Minute, true, func(ctx context.Context) (done bool, err error) {
+			//nolint:all
+			_ = wait.PollImmediateInfinite(cm.Refresh, func() (done bool, err error) {
 				return controller.Do.Poll(cm.ctx)
 			})
 		}()
@@ -94,7 +95,8 @@ func (cm *ControllerManager) Start() error {
 	}
 
 	go func() {
-		_ = wait.PollUntilContextTimeout(cm.ctx, cm.Refresh, time.Minute, true, func(ctx context.Context) (done bool, err error) {
+		//nolint:all
+		_ = wait.PollImmediateInfinite(cm.Refresh, func() (done bool, err error) {
 			return false, cm.Socket.Join()
 		})
 	}()

diff --git a/pkg/manifests/template/helm_test.go b/pkg/manifests/template/helm_test.go
@@ -19,6 +19,10 @@ var _ = Describe("Helm template", func() {
 	svc := &console.GetServiceDeploymentForAgent_ServiceDeployment{
 		Namespace: "default",
 		Name:      "test",
+		Cluster: &console.GetServiceDeploymentForAgent_ServiceDeployment_Cluster{
+			ID:   "123",
+			Name: "test",
+		},
 	}
 
 	r := gin.Default()

diff --git a/pkg/manifests/template/raw_test.go b/pkg/manifests/template/raw_test.go
@@ -25,6 +25,10 @@ var _ = Describe("Raw template", func() {
 					Value: name,
 				},
 			}
+			svc.Cluster = &console.GetServiceDeploymentForAgent_ServiceDeployment_Cluster{
+				ID:   "123",
+				Name: "test",
+			}
 			resp, err := NewRaw(dir).Render(svc, utilFactory)
 			Expect(err).NotTo(HaveOccurred())
 			Expect(len(resp)).To(Equal(1))