Add service metadata update unit test

fluxcd · Apr 4, 2020 · d34d60f · d34d60f
1 parent 3de8e99
commit d34d60f
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 10 deletions.
diff --git a/pkg/controller/finalizer.go b/pkg/controller/finalizer.go
@@ -55,7 +55,7 @@ func (c *Controller) finalize(old interface{}) error {
 	}
 
 	// Revert the Kubernetes service
-	router := c.routerFactory.KubernetesRouter(canary.Spec.TargetRef.Kind, labelSelector, map[string]string{}, ports)
+	router := c.routerFactory.KubernetesRouter(canary.Spec.TargetRef.Kind, labelSelector, ports)
 	if err := router.Finalize(canary); err != nil {
 		return fmt.Errorf("failed revert router: %w", err)
 	}

diff --git a/pkg/controller/scheduler.go b/pkg/controller/scheduler.go
@@ -108,7 +108,7 @@ func (c *Controller) advanceCanary(name string, namespace string) {
 	}
 
 	// init Kubernetes router
-	kubeRouter := c.routerFactory.KubernetesRouter(cd.Spec.TargetRef.Kind, labelSelector, map[string]string{}, ports)
+	kubeRouter := c.routerFactory.KubernetesRouter(cd.Spec.TargetRef.Kind, labelSelector, ports)
 	if err := kubeRouter.Initialize(cd); err != nil {
 		c.recordEventWarningf(cd, "%v", err)
 		return

diff --git a/pkg/router/factory.go b/pkg/router/factory.go
@@ -35,7 +35,7 @@ func NewFactory(kubeConfig *restclient.Config, kubeClient kubernetes.Interface,
 }
 
 // KubernetesRouter returns a KubernetesRouter interface implementation
-func (factory *Factory) KubernetesRouter(kind string, labelSelector string, annotations map[string]string, ports map[string]int32) KubernetesRouter {
+func (factory *Factory) KubernetesRouter(kind string, labelSelector string, ports map[string]int32) KubernetesRouter {
 	switch kind {
 	case "Service":
 		return &KubernetesNoopRouter{}
@@ -45,7 +45,6 @@ func (factory *Factory) KubernetesRouter(kind string, labelSelector string, anno
 			flaggerClient: factory.flaggerClient,
 			kubeClient:    factory.kubeClient,
 			labelSelector: labelSelector,
-			annotations:   annotations,
 			ports:         ports,
 		}
 	}

diff --git a/pkg/router/kubernetes_default.go b/pkg/router/kubernetes_default.go
@@ -24,7 +24,6 @@ type KubernetesDefaultRouter struct {
 	flaggerClient clientset.Interface
 	logger        *zap.SugaredLogger
 	labelSelector string
-	annotations   map[string]string
 	ports         map[string]int32
 }
 
@@ -119,15 +118,12 @@ func (c *KubernetesDefaultRouter) reconcileService(canary *flaggerv1.Canary, nam
 	if metadata.Labels == nil {
 		metadata.Labels = make(map[string]string)
 	}
+	metadata.Labels[c.labelSelector] = name
+
 	if metadata.Annotations == nil {
 		metadata.Annotations = make(map[string]string)
 	}
 
-	metadata.Labels[c.labelSelector] = name
-	for k, v := range c.annotations {
-		metadata.Annotations[k] = v
-	}
-
 	c.logger.With("canary", fmt.Sprintf("%s.%s", canary.Name, canary.Namespace)).
 		Debugw(fmt.Sprintf("Creating Service %s", name), "metadata", metadata, "service_configuration", canary.Spec.Service)
 

diff --git a/pkg/router/kubernetes_default_test.go b/pkg/router/kubernetes_default_test.go
@@ -413,4 +413,18 @@ func TestServiceRouter_ReconcileMetadata(t *testing.T) {
 	require.NoError(t, err)
 	assert.Equal(t, 0, len(primarySvc.Annotations))
 	assert.Equal(t, "podinfo-primary", primarySvc.Labels["app"])
+
+	mocks.canary.Spec.Service.Apex = &flaggerv1.CustomMetadata{
+		Labels:      map[string]string{"test": "test1"},
+		Annotations: map[string]string{"test1": "test"},
+	}
+
+	err = router.Reconcile(mocks.canary)
+	require.NoError(t, err)
+
+	apexSvc, err = mocks.kubeClient.CoreV1().Services("default").Get("podinfo", metav1.GetOptions{})
+	require.NoError(t, err)
+	assert.Equal(t, "test", apexSvc.Annotations["test1"])
+	assert.Equal(t, "test1", apexSvc.Labels["test"])
+	assert.Equal(t, "podinfo", apexSvc.Labels["app"])
 }