feat: support setting connect inject deployment replicas (#1029)

hashicorp · Jul 13, 2021 · 9b4dba0 · 9b4dba0
1 parent 83b32de
commit 9b4dba0
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 1 deletion.
diff --git a/templates/connect-inject-deployment.yaml b/templates/connect-inject-deployment.yaml
@@ -19,7 +19,7 @@ metadata:
     heritage: {{ .Release.Service }}
     release: {{ .Release.Name }}
 spec:
-  replicas: 2
+  replicas: {{ .Values.connectInject.replicas }}
   selector:
     matchLabels:
       app: {{ template "consul.name" . }}

diff --git a/test/unit/connect-inject-deployment.bats b/test/unit/connect-inject-deployment.bats
@@ -1446,4 +1446,31 @@ EOF
       yq '.spec.template.spec.containers[0].command | any(contains("-enable-openshift"))' | tee /dev/stderr)
 
   [ "${actual}" = "true" ]
+}
+
+
+#--------------------------------------------------------------------
+# replicas
+
+@test "connectInject/Deployment: replicas defaults to 2" {
+  cd `chart_dir`
+  local actual=$(helm template \
+      -s templates/connect-inject-deployment.yaml  \
+      --set 'connectInject.enabled=true' \
+      . | tee /dev/stderr |
+      yq '.spec.replicas' | tee /dev/stderr)
+
+  [ "${actual}" = "2" ]
+}
+
+@test "connectInject/Deployment: replicas can be set" {
+  cd `chart_dir`
+  local actual=$(helm template \
+      -s templates/connect-inject-deployment.yaml  \
+      --set 'connectInject.enabled=true' \
+      --set 'connectInject.replicas=3' \
+      . | tee /dev/stderr |
+      yq '.spec.replicas' | tee /dev/stderr)
+
+  [ "${actual}" = "3" ]
 }
diff --git a/values.yaml b/values.yaml
@@ -1376,6 +1376,9 @@ connectInject:
   # global.enabled.
   enabled: false
 
+  # The number of deployment replicas.
+  replicas: 2
+
   # Image for consul-k8s that contains the injector
   # @type: string
   image: null