From bc26a58044a34f6da53bdbd53670ca77ec2fb110 Mon Sep 17 00:00:00 2001
From: PoAn Yang <payang@apache.org>
Date: Thu, 23 Nov 2023 00:31:31 +0800
Subject: [PATCH] [YUNIKORN-2135] Add integration test code coverage

Signed-off-by: PoAn Yang <payang@apache.org>
---
 helm-charts/yunikorn/templates/deployment.yaml | 14 ++++++++++++++
 helm-charts/yunikorn/values.yaml               |  2 ++
 2 files changed, 16 insertions(+)

diff --git a/helm-charts/yunikorn/templates/deployment.yaml b/helm-charts/yunikorn/templates/deployment.yaml
index 399ba12..e367809 100644
--- a/helm-charts/yunikorn/templates/deployment.yaml
+++ b/helm-charts/yunikorn/templates/deployment.yaml
@@ -93,6 +93,13 @@ spec:
               valueFrom:
                 fieldRef:
                   fieldPath: metadata.namespace
+          {{- if .Values.enableGoCoverDir }}
+            - name: GOCOVERDIR
+              value: /go-cover-dir
+          volumeMounts:
+            - name: go-cover-dir
+              mountPath: /go-cover-dir/
+          {{- end }}
           resources:
             {{- toYaml .Values.resources | nindent 12 }}
         - name: yunikorn-scheduler-web
@@ -104,3 +111,10 @@ spec:
               protocol: TCP
           resources:
             {{- toYaml .Values.web.resources | nindent 12 }}
+      {{- if .Values.enableGoCoverDir }}
+      volumes:
+        - name: go-cover-dir
+          hostPath:
+            path: /go-cover-dir/
+            type: DirectoryOrCreate
+      {{- end }}
diff --git a/helm-charts/yunikorn/values.yaml b/helm-charts/yunikorn/values.yaml
index 3b6f149..e23fff4 100644
--- a/helm-charts/yunikorn/values.yaml
+++ b/helm-charts/yunikorn/values.yaml
@@ -124,3 +124,5 @@ enableSchedulerPlugin: false
 
 podLabels: {}
 podAnnotations: {}
+
+enableGoCoverDir: false