2023-01-09 15:37:44 -05:00
|
|
|
package repodb_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"encoding/json"
|
|
|
|
"errors"
|
|
|
|
"fmt"
|
|
|
|
"io"
|
|
|
|
"os"
|
|
|
|
"path"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
godigest "github.com/opencontainers/go-digest"
|
|
|
|
ispec "github.com/opencontainers/image-spec/specs-go/v1"
|
|
|
|
. "github.com/smartystreets/goconvey/convey"
|
|
|
|
|
|
|
|
zerr "zotregistry.io/zot/errors"
|
|
|
|
"zotregistry.io/zot/pkg/extensions/monitoring"
|
|
|
|
"zotregistry.io/zot/pkg/log"
|
2023-03-28 12:20:09 -05:00
|
|
|
"zotregistry.io/zot/pkg/meta/bolt"
|
|
|
|
"zotregistry.io/zot/pkg/meta/dynamo"
|
2023-01-09 15:37:44 -05:00
|
|
|
"zotregistry.io/zot/pkg/meta/repodb"
|
2023-03-28 12:20:09 -05:00
|
|
|
bolt_wrapper "zotregistry.io/zot/pkg/meta/repodb/boltdb-wrapper"
|
|
|
|
dynamo_wrapper "zotregistry.io/zot/pkg/meta/repodb/dynamodb-wrapper"
|
2023-01-09 15:37:44 -05:00
|
|
|
"zotregistry.io/zot/pkg/storage"
|
|
|
|
"zotregistry.io/zot/pkg/storage/local"
|
|
|
|
"zotregistry.io/zot/pkg/test"
|
|
|
|
"zotregistry.io/zot/pkg/test/mocks"
|
|
|
|
)
|
|
|
|
|
|
|
|
const repo = "repo"
|
|
|
|
|
|
|
|
var ErrTestError = errors.New("test error")
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
func TestParseStorageErrors(t *testing.T) {
|
2023-03-28 12:20:09 -05:00
|
|
|
Convey("ParseStorage", t, func() {
|
2023-01-09 15:37:44 -05:00
|
|
|
imageStore := mocks.MockedImageStore{
|
|
|
|
GetIndexContentFn: func(repo string) ([]byte, error) {
|
|
|
|
return nil, ErrTestError
|
|
|
|
},
|
|
|
|
GetRepositoriesFn: func() ([]string, error) {
|
|
|
|
return []string{"repo1", "repo2"}, nil
|
|
|
|
},
|
|
|
|
}
|
|
|
|
storeController := storage.StoreController{DefaultStore: imageStore}
|
|
|
|
repoDB := mocks.RepoDBMock{}
|
|
|
|
|
|
|
|
// sync repo fail
|
2023-03-10 13:37:29 -05:00
|
|
|
err := repodb.ParseStorage(repoDB, storeController, log.NewLogger("debug", ""))
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
|
|
|
|
Convey("getAllRepos errors", func() {
|
|
|
|
imageStore1 := mocks.MockedImageStore{
|
|
|
|
GetRepositoriesFn: func() ([]string, error) {
|
|
|
|
return []string{"repo1", "repo2"}, nil
|
|
|
|
},
|
|
|
|
}
|
|
|
|
imageStore2 := mocks.MockedImageStore{
|
|
|
|
GetRepositoriesFn: func() ([]string, error) {
|
|
|
|
return nil, ErrTestError
|
|
|
|
},
|
|
|
|
}
|
|
|
|
storeController := storage.StoreController{
|
|
|
|
DefaultStore: imageStore1,
|
|
|
|
SubStore: map[string]storage.ImageStore{
|
|
|
|
"a": imageStore2,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err := repodb.ParseStorage(repoDB, storeController, log.NewLogger("debug", ""))
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
})
|
|
|
|
|
2023-03-28 12:20:09 -05:00
|
|
|
Convey("Parse Repo", t, func() {
|
2023-01-09 15:37:44 -05:00
|
|
|
imageStore := mocks.MockedImageStore{}
|
|
|
|
storeController := storage.StoreController{DefaultStore: &imageStore}
|
|
|
|
repoDB := mocks.RepoDBMock{}
|
|
|
|
log := log.NewLogger("debug", "")
|
|
|
|
|
|
|
|
Convey("imageStore.GetIndexContent errors", func() {
|
|
|
|
imageStore.GetIndexContentFn = func(repo string) ([]byte, error) {
|
|
|
|
return nil, ErrTestError
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err := repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("json.Unmarshal errors", func() {
|
|
|
|
imageStore.GetIndexContentFn = func(repo string) ([]byte, error) {
|
|
|
|
return []byte("Invalid JSON"), nil
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err := repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("resetRepoMetaTags errors", func() {
|
|
|
|
imageStore.GetIndexContentFn = func(repo string) ([]byte, error) {
|
|
|
|
return []byte("{}"), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
Convey("repoDB.GetRepoMeta errors", func() {
|
|
|
|
repoDB.GetRepoMetaFn = func(repo string) (repodb.RepoMetadata, error) {
|
|
|
|
return repodb.RepoMetadata{}, ErrTestError
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err := repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("isManifestMetaPresent errors", func() {
|
|
|
|
indexContent := ispec.Index{
|
|
|
|
Manifests: []ispec.Descriptor{
|
|
|
|
{
|
|
|
|
Digest: godigest.FromString("manifest1"),
|
|
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
|
|
Annotations: map[string]string{
|
|
|
|
ispec.AnnotationRefName: "tag1",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
indexBlob, err := json.Marshal(indexContent)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
imageStore.GetIndexContentFn = func(repo string) ([]byte, error) {
|
|
|
|
return indexBlob, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
Convey("repoDB.GetManifestMeta errors", func() {
|
|
|
|
repoDB.GetManifestMetaFn = func(repo string, manifestDigest godigest.Digest) (repodb.ManifestMetadata, error) {
|
|
|
|
return repodb.ManifestMetadata{}, ErrTestError
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("manifestMetaIsPresent true", func() {
|
|
|
|
indexContent := ispec.Index{
|
|
|
|
Manifests: []ispec.Descriptor{
|
|
|
|
{
|
|
|
|
Digest: godigest.FromString("manifest1"),
|
|
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
|
|
Annotations: map[string]string{
|
|
|
|
ispec.AnnotationRefName: "tag1",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
indexBlob, err := json.Marshal(indexContent)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
imageStore.GetIndexContentFn = func(repo string) ([]byte, error) {
|
|
|
|
return indexBlob, nil
|
|
|
|
}
|
|
|
|
|
2023-03-09 13:41:48 -05:00
|
|
|
Convey("repoDB.SetRepoReference", func() {
|
|
|
|
repoDB.SetRepoReferenceFn = func(repo, tag string, manifestDigest godigest.Digest, mediaType string) error {
|
2023-01-09 15:37:44 -05:00
|
|
|
return ErrTestError
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("manifestMetaIsPresent false", func() {
|
|
|
|
indexContent := ispec.Index{
|
|
|
|
Manifests: []ispec.Descriptor{
|
|
|
|
{
|
|
|
|
Digest: godigest.FromString("manifest1"),
|
|
|
|
MediaType: ispec.MediaTypeImageManifest,
|
|
|
|
Annotations: map[string]string{
|
|
|
|
ispec.AnnotationRefName: "tag1",
|
|
|
|
},
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
indexBlob, err := json.Marshal(indexContent)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
imageStore.GetIndexContentFn = func(repo string) ([]byte, error) {
|
|
|
|
return indexBlob, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
repoDB.GetManifestMetaFn = func(repo string, manifestDigest godigest.Digest) (repodb.ManifestMetadata, error) {
|
|
|
|
return repodb.ManifestMetadata{}, zerr.ErrManifestMetaNotFound
|
|
|
|
}
|
|
|
|
|
|
|
|
Convey("GetImageManifest errors", func() {
|
|
|
|
imageStore.GetImageManifestFn = func(repo, reference string) ([]byte, godigest.Digest, string, error) {
|
|
|
|
return nil, "", "", ErrTestError
|
|
|
|
}
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("CheckIsImageSignature errors", func() {
|
|
|
|
// CheckIsImageSignature will fail because of a invalid json
|
|
|
|
imageStore.GetImageManifestFn = func(repo, reference string) ([]byte, godigest.Digest, string, error) {
|
|
|
|
return []byte("Invalid JSON"), "", "", nil
|
|
|
|
}
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
Convey("CheckIsImageSignature -> not signature", func() {
|
|
|
|
manifestContent := ispec.Manifest{}
|
|
|
|
manifestBlob, err := json.Marshal(manifestContent)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
imageStore.GetImageManifestFn = func(repo, reference string) ([]byte, godigest.Digest, string, error) {
|
|
|
|
return manifestBlob, "", "", nil
|
|
|
|
}
|
|
|
|
|
|
|
|
Convey("imgStore.GetBlobContent errors", func() {
|
|
|
|
imageStore.GetBlobContentFn = func(repo string, digest godigest.Digest) ([]byte, error) {
|
|
|
|
return nil, ErrTestError
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("json.Unmarshal(configBlob errors", func() {
|
|
|
|
imageStore.GetBlobContentFn = func(repo string, digest godigest.Digest) ([]byte, error) {
|
|
|
|
return []byte("invalid JSON"), nil
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
})
|
|
|
|
|
|
|
|
Convey("CheckIsImageSignature -> is signature", func() {
|
2023-05-10 12:15:33 -05:00
|
|
|
manifestContent := ispec.Manifest{
|
2023-03-10 13:37:29 -05:00
|
|
|
Subject: &ispec.Descriptor{
|
2023-01-09 15:37:44 -05:00
|
|
|
Digest: "123",
|
|
|
|
},
|
2023-03-10 13:37:29 -05:00
|
|
|
ArtifactType: "application/vnd.cncf.notary.signature",
|
2023-01-09 15:37:44 -05:00
|
|
|
}
|
2023-03-10 13:37:29 -05:00
|
|
|
|
2023-01-09 15:37:44 -05:00
|
|
|
manifestBlob, err := json.Marshal(manifestContent)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
imageStore.GetImageManifestFn = func(repo, reference string) ([]byte, godigest.Digest, string, error) {
|
|
|
|
return manifestBlob, "", "", nil
|
|
|
|
}
|
|
|
|
|
|
|
|
repoDB.AddManifestSignatureFn = func(repo string, signedManifestDigest godigest.Digest,
|
|
|
|
sm repodb.SignatureMetadata,
|
|
|
|
) error {
|
|
|
|
return ErrTestError
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseRepo("repo", repoDB, storeController, log)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
})
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2023-05-15 04:02:23 -05:00
|
|
|
func TestParseStorageWithBoltDB(t *testing.T) {
|
2023-01-09 15:37:44 -05:00
|
|
|
Convey("Boltdb", t, func() {
|
|
|
|
rootDir := t.TempDir()
|
|
|
|
|
2023-03-28 12:20:09 -05:00
|
|
|
boltDB, err := bolt.GetBoltDriver(bolt.DBParameters{
|
2023-03-23 13:08:11 -05:00
|
|
|
RootDir: rootDir,
|
|
|
|
})
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-28 12:20:09 -05:00
|
|
|
repoDB, err := bolt_wrapper.NewBoltDBWrapper(boltDB, log.NewLogger("debug", ""))
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
RunParseStorageTests(rootDir, repoDB)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestParseStorageDynamoWrapper(t *testing.T) {
|
|
|
|
skipIt(t)
|
|
|
|
|
|
|
|
Convey("Dynamodb", t, func() {
|
|
|
|
rootDir := t.TempDir()
|
|
|
|
|
2023-03-28 12:20:09 -05:00
|
|
|
params := dynamo.DBDriverParameters{
|
2023-03-23 13:08:11 -05:00
|
|
|
Endpoint: os.Getenv("DYNAMODBMOCK_ENDPOINT"),
|
|
|
|
Region: "us-east-2",
|
|
|
|
RepoMetaTablename: "RepoMetadataTable",
|
|
|
|
ManifestDataTablename: "ManifestDataTable",
|
|
|
|
IndexDataTablename: "IndexDataTable",
|
2023-04-24 13:13:15 -05:00
|
|
|
UserDataTablename: "UserDataTable",
|
2023-03-23 13:08:11 -05:00
|
|
|
VersionTablename: "Version",
|
2023-03-28 12:20:09 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
dynamoClient, err := dynamo.GetDynamoClient(params)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
dynamoWrapper, err := dynamo_wrapper.NewDynamoDBWrapper(dynamoClient, params, log.NewLogger("debug", ""))
|
2023-03-23 13:08:11 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
err = dynamoWrapper.ResetManifestDataTable()
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
err = dynamoWrapper.ResetRepoMetaTable()
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
RunParseStorageTests(rootDir, dynamoWrapper)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
func RunParseStorageTests(rootDir string, repoDB repodb.RepoDB) {
|
2023-05-15 04:02:23 -05:00
|
|
|
Convey("Test with simple case", func() {
|
2023-01-09 15:37:44 -05:00
|
|
|
imageStore := local.NewImageStore(rootDir, false, 0, false, false,
|
|
|
|
log.NewLogger("debug", ""), monitoring.NewMetricsServer(false, log.NewLogger("debug", "")), nil, nil)
|
|
|
|
|
|
|
|
storeController := storage.StoreController{DefaultStore: imageStore}
|
|
|
|
manifests := []ispec.Manifest{}
|
|
|
|
for i := 0; i < 3; i++ {
|
|
|
|
config, layers, manifest, err := test.GetRandomImageComponents(100)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
manifests = append(manifests, manifest)
|
|
|
|
|
|
|
|
err = test.WriteImageToFileSystem(
|
|
|
|
test.Image{
|
2023-02-27 14:23:18 -05:00
|
|
|
Config: config,
|
|
|
|
Layers: layers,
|
|
|
|
Manifest: manifest,
|
|
|
|
Reference: fmt.Sprintf("tag%d", i),
|
2023-01-09 15:37:44 -05:00
|
|
|
},
|
|
|
|
repo,
|
|
|
|
storeController)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
}
|
|
|
|
|
|
|
|
// add fake signature for tag1
|
|
|
|
signatureTag, err := test.GetCosignSignatureTagForManifest(manifests[1])
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
manifestBlob, err := json.Marshal(manifests[1])
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
signedManifestDigest := godigest.FromBytes(manifestBlob)
|
|
|
|
|
|
|
|
config, layers, manifest, err := test.GetRandomImageComponents(100)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
err = test.WriteImageToFileSystem(
|
|
|
|
test.Image{
|
2023-02-27 14:23:18 -05:00
|
|
|
Config: config,
|
|
|
|
Layers: layers,
|
|
|
|
Manifest: manifest,
|
|
|
|
Reference: signatureTag,
|
2023-01-09 15:37:44 -05:00
|
|
|
},
|
|
|
|
repo,
|
|
|
|
storeController)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
// remove tag2 from index.json
|
|
|
|
indexPath := path.Join(rootDir, repo, "index.json")
|
|
|
|
indexFile, err := os.Open(indexPath)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
buf, err := io.ReadAll(indexFile)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
var index ispec.Index
|
|
|
|
if err = json.Unmarshal(buf, &index); err == nil {
|
|
|
|
for _, manifest := range index.Manifests {
|
|
|
|
if val, ok := manifest.Annotations[ispec.AnnotationRefName]; ok && val == "tag2" {
|
|
|
|
delete(manifest.Annotations, ispec.AnnotationRefName)
|
|
|
|
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
buf, err = json.Marshal(index)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
err = os.WriteFile(indexPath, buf, 0o600)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseStorage(repoDB, storeController, log.NewLogger("debug", ""))
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
repos, err := repoDB.GetMultipleRepoMeta(
|
|
|
|
context.Background(),
|
|
|
|
func(repoMeta repodb.RepoMetadata) bool { return true },
|
|
|
|
repodb.PageInput{},
|
|
|
|
)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
So(len(repos), ShouldEqual, 1)
|
|
|
|
So(len(repos[0].Tags), ShouldEqual, 2)
|
|
|
|
|
|
|
|
for _, descriptor := range repos[0].Tags {
|
|
|
|
manifestMeta, err := repoDB.GetManifestMeta(repo, godigest.Digest(descriptor.Digest))
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
So(manifestMeta.ManifestBlob, ShouldNotBeNil)
|
|
|
|
So(manifestMeta.ConfigBlob, ShouldNotBeNil)
|
|
|
|
|
|
|
|
if descriptor.Digest == signedManifestDigest.String() {
|
2023-03-23 13:08:11 -05:00
|
|
|
So(manifestMeta.Signatures, ShouldNotBeEmpty)
|
2023-01-09 15:37:44 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
|
|
|
|
2023-05-15 04:02:23 -05:00
|
|
|
Convey("Accept orphan signatures", func() {
|
2023-01-09 15:37:44 -05:00
|
|
|
imageStore := local.NewImageStore(rootDir, false, 0, false, false,
|
|
|
|
log.NewLogger("debug", ""), monitoring.NewMetricsServer(false, log.NewLogger("debug", "")), nil, nil)
|
|
|
|
|
|
|
|
storeController := storage.StoreController{DefaultStore: imageStore}
|
|
|
|
// add an image
|
|
|
|
config, layers, manifest, err := test.GetRandomImageComponents(100)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
err = test.WriteImageToFileSystem(
|
|
|
|
test.Image{
|
2023-02-27 14:23:18 -05:00
|
|
|
Config: config,
|
|
|
|
Layers: layers,
|
|
|
|
Manifest: manifest,
|
|
|
|
Reference: "tag1",
|
2023-01-09 15:37:44 -05:00
|
|
|
},
|
|
|
|
repo,
|
|
|
|
storeController)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
// add mock cosign signature without pushing the signed image
|
2023-05-15 04:02:23 -05:00
|
|
|
image, err := test.GetRandomImage("")
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-05-15 04:02:23 -05:00
|
|
|
signatureTag, err := test.GetCosignSignatureTagForManifest(image.Manifest)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
missingImageDigest, err := image.Digest()
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
// get the body of the signature
|
|
|
|
config, layers, manifest, err = test.GetRandomImageComponents(100)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
err = test.WriteImageToFileSystem(
|
|
|
|
test.Image{
|
2023-02-27 14:23:18 -05:00
|
|
|
Config: config,
|
|
|
|
Layers: layers,
|
|
|
|
Manifest: manifest,
|
|
|
|
Reference: signatureTag,
|
2023-01-09 15:37:44 -05:00
|
|
|
},
|
|
|
|
repo,
|
|
|
|
storeController)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseStorage(repoDB, storeController, log.NewLogger("debug", ""))
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
repos, err := repoDB.GetMultipleRepoMeta(
|
|
|
|
context.Background(),
|
|
|
|
func(repoMeta repodb.RepoMetadata) bool { return true },
|
|
|
|
repodb.PageInput{},
|
|
|
|
)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-05-15 04:02:23 -05:00
|
|
|
for _, desc := range repos[0].Tags {
|
|
|
|
So(desc.Digest, ShouldNotResemble, missingImageDigest.String())
|
|
|
|
}
|
|
|
|
|
2023-01-09 15:37:44 -05:00
|
|
|
So(len(repos), ShouldEqual, 1)
|
|
|
|
So(repos[0].Tags, ShouldContainKey, "tag1")
|
|
|
|
So(repos[0].Tags, ShouldNotContainKey, signatureTag)
|
2023-05-15 04:02:23 -05:00
|
|
|
So(repos[0].Signatures, ShouldContainKey, missingImageDigest.String())
|
2023-01-09 15:37:44 -05:00
|
|
|
})
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
Convey("Check statistics after load", func() {
|
2023-01-09 15:37:44 -05:00
|
|
|
imageStore := local.NewImageStore(rootDir, false, 0, false, false,
|
|
|
|
log.NewLogger("debug", ""), monitoring.NewMetricsServer(false, log.NewLogger("debug", "")), nil, nil)
|
|
|
|
|
|
|
|
storeController := storage.StoreController{DefaultStore: imageStore}
|
2023-03-23 13:08:11 -05:00
|
|
|
// add an image
|
|
|
|
image, err := test.GetRandomImage("tag")
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
manifestDigest, err := image.Digest()
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
|
|
|
err = test.WriteImageToFileSystem(
|
2023-03-23 13:08:11 -05:00
|
|
|
image,
|
2023-01-09 15:37:44 -05:00
|
|
|
repo,
|
|
|
|
storeController)
|
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
err = repoDB.SetRepoReference(repo, "tag", manifestDigest, ispec.MediaTypeImageManifest)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
err = repoDB.IncrementRepoStars(repo)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
2023-03-23 13:08:11 -05:00
|
|
|
err = repoDB.IncrementImageDownloads(repo, "tag")
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
2023-03-23 13:08:11 -05:00
|
|
|
err = repoDB.IncrementImageDownloads(repo, "tag")
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
2023-03-23 13:08:11 -05:00
|
|
|
err = repoDB.IncrementImageDownloads(repo, "tag")
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
repoMeta, err := repoDB.GetRepoMeta(repo)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
So(repoMeta.Statistics[manifestDigest.String()].DownloadCount, ShouldEqual, 3)
|
|
|
|
So(repoMeta.Stars, ShouldEqual, 1)
|
2023-01-09 15:37:44 -05:00
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
err = repodb.ParseStorage(repoDB, storeController, log.NewLogger("debug", ""))
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
repoMeta, err = repoDB.GetRepoMeta(repo)
|
2023-01-09 15:37:44 -05:00
|
|
|
So(err, ShouldBeNil)
|
|
|
|
|
2023-03-23 13:08:11 -05:00
|
|
|
So(repoMeta.Statistics[manifestDigest.String()].DownloadCount, ShouldEqual, 3)
|
|
|
|
So(repoMeta.Stars, ShouldEqual, 1)
|
2023-01-09 15:37:44 -05:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2023-03-10 13:37:29 -05:00
|
|
|
func TestGetReferredSubject(t *testing.T) {
|
|
|
|
Convey("GetReferredSubject error", t, func() {
|
2023-05-10 12:15:33 -05:00
|
|
|
_, _, _, err := repodb.GetReferredSubject([]byte("bad json"), "digest", ispec.MediaTypeImageManifest)
|
2023-03-10 13:37:29 -05:00
|
|
|
So(err, ShouldNotBeNil)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2023-01-09 15:37:44 -05:00
|
|
|
func skipIt(t *testing.T) {
|
|
|
|
t.Helper()
|
|
|
|
|
|
|
|
if os.Getenv("S3MOCK_ENDPOINT") == "" {
|
|
|
|
t.Skip("Skipping testing without AWS S3 mock server")
|
|
|
|
}
|
|
|
|
}
|