Function WithTestImageConfig

internal/utils/oci/fake/client.go:38–78

(ctx context.Context, url string)

)

func WithTestImageConfig(ctx context.Context, url string) context.Context {
	ref, err := name.NewDigest(url)
	if err != nil {
		panic(err)
	}

	// Setup parent/base image mock
	parentURL := utils.WithDigest("registry.local/base-image")
	parentRef, err := name.ParseReference(parentURL)
	if err != nil {
		panic(err)
	}
	parentImage, err := mutate.Config(empty.Image, v1.Config{
		Labels: map[string]string{
			"io.k8s.display-name": "Base Image",
		},
	})
	if err != nil {
		panic(err)
	}

	// Setup child image mock
	image := mutate.Annotations(empty.Image, map[string]string{
		oci.BaseImageNameAnnotation: parentURL,
	}).(v1.Image)
	image, err = mutate.Config(image, v1.Config{
		Labels: map[string]string{
			"io.k8s.display-name": "Test Image",
		},
	})
	if err != nil {
		panic(err)
	}

	// Setup client
	client := &FakeClient{}
	client.On("Image", ref, mock.Anything).Return(image, nil)
	client.On("Image", parentRef, mock.Anything).Return(parentImage, nil)

	return oci.WithClient(ctx, client)
}

var _ oci.Client = &FakeClient{}