mirror of https://github.com/linkerd/linkerd2.git
41 lines
976 B
Go
41 lines
976 B
Go
package k8s
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
appsV1 "k8s.io/api/apps/v1"
|
|
coreV1 "k8s.io/api/core/v1"
|
|
metaV1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
)
|
|
|
|
func TestGetPodLabels(t *testing.T) {
|
|
t.Run("Maps proxy labels to prometheus labels", func(t *testing.T) {
|
|
pod := &coreV1.Pod{
|
|
ObjectMeta: metaV1.ObjectMeta{
|
|
Name: "test-pod",
|
|
Labels: map[string]string{
|
|
ControllerNSLabel: "linkerd-namespace",
|
|
appsV1.DefaultDeploymentUniqueLabelKey: "test-pth",
|
|
},
|
|
},
|
|
}
|
|
|
|
ownerKind := "deployment"
|
|
ownerName := "test-deployment"
|
|
|
|
expectedLabels := map[string]string{
|
|
"control_plane_ns": "linkerd-namespace",
|
|
"deployment": "test-deployment",
|
|
"pod": "test-pod",
|
|
"pod_template_hash": "test-pth",
|
|
}
|
|
|
|
podLabels := GetPodLabels(ownerKind, ownerName, pod)
|
|
|
|
if !reflect.DeepEqual(podLabels, expectedLabels) {
|
|
t.Fatalf("Expected pod labels [%v] but got [%v]", expectedLabels, podLabels)
|
|
}
|
|
})
|
|
}
|