cri: fix using the pinned label to pin image

Iceber · Iceber · commit 751b0c1867b2 · 2023-11-16T15:08:10.000+08:00
Signed-off-by: Iceber Gu <wei.cai-nat@daocloud.io> (cherry picked from commit 7f7ba31) Signed-off-by: Iceber Gu <caiwei95@hotmail.com>
diff --git a/integration/containerd_image_test.go b/integration/containerd_image_test.go
@@ -27,6 +27,7 @@ import (
 	"github.com/containerd/containerd"
 	"github.com/containerd/containerd/errdefs"
 	"github.com/containerd/containerd/namespaces"
+	"github.com/containerd/containerd/pkg/cri/labels"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"golang.org/x/net/context"
@@ -46,7 +47,8 @@ func TestContainerdImage(t *testing.T) {
 	}
 
 	t.Logf("pull the image into containerd")
-	_, err = containerdClient.Pull(ctx, testImage, containerd.WithPullUnpack, containerd.WithPullLabel("foo", "bar"))
+	lbs := map[string]string{"foo": "bar", labels.PinnedImageLabelKey: labels.PinnedImageLabelValue}
+	_, err = containerdClient.Pull(ctx, testImage, containerd.WithPullUnpack, containerd.WithPullLabels(lbs))
 	assert.NoError(t, err)
 	defer func() {
 		// Make sure the image is cleaned up in any case.
@@ -127,6 +129,13 @@ func TestContainerdImage(t *testing.T) {
 	img, err := containerdClient.GetImage(ctx, testImage)
 	assert.NoError(t, err)
 	assert.Equal(t, img.Labels()["foo"], "bar")
+	assert.Equal(t, img.Labels()[labels.ImageLabelKey], labels.ImageLabelValue)
+
+	t.Logf("the image should be pinned")
+	i, err = imageService.ImageStatus(&runtime.ImageSpec{Image: testImage})
+	require.NoError(t, err)
+	require.NotNil(t, i)
+	assert.True(t, i.Pinned)
 
 	t.Logf("should be able to start container with the image")
 	sb, sbConfig := PodSandboxConfigWithCleanup(t, "sandbox", "containerd-image")
diff --git a/pkg/cri/store/image/image.go b/pkg/cri/store/image/image.go
@@ -218,6 +218,7 @@ func (s *store) add(img Image) error {
 	}
 	// Or else, merge and sort the references.
 	i.References = sortReferences(util.MergeStringSlices(i.References, img.References))
+	i.Pinned = i.Pinned || img.Pinned
 	s.images[img.ID] = i
 	return nil
 }

Original file line number	Diff line number	Diff line change
`@@ -218,6 +218,7 @@ func (s *store) add(img Image) error {`
`218`	`218`	`}`
`219`	`219`	`// Or else, merge and sort the references.`
`220`	`220`	`i.References = sortReferences(util.MergeStringSlices(i.References, img.References))`
	`221`	`+ i.Pinned = i.Pinned \|\| img.Pinned`
`221`	`222`	`s.images[img.ID] = i`
`222`	`223`	`return nil`
`223`	`224`	`}`