freeform querry api

Signed-off-by: Laurentiu Niculae <themelopeus@gmail.com>

pkg/test/mocks/oci_mock.go

@@ -0,0 +1,121 @@
+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
+}