Unit tests for PR 450 (#611)

* Add unit tests for #450.
temporalio · Dec 10, 2024 · 51cb16c · 51cb16c
1 parent 56ce221
commit 51cb16c
Show file tree

Hide file tree

Showing 5 changed files with 87 additions and 12 deletions.
diff --git a/charts/temporal/templates/admintools-deployment.yaml b/charts/temporal/templates/admintools-deployment.yaml
@@ -24,7 +24,7 @@ spec:
       {{ include "temporal.serviceAccount" $ }}
       {{- if $.Values.admintools.additionalInitContainers }}
       initContainers:
-        {{- toYaml $.Values.admintools.additionalInitContainers | nindent 8}}
+        {{- toYaml $.Values.admintools.additionalInitContainers | nindent 8 }}
       {{- end }}
       containers:
         - name: admin-tools

diff --git a/charts/temporal/templates/server-job.yaml b/charts/temporal/templates/server-job.yaml
@@ -24,6 +24,9 @@ spec:
       {{ include "temporal.serviceAccount" $ }}
       restartPolicy: OnFailure
       initContainers:
+        {{- if $.Values.admintools.additionalInitContainers }}
+          {{- toYaml $.Values.admintools.additionalInitContainers | nindent 8 }}
+        {{- end }}
         {{- if $.Values.cassandra.enabled }}
         - name: check-cassandra-service
           image: busybox
@@ -56,7 +59,7 @@ spec:
               {{- end }}
           env:
               {{- include "temporal.admintools-env" (list $ $store) | nindent 12 }}
-              {{- with $.Values.server.additionalVolumeMounts }}
+              {{- with $.Values.admintools.additionalVolumeMounts }}
           volumeMounts:
                 {{- toYaml . | nindent 12 }}
               {{- end }}
@@ -83,7 +86,7 @@ spec:
             {{- end }}
           env:
             {{- include "temporal.admintools-env" (list $ $store) | nindent 12 }}
-            {{- with $.Values.server.additionalVolumeMounts }}
+            {{- with $.Values.admintools.additionalVolumeMounts }}
           volumeMounts:
               {{- toYaml . | nindent 12 }}
             {{- end }}
@@ -118,7 +121,7 @@ spec:
               {{- end }}
           env:
               {{- include "temporal.admintools-env" (list $ $store) | nindent 12 }}
-              {{- with $.Values.server.additionalVolumeMounts }}
+              {{- with $.Values.admintools.additionalVolumeMounts }}
           volumeMounts:
                 {{- toYaml . | nindent 12 }}
               {{- end }}
@@ -143,7 +146,7 @@ spec:
           env:
             - name: TEMPORAL_ADDRESS
               value: "{{ include "temporal.fullname" $ }}-frontend.{{ $.Release.Namespace }}.svc:{{ $.Values.server.frontend.service.port }}"
-              {{- with $.Values.server.additionalVolumeMounts }}
+              {{- with $.Values.admintools.additionalVolumeMounts }}
           volumeMounts:
                 {{- toYaml . | nindent 12 }}
               {{- end }}
@@ -178,19 +181,15 @@ spec:
       imagePullSecrets:
         {{- toYaml . | nindent 8 }}
       {{- end }}
-      {{- with $.Values.server.nodeSelector }}
+      {{- with $.Values.admintools.nodeSelector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}
       {{- end }}
-      {{- with $.Values.server.affinity }}
-      affinity:
-        {{- toYaml . | nindent 8 }}
-      {{- end }}
-      {{- with $.Values.server.tolerations }}
+      {{- with $.Values.admintools.tolerations }}
       tolerations:
         {{- toYaml . | nindent 8 }}
       {{- end }}
-      {{- with $.Values.server.additionalVolumes }}
+      {{- with $.Values.admintools.additionalVolumes }}
       volumes:
         {{- toYaml . | nindent 8 }}
       {{- end }}

diff --git a/charts/temporal/tests/admintools_deployment_test.yaml b/charts/temporal/tests/admintools_deployment_test.yaml
@@ -0,0 +1,31 @@
+suite: test admintools deployment
+templates:
+  - admintools-deployment.yaml
+tests:
+  - it: includes additional init containers
+    set:
+      admintools:
+        additionalInitContainers:
+          - name: my-init-container
+    asserts:
+      - equal:
+          path: spec.template.spec.initContainers[0].name
+          value: my-init-container
+  - it: includes additional volumes
+    set:
+      admintools:
+        additionalVolumes:
+          - name: my-volume
+    asserts:
+      - equal:
+          path: spec.template.spec.volumes[0].name
+          value: my-volume
+  - it: includes additional volume mounts
+    set:
+      admintools:
+        additionalVolumeMounts:
+          - name: my-volume
+    asserts:
+      - equal:
+          path: spec.template.spec.containers[0].volumeMounts[0].name
+          value: my-volume
diff --git a/charts/temporal/tests/server_deployment_test.yaml b/charts/temporal/tests/server_deployment_test.yaml
@@ -52,3 +52,17 @@ tests:
       - equal:
           path: spec.template.spec.containers[0].resources.requests.cpu
           value: 200m
+  - it: includes additional init containers
+    template: templates/server-deployment.yaml
+    documentSelector:
+      path: .kind
+      value: Deployment
+      matchMany: true
+    set:
+      server:
+        additionalInitContainers:
+          - name: my-init-container
+    asserts:
+      - equal:
+          path: spec.template.spec.initContainers[0].name
+          value: my-init-container
diff --git a/charts/temporal/tests/server_job_test.yaml b/charts/temporal/tests/server_job_test.yaml
@@ -0,0 +1,31 @@
+suite: test server job
+templates:
+  - server-job.yaml
+tests:
+  - it: includes additional init containers
+    set:
+      admintools:
+        additionalInitContainers:
+          - name: my-init-container
+    asserts:
+      - equal:
+          path: spec.template.spec.initContainers[0].name
+          value: my-init-container
+  - it: includes additional volumes
+    set:
+      admintools:
+        additionalVolumes:
+          - name: my-volume
+    asserts:
+      - equal:
+          path: spec.template.spec.volumes[0].name
+          value: my-volume
+  - it: includes additional volume mounts
+    set:
+      admintools:
+        additionalVolumeMounts:
+          - name: my-volume
+    asserts:
+      - equal:
+          path: spec.template.spec.initContainers[*].volumeMounts[0].name
+          value: my-volume