package k8s

import (
	"testing"

	"github.com/go-test/deep"
	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",
				Namespace: "test-ns",
				Labels: map[string]string{
					ControllerNSLabel:                      "linkerd-namespace",
					appsv1.DefaultDeploymentUniqueLabelKey: "test-pth",
				},
			},
			Spec: corev1.PodSpec{
				ServiceAccountName: "test-sa",
			},
		}

		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",
			"serviceaccount":    "test-sa",
		}

		podLabels := GetPodLabels(ownerKind, ownerName, pod)

		if diff := deep.Equal(podLabels, expectedLabels); diff != nil {
			t.Errorf("labels %+v", diff)
		}
	})
}