fix(capi-cluster): add labels and annotations to machinepool (#849)

Signed-off-by: David van der Spek <[email protected]>
pluralsh · Sep 14, 2023 · c504d31 · c504d31
1 parent 1696106
commit c504d31
Show file tree

Hide file tree

Showing 4 changed files with 91 additions and 2 deletions.
diff --git a/bootstrap/helm/cluster-api-cluster/Chart.yaml b/bootstrap/helm/cluster-api-cluster/Chart.yaml
@@ -2,5 +2,5 @@ apiVersion: v2
 name: cluster-api-cluster
 description: A Helm chart for Kubernetes
 type: application
-version: 0.1.43
+version: 0.1.44
 appVersion: v1.24.16
diff --git a/bootstrap/helm/cluster-api-cluster/templates/_helpers.tpl b/bootstrap/helm/cluster-api-cluster/templates/_helpers.tpl
@@ -208,6 +208,17 @@ metadata:
   name: {{ .name }}
   annotations:
     helm.sh/resource-policy: keep
+    {{- if (hasKey .values "annotations") -}}
+    {{- toYaml (merge .values.annotations .defaultVals.annotations)| nindent 4 }}
+    {{- else -}}
+    {{- toYaml .defaultVals.annotations | nindent 4 }}
+    {{- end }}
+  labels:
+    {{- if (hasKey .values "labels") -}}
+    {{- toYaml (merge .values.labels .defaultVals.labels)| nindent 4 }}
+    {{- else -}}
+    {{- toYaml .defaultVals.labels | nindent 4 }}
+    {{- end }}
 spec:
   clusterName: {{ .ctx.Values.cluster.name }}
   replicas: {{ $replicas }}

diff --git a/bootstrap/helm/cluster-api-cluster/tests/aws_machinepools_test.yaml b/bootstrap/helm/cluster-api-cluster/tests/aws_machinepools_test.yaml
@@ -30,6 +30,8 @@ tests:
       type: managed
       cluster.aws.region: abc
       cluster.kubernetesVersion: v1.2.3
+      workers.defaults.aws.labels:
+        test: test
       workers.defaults.aws.spec.availabilityZones:
       - us-east-1a
       - us-east-1b
@@ -39,11 +41,78 @@ tests:
           path: spec.amiType
           value: AL2_x86_64
         documentIndex: 0
+      - isSubset:
+          path: metadata.labels
+          content:
+            test: test
+      - isSubset:
+          path: metadata.annotations
+          content:
+            cluster.x-k8s.io/replicas-managed-by: external-autoscaler
       - equal:
           path: spec.diskSize
           value: 50
         documentIndex: 2
     template: aws/machinepools.yaml
+  - it: test override
+    set:
+      cluster.name: test
+      provider: aws
+      type: managed
+      cluster.aws.region: abc
+      cluster.kubernetesVersion: v1.2.3
+      workers.defaults.aws.spec.availabilityZones:
+      - us-east-1a
+      - us-east-1b
+      - us-east-1c
+      workers.aws.small-burst-spot:
+        labels:
+          test: test
+        annotations:
+          test: test
+        spec:
+          amiType: AL2_x86_64_GPU
+          diskSize: 100
+    asserts:
+      - equal:
+          path: spec.eksNodegroupName
+          value: small-burst-spot
+        documentIndex: 22
+      - equal:
+          path: metadata.name
+          value: small-burst-spot
+        documentIndex: 23
+      - isSubset:
+          path: metadata.labels
+          content:
+            test: test
+        documentIndex: 22
+      - isSubset:
+          path: metadata.annotations
+          content:
+            cluster.x-k8s.io/replicas-managed-by: external-autoscaler
+            test: test
+        documentIndex: 22
+      - isSubset:
+          path: metadata.labels
+          content:
+            test: test
+        documentIndex: 23
+      - isSubset:
+          path: metadata.annotations
+          content:
+            cluster.x-k8s.io/replicas-managed-by: external-autoscaler
+            test: test
+        documentIndex: 23
+      - equal:
+          path: spec.diskSize
+          value: 100
+        documentIndex: 22
+      - equal:
+          path: spec.amiType
+          value: AL2_x86_64_GPU
+        documentIndex: 22
+    template: aws/machinepools.yaml
   - it: test large-burst-spot
     set:
       cluster.name: test

diff --git a/bootstrap/helm/cluster-api-cluster/tests/gcp_machinepools_test.yaml b/bootstrap/helm/cluster-api-cluster/tests/gcp_machinepools_test.yaml
@@ -6,6 +6,8 @@ tests:
     set:
       provider: gcp
       type: managed
+      workers.defaults.gcp.labels:
+        test: test
     asserts:
       - hasDocuments:
           count: 6
@@ -51,6 +53,14 @@ tests:
         equal:
           path: spec.machineType
           value: e2-standard-2
+      - isSubset:
+          path: metadata.labels
+          content:
+            test: test
+      - isSubset:
+          path: metadata.annotations
+          content:
+            cluster.x-k8s.io/replicas-managed-by: external-autoscaler
   - it: should have custom kubernetes version set
     set:
       provider: gcp
@@ -67,4 +77,3 @@ tests:
     asserts:
       - hasDocuments:
           count: 0
-