package injector

import (
	"fmt"
	"reflect"
	"testing"

	"github.com/linkerd/linkerd2/controller/proxy-injector/fake"
	k8sPkg "github.com/linkerd/linkerd2/pkg/k8s"
	"k8s.io/api/core/v1"
)

func TestPatch(t *testing.T) {
	fixture := fake.NewFactory()

	trustAnchors, err := fixture.Volume("inject-trust-anchors-volume-spec.yaml")
	if err != nil {
		t.Fatal("Unexpected error: ", err)
	}

	secrets, err := fixture.Volume("inject-linkerd-secrets-volume-spec.yaml")
	if err != nil {
		t.Fatal("Unexpected error: ", err)
	}

	sidecar, err := fixture.Container("inject-sidecar-container-spec.yaml")
	if err != nil {
		t.Fatal("Unexpected error: ", err)
	}

	init, err := fixture.Container("inject-init-container-spec.yaml")
	if err != nil {
		t.Fatal("Unexpected error: ", err)
	}

	var (
		controllerNamespace = "linkerd"
		createdBy           = "linkerd/cli v18.8.4"
		initContainerIndex  = 5
	)

	actual := NewPatch()
	actual.addContainer(sidecar)
	actual.addInitContainerRoot()
	actual.addInitContainer(init, initContainerIndex)
	actual.addVolumeRoot()
	actual.addVolume(trustAnchors)
	actual.addVolume(secrets)
	actual.addPodLabels(map[string]string{
		k8sPkg.ControllerNSLabel:    controllerNamespace,
		k8sPkg.ProxyAutoInjectLabel: k8sPkg.ProxyAutoInjectCompleted,
	})
	actual.addDeploymentLabels(map[string]string{
		k8sPkg.ControllerNSLabel:    controllerNamespace,
		k8sPkg.ProxyAutoInjectLabel: k8sPkg.ProxyAutoInjectCompleted,
	})
	actual.addPodAnnotations(map[string]string{
		k8sPkg.CreatedByAnnotation: createdBy,
	})

	expected := NewPatch()
	expected.patchOps = []*patchOp{
		&patchOp{Op: "add", Path: patchPathContainer, Value: sidecar},
		&patchOp{Op: "add", Path: patchPathInitContainerRoot, Value: []*v1.Container{}},
		&patchOp{Op: "add", Path: fmt.Sprintf(patchPathInitContainer, initContainerIndex), Value: init},
		&patchOp{Op: "add", Path: patchPathVolumeRoot, Value: []*v1.Volume{}},
		&patchOp{Op: "add", Path: patchPathVolume, Value: trustAnchors},
		&patchOp{Op: "add", Path: patchPathVolume, Value: secrets},
		&patchOp{Op: "add", Path: patchPathPodLabels, Value: map[string]string{
			k8sPkg.ControllerNSLabel:    controllerNamespace,
			k8sPkg.ProxyAutoInjectLabel: k8sPkg.ProxyAutoInjectCompleted,
		}},
		&patchOp{Op: "add", Path: patchPathDeploymentLabels, Value: map[string]string{
			k8sPkg.ControllerNSLabel:    controllerNamespace,
			k8sPkg.ProxyAutoInjectLabel: k8sPkg.ProxyAutoInjectCompleted,
		}},
		&patchOp{Op: "add", Path: patchPathPodAnnotations, Value: map[string]string{k8sPkg.CreatedByAnnotation: createdBy}},
	}

	if !reflect.DeepEqual(actual, expected) {
		t.Errorf("Content mismatch\nExpected: %s\nActual: %s", expected, actual)
	}
}