package update

import (
	"testing"

	"github.com/google/go-containerregistry/pkg/name"
	. "github.com/onsi/gomega"
	"k8s.io/apimachinery/pkg/types"
	"sigs.k8s.io/kustomize/kyaml/yaml"
)

// mustRef creates an imageRef for use in tests. It panics if the ref
// given is invalid.
func mustRef(ref string) imageRef {
	r, err := name.ParseReference(ref)
	if err != nil {
		panic(err)
	}
	return imageRef{r, types.NamespacedName{}}
}

func TestMustRef(t *testing.T) {
	g := NewWithT(t)

	t.Run("gives each component of an image ref", func(t *testing.T) {
		ref := mustRef("helloworld:v1.0.1")
		g.Expect(ref.String()).To(Equal("helloworld:v1.0.1"))
		g.Expect(ref.Identifier()).To(Equal("v1.0.1"))
		g.Expect(ref.Repository()).To(Equal("library/helloworld"))
		g.Expect(ref.Registry()).To(Equal("index.docker.io"))
		g.Expect(ref.Name()).To(Equal("index.docker.io/library/helloworld:v1.0.1"))
	})

	t.Run("deals with hostnames and digests", func(t *testing.T) {
		image := "localhost:5000/org/helloworld@sha256:6745aaad46d795c9836632e1fb62f24b7e7f4c843144da8e47a5465c411a14be"
		ref := mustRef(image)
		g.Expect(ref.String()).To(Equal(image))
		g.Expect(ref.Identifier()).To(Equal("sha256:6745aaad46d795c9836632e1fb62f24b7e7f4c843144da8e47a5465c411a14be"))
		g.Expect(ref.Repository()).To(Equal("org/helloworld"))
		g.Expect(ref.Registry()).To(Equal("localhost:5000"))
		g.Expect(ref.Name()).To(Equal(image))
	})
}

func TestUpdateResults(t *testing.T) {
	g := NewWithT(t)

	var result Result
	objectNames := []ObjectIdentifier{
		{yaml.ResourceIdentifier{
			NameMeta: yaml.NameMeta{Namespace: "ns", Name: "foo"},
		}},
		{yaml.ResourceIdentifier{
			NameMeta: yaml.NameMeta{Namespace: "ns", Name: "bar"},
		}},
	}

	result = Result{
		Files: map[string]FileResult{
			"foo.yaml": {
				Objects: map[ObjectIdentifier][]ImageRef{
					objectNames[0]: {
						mustRef("image:v1.0"),
						mustRef("other:v2.0"),
					},
				},
			},
			"bar.yaml": {
				Objects: map[ObjectIdentifier][]ImageRef{
					objectNames[1]: {
						mustRef("image:v1.0"),
						mustRef("other:v2.0"),
					},
				},
			},
		},
	}

	g.Expect(result.Images()).To(Equal([]ImageRef{
		mustRef("image:v1.0"),
		mustRef("other:v2.0"),
	}))

	g.Expect(result.Objects()).To(Equal(map[ObjectIdentifier][]ImageRef{
		objectNames[0]: {
			mustRef("image:v1.0"),
			mustRef("other:v2.0"),
		},
		objectNames[1]: {
			mustRef("image:v1.0"),
			mustRef("other:v2.0"),
		},
	}))
}