metadata_test.go

package lifecycle_test

import (
	"testing"

	"github.com/sclevine/spec"

	"github.com/buildpacks/lifecycle"
	h "github.com/buildpacks/lifecycle/testhelpers"
)

func TestStackMetadata(t *testing.T) {
	spec.Run(t, "Test StackMetadata", testMetadata)
}

func testMetadata(t *testing.T, when spec.G, it spec.S) {
	when("BestRunImageMirror", func() {
		var stackMD *lifecycle.StackMetadata

		it.Before(func() {
			stackMD = &lifecycle.StackMetadata{RunImage: lifecycle.StackRunImageMetadata{
				Image: "first.com/org/repo",
				Mirrors: []string{
					"myorg/myrepo",
					"zonal.gcr.io/org/repo",
					"gcr.io/org/repo",
				},
			}}
		})

		when("repoName is dockerhub", func() {
			it("returns the dockerhub image", func() {
				name, err := stackMD.BestRunImageMirror("index.docker.io")
				h.AssertNil(t, err)
				h.AssertEq(t, name, "myorg/myrepo")
			})
		})

		when("registry is gcr.io", func() {
			it("returns the gcr.io image", func() {
				name, err := stackMD.BestRunImageMirror("gcr.io")
				h.AssertNil(t, err)
				h.AssertEq(t, name, "gcr.io/org/repo")
			})

			when("registry is zonal.gcr.io", func() {
				it("returns the gcr image", func() {
					name, err := stackMD.BestRunImageMirror("zonal.gcr.io")
					h.AssertNil(t, err)
					h.AssertEq(t, name, "zonal.gcr.io/org/repo")
				})
			})

			when("registry is missingzone.gcr.io", func() {
				it("returns the run image", func() {
					name, err := stackMD.BestRunImageMirror("missingzone.gcr.io")
					h.AssertNil(t, err)
					h.AssertEq(t, name, "first.com/org/repo")
				})
			})
		})

		when("one of the images is non-parsable", func() {
			it.Before(func() {
				stackMD.RunImage.Mirrors = []string{"as@ohd@as@op", "gcr.io/myorg/myrepo"}
			})

			it("skips over it", func() {
				name, err := stackMD.BestRunImageMirror("gcr.io")
				h.AssertNil(t, err)
				h.AssertEq(t, name, "gcr.io/myorg/myrepo")
			})
		})
	})
}