fix(image): force-pull when digest pinned but missing locally

cmd/core/ensure_image_test.go

@@ -0,0 +1,119 @@
+package core
+
+import (
+	"context"
+	"testing"
+
+	"github.com/cocoonstack/cocoon/gc"
+	imagebackend "github.com/cocoonstack/cocoon/images"
+	"github.com/cocoonstack/cocoon/progress"
+	"github.com/cocoonstack/cocoon/types"
+)
+
+type fakeImageBackend struct {
+	typ          string
+	inspectByRef map[string]*types.Image
+	pullRefs     []string
+	pullForce    []bool
+}
+
+func (f *fakeImageBackend) Type() string { return f.typ }
+
+func (f *fakeImageBackend) Pull(_ context.Context, ref string, force bool, _ progress.Tracker) error {
+	f.pullRefs = append(f.pullRefs, ref)
+	f.pullForce = append(f.pullForce, force)
+	return nil
+}
+
+func (f *fakeImageBackend) Inspect(_ context.Context, ref string) (*types.Image, error) {
+	return f.inspectByRef[ref], nil
+}
+
+func (f *fakeImageBackend) Import(context.Context, string, progress.Tracker, ...string) error {
+	return nil
+}
+
+func (f *fakeImageBackend) List(context.Context) ([]*types.Image, error) { return nil, nil }
+func (f *fakeImageBackend) Delete(context.Context, []string) ([]string, error) {
+	return nil, nil
+}
+func (f *fakeImageBackend) RegisterGC(*gc.Orchestrator) {}
+func (f *fakeImageBackend) Config(context.Context, []*types.VMConfig) ([][]*types.StorageConfig, []*types.BootConfig, error) {
+	return nil, nil, nil
+}
+
+// Regression guard for issue 37: pinned digest with no local hit must force-pull
+// to bypass cloudimg's URL-keyed cache.
+func TestEnsureImage_ForceWhenDigestPinned(t *testing.T) {
+	const (
+		url    = "https://epoch.example/dl/simular/win11"
+		digest = "sha256:adafd938488daa114be898848eb24b9b0afffc21ac18f8b11f3f0057644b11e1"
+	)
+
+	tests := []struct {
+		name        string
+		vmCfg       *types.VMConfig
+		wantPullRef string
+		wantForce   bool
+	}{
+		{
+			name: "digest pinned -> force=true",
+			vmCfg: &types.VMConfig{
+				Config: types.Config{
+					Image:       url,
+					ImageDigest: digest,
+					ImageType:   types.ImageTypeCloudImg,
+				},
+			},
+			wantPullRef: url,
+			wantForce:   true,
+		},
+		{
+			name: "no digest -> force=false",
+			vmCfg: &types.VMConfig{
+				Config: types.Config{
+					Image:     url,
+					ImageType: types.ImageTypeCloudImg,
+				},
+			},
+			wantPullRef: url,
+			wantForce:   false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			f := &fakeImageBackend{typ: types.ImageTypeCloudImg}
+			EnsureImage(t.Context(), []imagebackend.Images{f}, tt.vmCfg)
+			if len(f.pullRefs) != 1 {
+				t.Fatalf("Pull invocations = %d, want 1", len(f.pullRefs))
+			}
+			if f.pullRefs[0] != tt.wantPullRef {
+				t.Errorf("Pull ref = %q, want %q", f.pullRefs[0], tt.wantPullRef)
+			}
+			if f.pullForce[0] != tt.wantForce {
+				t.Errorf("Pull force = %v, want %v", f.pullForce[0], tt.wantForce)
+			}
+		})
+	}
+}
+
+func TestEnsureImage_SkipsPullWhenDigestLocal(t *testing.T) {
+	const digest = "sha256:adafd938488daa114be898848eb24b9b0afffc21ac18f8b11f3f0057644b11e1"
+	f := &fakeImageBackend{
+		typ: types.ImageTypeCloudImg,
+		inspectByRef: map[string]*types.Image{
+			digest: {ID: digest, Name: "win11", Type: types.ImageTypeCloudImg},
+		},
+	}
+	EnsureImage(t.Context(), []imagebackend.Images{f}, &types.VMConfig{
+		Config: types.Config{
+			Image:       "https://epoch.example/dl/simular/win11",
+			ImageDigest: digest,
+			ImageType:   types.ImageTypeCloudImg,
+		},
+	})
+	if len(f.pullRefs) != 0 {
+		t.Errorf("Pull called %d time(s), want 0", len(f.pullRefs))
+	}
+}

cmd/core/helpers.go

@@ -249,7 +249,9 @@ func EnsureImage(ctx context.Context, backends []imagebackend.Images, vmCfg *typ
 		// version recorded at snapshot time, not whatever the tag points to now.
 		pullRef := digestPullRef(vmCfg.Image, vmCfg.ImageDigest, vmCfg.ImageType)
 		logger.Infof(ctx, "base image not found locally, pulling %s ...", pullRef)
-		if pullErr := b.Pull(ctx, pullRef, false, progress.Nop); pullErr != nil {
+		// Pinned digest with no local hit: force past cloudimg's URL-keyed cache.
+		needForce := vmCfg.ImageDigest != ""
+		if pullErr := b.Pull(ctx, pullRef, needForce, progress.Nop); pullErr != nil {
 			logger.Warnf(ctx, "auto-pull %s failed (imported image?): %v — clone may fail if base layers are missing", pullRef, pullErr)
 			return
 		}