zot/pkg/test/mocks/oci_mock.go

package mocks

import (
	"time"

	v1 "github.com/google/go-containerregistry/pkg/v1"
	godigest "github.com/opencontainers/go-digest"
	ispec "github.com/opencontainers/image-spec/specs-go/v1"
	"zotregistry.io/zot/pkg/extensions/search/common"
)

type OciLayoutUtilsMock struct {
	GetImageManifestsFn         func(image string) ([]ispec.Descriptor, error)
	GetImageBlobManifestFn      func(imageDir string, digest godigest.Digest) (v1.Manifest, error)
	GetImageInfoFn              func(imageDir string, hash v1.Hash) (ispec.Image, error)
	IsValidImageFormatFn        func(image string) (bool, error)
	GetImageTagsWithTimestampFn func(repo string) ([]common.TagInfo, error)
	GetImageLastUpdatedFn       func(repo string, manifestDigest godigest.Digest) time.Time
	GetImagePlatformFn          func(repo string, manifestDigest godigest.Digest) (string, string)
	GetImageVendorFn            func(repo string, manifestDigest godigest.Digest) string
	GetImageManifestSizeFn      func(repo string, manifestDigest godigest.Digest) int64
	GetImageConfigSizeFn        func(repo string, manifestDigest godigest.Digest) int64
	GetRepoLastUpdatedFn        func(repo string) (time.Time, error)
	GetExpandedRepoInfoFn       func(name string) (common.RepoInfo, error)
}

func (olum OciLayoutUtilsMock) GetImageManifests(image string) ([]ispec.Descriptor, error) {
	if olum.GetImageBlobManifestFn != nil {
		return olum.GetImageManifestsFn(image)
	}

	return []ispec.Descriptor{}, nil
}

func (olum OciLayoutUtilsMock) GetImageBlobManifest(imageDir string, digest godigest.Digest) (v1.Manifest, error) {
	if olum.GetImageBlobManifestFn != nil {
		return olum.GetImageBlobManifestFn(imageDir, digest)
	}

	return v1.Manifest{}, nil
}

func (olum OciLayoutUtilsMock) GetImageInfo(imageDir string, hash v1.Hash) (ispec.Image, error) {
	if olum.GetImageInfoFn != nil {
		return olum.GetImageInfoFn(imageDir, hash)
	}

	return ispec.Image{}, nil
}

func (olum OciLayoutUtilsMock) IsValidImageFormat(image string) (bool, error) {
	if olum.IsValidImageFormatFn != nil {
		return olum.IsValidImageFormatFn(image)
	}

	return true, nil
}

func (olum OciLayoutUtilsMock) GetImageTagsWithTimestamp(repo string) ([]common.TagInfo, error) {
	if olum.GetImageTagsWithTimestampFn != nil {
		return olum.GetImageTagsWithTimestampFn(repo)
	}

	return []common.TagInfo{}, nil
}

func (olum OciLayoutUtilsMock) GetImageLastUpdated(repo string, manifestDigest godigest.Digest) time.Time {
	if olum.GetImageLastUpdatedFn != nil {
		return olum.GetImageLastUpdatedFn(repo, manifestDigest)
	}

	return time.Time{}
}

func (olum OciLayoutUtilsMock) GetImagePlatform(repo string, manifestDigest godigest.Digest) (string, string) {
	if olum.GetImagePlatformFn != nil {
		return olum.GetImagePlatformFn(repo, manifestDigest)
	}

	return "", ""
}

func (olum OciLayoutUtilsMock) GetImageVendor(repo string, manifestDigest godigest.Digest) string {
	if olum.GetImageVendorFn != nil {
		return olum.GetImageVendorFn(repo, manifestDigest)
	}

	return ""
}

func (olum OciLayoutUtilsMock) GetImageManifestSize(repo string, manifestDigest godigest.Digest) int64 {
	if olum.GetImageManifestSizeFn != nil {
		return olum.GetImageManifestSizeFn(repo, manifestDigest)
	}

	return 0
}

func (olum OciLayoutUtilsMock) GetImageConfigSize(repo string, manifestDigest godigest.Digest) int64 {
	if olum.GetImageConfigSizeFn != nil {
		return olum.GetImageConfigSizeFn(repo, manifestDigest)
	}

	return 0
}

func (olum OciLayoutUtilsMock) GetRepoLastUpdated(repo string) (time.Time, error) {
	if olum.GetRepoLastUpdatedFn != nil {
		return olum.GetRepoLastUpdatedFn(repo)
	}

	return time.Time{}, nil
}

func (olum OciLayoutUtilsMock) GetExpandedRepoInfo(name string) (common.RepoInfo, error) {
	if olum.GetExpandedRepoInfoFn != nil {
		return olum.GetExpandedRepoInfoFn(name)
	}

	return common.RepoInfo{}, nil
}