From 6258e87b072a3f35e3b35c53755b2e8b32c41387 Mon Sep 17 00:00:00 2001
From: Will Kutler <wkutler@redhat.com>
Date: Thu, 12 Jan 2023 14:51:49 -0500
Subject: [PATCH] add tests for evaluation metrics

Signed-off-by: Will Kutler <wkutler@redhat.com>
---
 test/e2e/case28_evauluation_metric_test.go    | 122 ++++++++++++++++++
 .../case28_policy.yaml                        |  18 +++
 2 files changed, 140 insertions(+)
 create mode 100644 test/e2e/case28_evauluation_metric_test.go
 create mode 100644 test/resources/case28_evaluation_metric/case28_policy.yaml

diff --git a/test/e2e/case28_evauluation_metric_test.go b/test/e2e/case28_evauluation_metric_test.go
new file mode 100644
index 00000000..db36086f
--- /dev/null
+++ b/test/e2e/case28_evauluation_metric_test.go
@@ -0,0 +1,122 @@
+package e2e
+
+import (
+	"fmt"
+	"os/exec"
+	"strconv"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+
+	"open-cluster-management.io/config-policy-controller/test/utils"
+)
+
+var _ = Describe("Test config policy evaluation metrics", Ordered, func() {
+	const (
+		policyName = "policy-cfgmap-watch"
+		policyYaml = "../resources/case28_evaluation_metric/case28_policy.yaml"
+	)
+
+	prePolicyEvalDuration := -1
+
+	It("should record number of ", func() {
+		evalMetric := utils.GetMetrics(
+			"config_policies_evaluation_duration_seconds_bucket", fmt.Sprintf(`le=\"%d\"`, 30))
+		Expect(len(evalMetric) != 0)
+		numEvals, err := strconv.Atoi(evalMetric[0])
+		Expect(err == nil)
+		prePolicyEvalDuration = numEvals
+		Expect(prePolicyEvalDuration > -1)
+	})
+
+	It("should create policy", func() {
+		By("Creating " + policyYaml)
+		utils.Kubectl("apply",
+			"-f", policyYaml,
+			"-n", testNamespace)
+		By("Verifying the policies were created")
+		plc := utils.GetWithTimeout(
+			clientManagedDynamic, gvrConfigPolicy, policyName, testNamespace, true, defaultTimeoutSeconds,
+		)
+		Expect(plc).NotTo(BeNil())
+	})
+
+	It("should update evaluation duration after the configurationpolicy is created", func() {
+		By("Checking metric bucket for evaluation duration (30 seconds or less)")
+		Eventually(func() interface{} {
+			metric := utils.GetMetrics(
+				"config_policies_evaluation_duration_seconds_bucket", fmt.Sprintf(`le=\"%d\"`, 30))
+			if len(metric) == 0 {
+				return false
+			}
+			numEvals, err := strconv.Atoi(metric[0])
+			if err != nil {
+				return false
+			}
+
+			return numEvals > prePolicyEvalDuration
+		}, defaultTimeoutSeconds, 1).Should(Equal(true))
+	})
+
+	It("should correctly report total evaluations for the configurationpolicy", func() {
+		By("Checking metric endpoint for total evaluations")
+		Eventually(func() interface{} {
+			metric := utils.GetMetrics(
+				"config_policy_evaluation_total", fmt.Sprintf(`name=\"%s\"`, policyName))
+			if len(metric) == 0 {
+				return false
+			}
+			numEvals, err := strconv.Atoi(metric[0])
+			if err != nil {
+				return false
+			}
+
+			return numEvals > 0
+		}, defaultTimeoutSeconds, 1).Should(Equal(true))
+	})
+
+	It("should report total evaluation duration for the configurationpolicy", func() {
+		By("Checking metric endpoint for total evaluation duration")
+		Eventually(func() interface{} {
+			metric := utils.GetMetrics(
+				"config_policy_evaluation_seconds_total", fmt.Sprintf(`name=\"%s\"`, policyName))
+			if len(metric) == 0 {
+				return false
+			}
+			numEvals, err := strconv.ParseFloat(metric[0], 64)
+			if err != nil {
+				return false
+			}
+
+			return numEvals > 0
+		}, defaultTimeoutSeconds, 1).Should(Equal(true))
+	})
+
+	cleanup := func() {
+		// Delete the policies and ignore any errors (in case it was deleted previously)
+		cmd := exec.Command("kubectl", "delete",
+			"-f", policyYaml,
+			"-n", testNamespace)
+		_, _ = cmd.CombinedOutput()
+		opt := metav1.ListOptions{}
+		utils.ListWithTimeout(
+			clientManagedDynamic, gvrConfigPolicy, opt, 0, false, defaultTimeoutSeconds)
+	}
+
+	It("should clean up", cleanup)
+
+	It("should have no evaluation metrics for deleted policies after clean up", func() {
+		By("Checking evaluation metrics for deleted config policy")
+		Eventually(func() interface{} {
+			return utils.GetMetrics(
+				"config_policy_evaluation_total", fmt.Sprintf(`name=\"%s\"`, policyName))
+		}, defaultTimeoutSeconds, 1).Should(Equal([]string{}))
+		Eventually(func() interface{} {
+			return utils.GetMetrics(
+				"config_policy_evaluation_seconds_total", fmt.Sprintf(`name=\"%s\"`, policyName))
+		}, defaultTimeoutSeconds, 1).Should(Equal([]string{}))
+	})
+
+	AfterAll(cleanup)
+})
diff --git a/test/resources/case28_evaluation_metric/case28_policy.yaml b/test/resources/case28_evaluation_metric/case28_policy.yaml
new file mode 100644
index 00000000..5dc84101
--- /dev/null
+++ b/test/resources/case28_evaluation_metric/case28_policy.yaml
@@ -0,0 +1,18 @@
+apiVersion: policy.open-cluster-management.io/v1
+kind: ConfigurationPolicy
+metadata:
+  name: policy-cfgmap-watch
+spec:
+  remediationAction: inform
+  namespaceSelector:
+    exclude: ["kube-*"]
+    include: ["default"]
+  object-templates:
+    - complianceType: musthave
+      objectDefinition:
+        apiVersion: v1
+        kind: ConfigMap
+        metadata:
+          name: case28-map
+        data:
+          fieldToUpdate: "1"