129 lines
4.0 KiB
Go
129 lines
4.0 KiB
Go
/*
|
|
Copyright The containerd Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package images
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/containerd/containerd/errdefs"
|
|
criconfig "github.com/containerd/containerd/pkg/cri/config"
|
|
imagestore "github.com/containerd/containerd/pkg/cri/store/image"
|
|
snapshotstore "github.com/containerd/containerd/pkg/cri/store/snapshot"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const (
|
|
testImageFSPath = "/test/image/fs/path"
|
|
testRootDir = "/test/root"
|
|
testStateDir = "/test/state"
|
|
// Use an image id as test sandbox image to avoid image name resolve.
|
|
// TODO(random-liu): Change this to image name after we have complete image
|
|
// management unit test framework.
|
|
testSandboxImage = "sha256:c75bebcdd211f41b3a460c7bf82970ed6c75acaab9cd4c9a4e125b03ca113798"
|
|
)
|
|
|
|
// newTestCRIService creates a fake criService for test.
|
|
func newTestCRIService() *CRIImageService {
|
|
return &CRIImageService{
|
|
config: testConfig,
|
|
imageFSPath: testImageFSPath,
|
|
imageStore: imagestore.NewStore(nil),
|
|
snapshotStore: snapshotstore.NewStore(),
|
|
}
|
|
}
|
|
|
|
var testConfig = criconfig.Config{
|
|
RootDir: testRootDir,
|
|
StateDir: testStateDir,
|
|
PluginConfig: criconfig.PluginConfig{
|
|
SandboxImage: testSandboxImage,
|
|
TolerateMissingHugetlbController: true,
|
|
},
|
|
}
|
|
|
|
func TestLocalResolve(t *testing.T) {
|
|
image := imagestore.Image{
|
|
ID: "sha256:c75bebcdd211f41b3a460c7bf82970ed6c75acaab9cd4c9a4e125b03ca113799",
|
|
ChainID: "test-chain-id-1",
|
|
References: []string{
|
|
"docker.io/library/busybox:latest",
|
|
"docker.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
|
|
},
|
|
Size: 10,
|
|
}
|
|
c := newTestCRIService()
|
|
var err error
|
|
c.imageStore, err = imagestore.NewFakeStore([]imagestore.Image{image})
|
|
assert.NoError(t, err)
|
|
|
|
for _, ref := range []string{
|
|
"sha256:c75bebcdd211f41b3a460c7bf82970ed6c75acaab9cd4c9a4e125b03ca113799",
|
|
"busybox",
|
|
"busybox:latest",
|
|
"busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
|
|
"library/busybox",
|
|
"library/busybox:latest",
|
|
"library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
|
|
"docker.io/busybox",
|
|
"docker.io/busybox:latest",
|
|
"docker.io/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
|
|
"docker.io/library/busybox",
|
|
"docker.io/library/busybox:latest",
|
|
"docker.io/library/busybox@sha256:e6693c20186f837fc393390135d8a598a96a833917917789d63766cab6c59582",
|
|
} {
|
|
img, err := c.LocalResolve(ref)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, image, img)
|
|
}
|
|
img, err := c.LocalResolve("randomid")
|
|
assert.Equal(t, errdefs.IsNotFound(err), true)
|
|
assert.Equal(t, imagestore.Image{}, img)
|
|
}
|
|
|
|
func TestRuntimeSnapshotter(t *testing.T) {
|
|
defaultRuntime := criconfig.Runtime{
|
|
Snapshotter: "",
|
|
}
|
|
|
|
fooRuntime := criconfig.Runtime{
|
|
Snapshotter: "devmapper",
|
|
}
|
|
|
|
for desc, test := range map[string]struct {
|
|
runtime criconfig.Runtime
|
|
expectSnapshotter string
|
|
}{
|
|
"should return default snapshotter when runtime.Snapshotter is not set": {
|
|
runtime: defaultRuntime,
|
|
expectSnapshotter: criconfig.DefaultConfig().Snapshotter,
|
|
},
|
|
"should return overridden snapshotter when runtime.Snapshotter is set": {
|
|
runtime: fooRuntime,
|
|
expectSnapshotter: "devmapper",
|
|
},
|
|
} {
|
|
t.Run(desc, func(t *testing.T) {
|
|
cri := newTestCRIService()
|
|
cri.config = criconfig.Config{
|
|
PluginConfig: criconfig.DefaultConfig(),
|
|
}
|
|
assert.Equal(t, test.expectSnapshotter, cri.RuntimeSnapshotter(context.Background(), test.runtime))
|
|
})
|
|
}
|
|
}
|