Files
zot/pkg/extensions/sync/sync_internal_test.go
T
Andrei Aaron dfb5d1df54 fix: make config read/write thread safe (#3432)
* fix: make config read/write thread safe and fix some other similar issues

1. The config config has a lock, and safe methods to update and read the attributes
2. The config has methods to retrieve copies of specific attributes, such as the extyensions config, the auth config, and the authz config.
These are needed, as the config object may mutate in the middle of an auth/authz requests, and we avoid partial configuration being applied for that request.
3. Fix an issue with the monitoring server not stopping when the controller is shut down.
4. Fix an issue with the HTPasswdWatcher not stopping when the background tasks are supposed to finish.
5. Fix some tests using hardcoded ports.

Moved some of the methods which were on the main config to the auth, access control and extension configs

Signed-off-by: Andrei Aaron <andreifdaaron@gmail.com>
2025-10-18 11:20:58 +03:00

826 lines
28 KiB
Go

//go:build sync
// +build sync
package sync
import (
"bytes"
"context"
"encoding/json"
"errors"
"os"
"testing"
"time"
godigest "github.com/opencontainers/go-digest"
ispec "github.com/opencontainers/image-spec/specs-go/v1"
"github.com/regclient/regclient/types/ref"
. "github.com/smartystreets/goconvey/convey"
zerr "zotregistry.dev/zot/v2/errors"
"zotregistry.dev/zot/v2/pkg/extensions/config"
syncconf "zotregistry.dev/zot/v2/pkg/extensions/config/sync"
"zotregistry.dev/zot/v2/pkg/extensions/lint"
"zotregistry.dev/zot/v2/pkg/extensions/monitoring"
"zotregistry.dev/zot/v2/pkg/log"
mTypes "zotregistry.dev/zot/v2/pkg/meta/types"
"zotregistry.dev/zot/v2/pkg/storage"
"zotregistry.dev/zot/v2/pkg/storage/cache"
storageConstants "zotregistry.dev/zot/v2/pkg/storage/constants"
"zotregistry.dev/zot/v2/pkg/storage/local"
. "zotregistry.dev/zot/v2/pkg/test/image-utils"
"zotregistry.dev/zot/v2/pkg/test/mocks"
)
func TestService(t *testing.T) {
Convey("trigger fetch tags error", t, func() {
conf := syncconf.RegistryConfig{
URLs: []string{"http://localhost"},
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
err = service.SyncRepo(context.Background(), "repo")
So(err, ShouldNotBeNil)
})
Convey("test context cancellation in SyncRepo without mock", t, func() {
conf := syncconf.RegistryConfig{
URLs: []string{"http://localhost"},
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
// Create a context that's already cancelled
ctx, cancel := context.WithCancel(context.Background())
cancel() // Cancel immediately
err = service.SyncRepo(ctx, "repo")
So(err, ShouldNotBeNil)
// This will fail at getTags before reaching the cancellation check
})
Convey("test context cancellation in SyncRepo with mock", t, func() {
conf := syncconf.RegistryConfig{
URLs: []string{"http://localhost"},
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
// Create a mock remote that returns tags so we can reach the loop
mockRemote := &mocks.SyncRemoteMock{
GetTagsFn: func(ctx context.Context, repo string) ([]string, error) {
return []string{"tag1", "tag2", "tag3"}, nil
},
}
service.remote = mockRemote
// Create a context that's already cancelled
ctx, cancel := context.WithCancel(context.Background())
cancel() // Cancel immediately
err = service.SyncRepo(ctx, "repo")
So(err, ShouldNotBeNil)
So(errors.Is(err, context.Canceled), ShouldBeTrue)
})
Convey("test SyncReferrers ReferrerList error", t, func() {
conf := syncconf.RegistryConfig{
URLs: []string{"http://localhost"},
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
// Create a minimal mock remote that only returns tags
mockRemote := &mocks.SyncRemoteMock{
GetTagsFn: func(ctx context.Context, repo string) ([]string, error) {
return []string{"tag1"}, nil
},
}
service.remote = mockRemote
// Set rc to nil to force a panic at ReferrerList call
service.rc = nil
// Use defer to catch the panic - this confirms we reached the ReferrerList call
var panicOccurred bool
defer func() {
if r := recover(); r != nil {
panicOccurred = true
t.Logf("SyncReferrers panic (expected): %v", r)
}
}()
ctx := context.Background()
err = service.SyncReferrers(ctx, "repo", "tag1", []string{"signature"})
// We expect a panic when rc is nil, which confirms we reached the ReferrerList call
So(panicOccurred, ShouldBeTrue)
})
Convey("test syncImage ReferrerList error with OnlySigned", t, func() {
onlySigned := true
conf := syncconf.RegistryConfig{
URLs: []string{"http://invalid-registry-that-does-not-exist:9999"},
OnlySigned: &onlySigned,
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
// Create a mock remote that returns necessary data
mockRemote := &mocks.SyncRemoteMock{
GetImageReferenceFn: func(repo string, tag string) (ref.Ref, error) {
return ref.New("invalid-registry-that-does-not-exist:9999/" + repo + ":" + tag)
},
GetDigestFn: func(ctx context.Context, repo, tag string) (godigest.Digest, error) {
return godigest.Digest("sha256:abc123"), nil
},
}
service.remote = mockRemote
// Create a mock destination
mockDest := &mocks.SyncDestinationMock{
GetImageReferenceFn: func(repo string, tag string) (ref.Ref, error) {
return ref.New("local/" + repo + ":" + tag)
},
}
service.destination = mockDest
ctx := context.Background()
err = service.syncImage(ctx, "localrepo", "remoterepo", "tag1", []string{}, true)
// We expect an error when ReferrerList fails (network/connection error in this case)
So(err, ShouldNotBeNil)
})
Convey("test LoadOrStore continue path by pre-populating requestStore", t, func() {
// Strategy: Pre-populate requestStore to force LoadOrStore to return true
maxRetries := 2
retryDelay := 100 * time.Millisecond
conf := syncconf.RegistryConfig{
URLs: []string{"http://localhost:32768"}, // Invalid port to force errors
MaxRetries: &maxRetries,
RetryDelay: &retryDelay,
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
onDemand := NewOnDemand(log.NewTestLogger())
onDemand.Add(service)
ctx := context.Background()
// Step 1: Verify empty requestStore initially
initialImageCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
initialImageCount++
return true
})
So(initialImageCount, ShouldEqual, 0)
// Step 2: Pre-populate image background retry request
duplicateImageReq := request{
repo: "test-duplicate-repo",
reference: "test-duplicate-tag",
serviceID: 0,
isBackground: true,
}
onDemand.requestStore.Store(duplicateImageReq, struct{}{})
// Step 3: Verify we now have 1 request
preImageCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
preImageCount++
return true
})
So(preImageCount, ShouldEqual, 1) // Should be 1 after pre-population
// Step 4: Trigger sync - should execute continue path
err = onDemand.SyncImage(ctx, "test-duplicate-repo", "test-duplicate-tag")
So(err, ShouldNotBeNil) // Should still error due to invalid registry
// Step 5: Verify CONTINUE PATH EXECUTED - no new requests created
postImageCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
postImageCount++
return true
})
So(postImageCount, ShouldEqual, preImageCount) // Count unchanged = continue executed
// Step 6: Verify image request unchanged (proves no background goroutine started)
value, exists := onDemand.requestStore.Load(duplicateImageReq)
So(exists, ShouldBeTrue)
So(value, ShouldEqual, struct{}{}) // Should still be pre-populated value
// Step 7: Verify current state before referrer test - we should have 1 request
initialReferrerCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
initialReferrerCount++
return true
})
So(initialReferrerCount, ShouldEqual, 1) // Should have 1 from image test
// Step 8: Pre-populate referrer background retry request
duplicateReferrerReq := request{
repo: "test-duplicate-referrer-repo",
reference: "sha256:duplicate",
serviceID: 0,
isBackground: true,
}
onDemand.requestStore.Store(duplicateReferrerReq, struct{}{})
// Step 9: Verify we now have 2 requests (image + referrer)
preReferrerCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
preReferrerCount++
return true
})
So(preReferrerCount, ShouldEqual, 2) // Should have 2 after referrer pre-population
// Step 10: Trigger referrer sync - should execute continue path
err = onDemand.SyncReferrers(ctx, "test-duplicate-referrer-repo", "sha256:duplicate", []string{"signature"})
So(err, ShouldNotBeNil) // Should still error due to invalid registry
// Step 11: Verify CONTINUE PATH EXECUTED - no new referrer requests created
postReferrerCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
postReferrerCount++
return true
})
So(postReferrerCount, ShouldEqual, preReferrerCount) // Count unchanged = continue executed
// Step 12: Verify referrer request unchanged (proves no background goroutine started)
value, exists = onDemand.requestStore.Load(duplicateReferrerReq)
So(exists, ShouldBeTrue)
So(value, ShouldEqual, struct{}{}) // Should still be pre-populated value
// Step 13: Final verification - exactly 2 requests total (both pre-populated, none deleted)
finalCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
finalCount++
return true
})
So(finalCount, ShouldEqual, 2)
})
Convey("test continue paths for specific error types in on-demand sync", t, func() {
// Strategy: Create multiple services where one returns ErrSyncImageFilteredOut
// This will trigger the continue path when looping through services
ctx := context.Background()
// Create first service with content filtering
maxRetries := 2
retryDelay := 100 * time.Millisecond
conf1 := syncconf.RegistryConfig{
URLs: []string{"http://localhost:32768"}, // Invalid port to force errors
MaxRetries: &maxRetries,
RetryDelay: &retryDelay,
Content: []syncconf.Content{{
Prefix: "different-prefix", // Won't match our test repo
}},
}
service1, err := New(conf1, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
// Create second service for normal processing
conf2 := syncconf.RegistryConfig{
URLs: []string{"http://localhost:32768"}, // Invalid port to force errors
MaxRetries: &maxRetries,
RetryDelay: &retryDelay,
Content: []syncconf.Content{{
Prefix: "test-repo", // Will match our test repo
}},
}
service2, err := New(conf2, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
onDemand := NewOnDemand(log.NewTestLogger())
onDemand.Add(service1) // First service will return ErrSyncImageFilteredOut
onDemand.Add(service2) // Second service will process normally but error on network
// Test ErrSyncImageFilteredOut continue path in SyncImage
// The first service returns ErrSyncImageFilteredOut (causes continue), second service processes but errors
err = onDemand.SyncImage(ctx, "test-filtered-repo", "test-tag")
So(err, ShouldNotBeNil) // Should get error from second service (network error)
// Test ErrSyncImageFilteredOut continue path in SyncReferrers
err = onDemand.SyncReferrers(ctx, "test-filtered-referrer-repo", "sha256:test", []string{"signature"})
So(err, ShouldNotBeNil) // Should get error from second service (network error)
})
Convey("test continue paths in both SyncImage and SyncReferrers", t, func() {
// Helper function to create filter test scenario
createFilteredService := func(prefix string) (*BaseService, *BaseOnDemand) {
maxRetries := 2
retryDelay := 100 * time.Millisecond
conf := syncconf.RegistryConfig{
URLs: []string{"http://localhost:32768"}, // Invalid port to force errors
MaxRetries: &maxRetries,
RetryDelay: &retryDelay,
Content: []syncconf.Content{{
Prefix: prefix, // Won't match our test repo
}},
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
onDemand := NewOnDemand(log.NewTestLogger())
onDemand.Add(service)
return service, onDemand
}
ctx := context.Background()
// Test SyncReferrers continue path
Convey("SyncReferrers continue path", func() {
_, onDemand := createFilteredService("different-referrer-prefix")
err := onDemand.SyncReferrers(ctx, "test-unfiltered-referrer-repo", "sha256:test", []string{"signature"})
So(err, ShouldNotBeNil) // Should get error from sync process (filtered out -> continue -> network error)
})
// Test SyncImage continue path
Convey("SyncImage continue path", func() {
_, onDemand := createFilteredService("different-image-prefix")
err := onDemand.SyncImage(ctx, "test-unfiltered-image-repo", "test-tag")
So(err, ShouldNotBeNil) // Should get error from sync process (filtered out -> continue -> network error)
})
})
Convey("test background retry goroutines for both SyncImage and SyncReferrers", t, func() {
// Helper function to create background retry test scenario
createBackgroundRetryService := func(prefix string) (*BaseService, *BaseOnDemand, time.Duration) {
maxRetries := 2
retryDelay := 1 * time.Second
conf := syncconf.RegistryConfig{
URLs: []string{"http://localhost:32768"}, // Invalid port to force errors
MaxRetries: &maxRetries, // Enable retries so CanRetryOnError() returns true
RetryDelay: &retryDelay,
Content: []syncconf.Content{{
Prefix: prefix, // Will match our repo
}},
}
service, err := New(conf, "", nil, os.TempDir(), storage.StoreController{}, mocks.MetaDBMock{}, log.NewTestLogger())
So(err, ShouldBeNil)
onDemand := NewOnDemand(log.NewTestLogger())
onDemand.Add(service)
return service, onDemand, retryDelay
}
// Test SyncReferrers background retry
Convey("SyncReferrers background retry", func() {
_, onDemand, retryDelay := createBackgroundRetryService("test-background-retry")
ctx := context.Background()
// Verify initial requestStore is empty
initialCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
initialCount++
return true
})
So(initialCount, ShouldEqual, 0)
// Call SyncReferrers - should trigger background retry since:
// 1. Network error (not a continue-path error)
// 2. CanRetryOnError() returns true (maxRetries > 0)
// 3. No existing background retry
err := onDemand.SyncReferrers(ctx, "test-background-retry", "sha256:background", []string{"signature"})
So(err, ShouldNotBeNil) // Should get original network error
// Wait for background goroutine to start and store request
time.Sleep(50 * time.Millisecond)
// Verify background retry request was stored
reqBackground := request{
repo: "test-background-retry",
reference: "sha256:background",
serviceID: 0,
isBackground: true,
}
_, existsBackground := onDemand.requestStore.Load(reqBackground)
So(existsBackground, ShouldBeTrue) // Background retry request should exist
// Wait for background retry to complete and cleanup (3x retry delay)
time.Sleep(3 * retryDelay)
// Verify background retry request was cleaned up after completion
_, existsAfterCleanup := onDemand.requestStore.Load(reqBackground)
So(existsAfterCleanup, ShouldBeFalse) // Should be cleaned up by defer function after retry
})
// Test SyncImage background retry
Convey("SyncImage background retry", func() {
_, onDemand, retryDelay := createBackgroundRetryService("test-background-image-retry")
ctx := context.Background()
// Verify initial requestStore is empty
initialCount := 0
onDemand.requestStore.Range(func(key, value interface{}) bool {
initialCount++
return true
})
So(initialCount, ShouldEqual, 0)
// Call SyncImage - should trigger background retry since:
// 1. Network error (not a continue-path error)
// 2. CanRetryOnError() returns true (maxRetries > 0)
// 3. No existing background retry
err := onDemand.SyncImage(ctx, "test-background-image-retry", "background-image-tag")
So(err, ShouldNotBeNil) // Should get original network error
// Wait for background goroutine to start and store request
time.Sleep(50 * time.Millisecond)
// Verify background retry request was stored
reqBackground := request{
repo: "test-background-image-retry",
reference: "background-image-tag",
serviceID: 0,
isBackground: true,
}
_, existsBackground := onDemand.requestStore.Load(reqBackground)
So(existsBackground, ShouldBeTrue) // Background retry request should exist
// Wait for background retry to complete and cleanup (3x retry delay)
time.Sleep(3 * retryDelay)
// Verify background retry request was cleaned up after completion
_, existsAfterCleanup := onDemand.requestStore.Load(reqBackground)
So(existsAfterCleanup, ShouldBeFalse) // Should be cleaned up by defer function after retry
})
})
Convey("test assured channel waiting path", t, func() {
// Strategy: Pre-populate requestStore with a channel to GUARANTEE the channel waiting code path
// This ensures the "waiting on channel" message is logged and the channel receive happens
Convey("SyncImage assured channel waiting", func() {
onDemand := NewOnDemand(log.NewTestLogger())
// Create request and pre-populate with a channel that we control
req := request{
repo: "test-guaranteed-channel-image",
reference: "guaranteed-image-tag",
serviceID: 0,
isBackground: false,
}
// Create a channel that we control completely
pendingChannel := make(chan error, 1)
onDemand.requestStore.Store(req, pendingChannel)
// Start request that will wait on our channel
requestCompleted := make(chan error)
go func() {
err := onDemand.SyncImage(context.Background(), "test-guaranteed-channel-image", "guaranteed-image-tag")
requestCompleted <- err
}()
// Wait a moment for the request to reach the channel waiting code
time.Sleep(50 * time.Millisecond)
// Send error through our controlled channel - this proves channel waiting worked
pendingChannel <- errors.New("guaranteed channel error")
// Verify the request got our controlled error
err := <-requestCompleted
So(err, ShouldNotBeNil)
So(err.Error(), ShouldEqual, "guaranteed channel error")
})
Convey("SyncReferrers assured channel waiting", func() {
onDemand := NewOnDemand(log.NewTestLogger())
// Create request and pre-populate with a channel that we control
req := request{
repo: "test-guaranteed-channel-referrers",
reference: "sha256:guaranteed",
serviceID: 0,
isBackground: false,
}
// Create a channel that we control completely
pendingChannel := make(chan error, 1)
onDemand.requestStore.Store(req, pendingChannel)
// Start request that will wait on our channel
requestCompleted := make(chan error)
go func() {
err := onDemand.SyncReferrers(context.Background(), "test-guaranteed-channel-referrers", "sha256:guaranteed", []string{"signature"})
requestCompleted <- err
}()
// Wait a moment for the request to reach the channel waiting code
time.Sleep(50 * time.Millisecond)
// Send error through our controlled channel - this proves channel waiting worked
pendingChannel <- errors.New("guaranteed referrer channel error")
// Verify the request got our controlled error
err := <-requestCompleted
So(err, ShouldNotBeNil)
So(err.Error(), ShouldEqual, "guaranteed referrer channel error")
})
})
}
func TestDestinationRegistry(t *testing.T) {
Convey("make StoreController", t, func() {
dir := t.TempDir()
log := log.NewTestLogger()
metrics := monitoring.NewMetricsServer(false, log)
cacheDriver, _ := storage.Create("boltdb", cache.BoltDBDriverParameters{
RootDir: dir,
Name: "cache",
UseRelPaths: true,
}, log)
syncImgStore := local.NewImageStore(dir, true, true, log, metrics, nil, cacheDriver, nil, nil)
repoName := "repo"
storeController := storage.StoreController{DefaultStore: syncImgStore}
registry := NewDestinationRegistry(storeController, storeController, nil, log)
imageReference, err := registry.GetImageReference(repoName, "1.0")
So(err, ShouldBeNil)
So(imageReference, ShouldNotBeNil)
imgStore := getImageStoreFromImageReference(repoName, imageReference, log)
// create a blob/layer
upload, err := imgStore.NewBlobUpload(repoName)
So(err, ShouldBeNil)
So(upload, ShouldNotBeEmpty)
content := []byte("this is a blob1")
buf := bytes.NewBuffer(content)
buflen := buf.Len()
digest := godigest.FromBytes(content)
So(digest, ShouldNotBeNil)
blob, err := imgStore.PutBlobChunkStreamed(repoName, upload, buf)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
bdgst1 := digest
bsize1 := len(content)
err = imgStore.FinishBlobUpload(repoName, upload, buf, digest)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
// push index image
var index ispec.Index
index.SchemaVersion = 2
index.MediaType = ispec.MediaTypeImageIndex
for i := 0; i < 4; i++ {
// upload image config blob
upload, err := imgStore.NewBlobUpload(repoName)
So(err, ShouldBeNil)
So(upload, ShouldNotBeEmpty)
cblob, cdigest := GetRandomImageConfig()
buf := bytes.NewBuffer(cblob)
buflen := buf.Len()
blob, err := imgStore.PutBlobChunkStreamed(repoName, upload, buf)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
err = imgStore.FinishBlobUpload(repoName, upload, buf, cdigest)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
// create a manifest
manifest := ispec.Manifest{
Config: ispec.Descriptor{
MediaType: ispec.MediaTypeImageConfig,
Digest: cdigest,
Size: int64(len(cblob)),
},
Layers: []ispec.Descriptor{
{
MediaType: ispec.MediaTypeImageLayer,
Digest: bdgst1,
Size: int64(bsize1),
},
},
}
manifest.SchemaVersion = 2
content, err = json.Marshal(manifest)
So(err, ShouldBeNil)
digest = godigest.FromBytes(content)
So(digest, ShouldNotBeNil)
_, _, err = imgStore.PutImageManifest(repoName, digest.String(), ispec.MediaTypeImageManifest, content)
So(err, ShouldBeNil)
index.Manifests = append(index.Manifests, ispec.Descriptor{
Digest: digest,
MediaType: ispec.MediaTypeImageManifest,
Size: int64(len(content)),
})
}
// upload index image
indexContent, err := json.Marshal(index)
So(err, ShouldBeNil)
indexDigest := godigest.FromBytes(indexContent)
So(indexDigest, ShouldNotBeNil)
_, _, err = imgStore.PutImageManifest(repoName, "1.0", ispec.MediaTypeImageIndex, indexContent)
So(err, ShouldBeNil)
Convey("sync index image", func() {
ok, err := registry.CanSkipImage(repoName, "1.0", indexDigest)
So(ok, ShouldBeFalse)
So(err, ShouldBeNil)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldBeNil)
})
Convey("CleanupImage()", func() {
ok, err := registry.CanSkipImage(repoName, "1.0", indexDigest)
So(ok, ShouldBeFalse)
So(err, ShouldBeNil)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldBeNil)
err = registry.CleanupImage(imageReference, repoName)
So(err, ShouldBeNil)
})
Convey("trigger GetImageManifest error in CommitImage()", func() {
err = os.Chmod(imgStore.BlobPath(repoName, indexDigest), 0o000)
So(err, ShouldBeNil)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldNotBeNil)
})
Convey("trigger linter error in CommitImage()", func() {
defaultVal := true
linter := lint.NewLinter(&config.LintConfig{
BaseConfig: config.BaseConfig{
Enable: &defaultVal,
},
MandatoryAnnotations: []string{"annot1"},
}, log)
syncImgStore := local.NewImageStore(dir, true, true, log, metrics, linter, cacheDriver, nil, nil)
repoName := "repo"
storeController := storage.StoreController{DefaultStore: syncImgStore}
registry := NewDestinationRegistry(storeController, storeController, nil, log)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldBeNil)
})
Convey("trigger GetBlobContent on manifest error in CommitImage()", func() {
err = os.Chmod(imgStore.BlobPath(repoName, digest), 0o000)
So(err, ShouldBeNil)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldNotBeNil)
})
Convey("trigger copyBlob() error in CommitImage()", func() {
err = os.Chmod(imgStore.BlobPath(repoName, bdgst1), 0o000)
So(err, ShouldBeNil)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldNotBeNil)
})
Convey("trigger PutImageManifest error on index manifest in CommitImage()", func() {
err = os.MkdirAll(syncImgStore.BlobPath(repoName, indexDigest), storageConstants.DefaultDirPerms)
So(err, ShouldBeNil)
err = os.Chmod(syncImgStore.BlobPath(repoName, indexDigest), 0o000)
So(err, ShouldBeNil)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldNotBeNil)
})
Convey("trigger metaDB error on index manifest in CommitImage()", func() {
storeController := storage.StoreController{DefaultStore: syncImgStore}
registry := NewDestinationRegistry(storeController, storeController, mocks.MetaDBMock{
SetRepoReferenceFn: func(ctx context.Context, repo string, reference string, imageMeta mTypes.ImageMeta) error {
if reference == "1.0" {
return zerr.ErrRepoMetaNotFound
}
return nil
},
}, log)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldNotBeNil)
})
Convey("trigger metaDB error on image manifest in CommitImage()", func() {
storeController := storage.StoreController{DefaultStore: syncImgStore}
registry := NewDestinationRegistry(storeController, storeController, mocks.MetaDBMock{
SetRepoReferenceFn: func(ctx context.Context, repo, reference string, imageMeta mTypes.ImageMeta) error {
return zerr.ErrRepoMetaNotFound
},
}, log)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldNotBeNil)
})
Convey("push image", func() {
imageReference, err := registry.GetImageReference(repoName, "2.0")
So(err, ShouldBeNil)
So(imageReference, ShouldNotBeNil)
imgStore := getImageStoreFromImageReference(repoName, imageReference, log)
// upload image
// create a blob/layer
upload, err := imgStore.NewBlobUpload(repoName)
So(err, ShouldBeNil)
So(upload, ShouldNotBeEmpty)
content := []byte("this is a blob1")
buf := bytes.NewBuffer(content)
buflen := buf.Len()
digest := godigest.FromBytes(content)
So(digest, ShouldNotBeNil)
blob, err := imgStore.PutBlobChunkStreamed(repoName, upload, buf)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
bdgst1 := digest
bsize1 := len(content)
err = imgStore.FinishBlobUpload(repoName, upload, buf, digest)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
// upload image config blob
upload, err = imgStore.NewBlobUpload(repoName)
So(err, ShouldBeNil)
So(upload, ShouldNotBeEmpty)
cblob, cdigest := GetRandomImageConfig()
buf = bytes.NewBuffer(cblob)
buflen = buf.Len()
blob, err = imgStore.PutBlobChunkStreamed(repoName, upload, buf)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
err = imgStore.FinishBlobUpload(repoName, upload, buf, cdigest)
So(err, ShouldBeNil)
So(blob, ShouldEqual, buflen)
// create a manifest
manifest := ispec.Manifest{
Config: ispec.Descriptor{
MediaType: ispec.MediaTypeImageConfig,
Digest: cdigest,
Size: int64(len(cblob)),
},
Layers: []ispec.Descriptor{
{
MediaType: ispec.MediaTypeImageLayer,
Digest: bdgst1,
Size: int64(bsize1),
},
},
}
manifest.SchemaVersion = 2
content, err = json.Marshal(manifest)
So(err, ShouldBeNil)
digest = godigest.FromBytes(content)
So(digest, ShouldNotBeNil)
_, _, err = imgStore.PutImageManifest(repoName, "2.0", ispec.MediaTypeImageManifest, content)
So(err, ShouldBeNil)
Convey("sync image", func() {
ok, err := registry.CanSkipImage(repoName, "2.0", digest)
So(ok, ShouldBeFalse)
So(err, ShouldBeNil)
err = registry.CommitAll(repoName, imageReference)
So(err, ShouldBeNil)
})
})
})
}