add to deployments

Author: w0ng952

pkg/templates/deployment.go

@@ -31,6 +31,7 @@ type templateParameterDeployment struct {
 	Criticality         string
 	Sensitivity         string
 	Deployment          *apps.Deployment
+	Priority            string
 }
 
 // CreateFromDeployment
@@ -74,6 +75,8 @@ func (a *PrometheusRuleTemplateManager) CreateFromDeployment(deployment *apps.De
 			continue
 		}
 
+		params.Priority = deployment.GetAnnotations()[priorityAnnotation]
+
 		templateName := strings.TrimLeft(k, fmt.Sprintf("%s/", heimPrefix))
 		logger.Infow("template selected", "template", templateName)
 		template, ok := a.templates[templateName]

pkg/templates/deployment_test.go

@@ -24,6 +24,7 @@ var (
 				environmentAnnotation: "testing",
 				criticalityAnnotation: "low",
 				sensitivityAnnotation: "public",
+				priorityAnnotation:    "p1",
 				"com.uswitch.heimdall/replicas-availability-deployment": "1",
 			},
 			OwnerReferences: []metav1.OwnerReference{},
@@ -51,4 +52,8 @@ kube_deployment_spec_replicas{namespace="testNamespace", deployment="testApp"} <
 	assert.Assert(t, is.Len(promrules, 1))
 	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Expr.StrVal, expr)
 	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["owner"], "testDeploymentOwner")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["environment"], "testing")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["criticality"], "low")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["sensitivity"], "public")
+	assert.Equal(t, promrules[0].Spec.Groups[0].Rules[0].Labels["priority"], "p1")
 }