containerd/integration/client/lease_test.go
Paul "TBBle" Hampson b375f90128 Use k8s.gcr.io/pause:3.4.1 in pull-only tests
This reduces the need to pull random images from docker.io, and should
greatly reduce the tendancy to hit their hourly rate-limit during
integration test runs.

TestImagePullSomePlatforms uses k8s.gcr.io/pause:3.2 so that it does not
see the content pulled by TestImagePullAllPlatforms. This image is
multi-arch, but not multi-os.

Signed-off-by: Paul "TBBle" Hampson <Paul.Hampson@Pobox.com>
2021-03-13 13:08:12 +11:00

142 lines
3.4 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 containerd
import (
"runtime"
"testing"
. "github.com/containerd/containerd"
"github.com/containerd/containerd/errdefs"
"github.com/containerd/containerd/images"
"github.com/containerd/containerd/leases"
"github.com/opencontainers/image-spec/identity"
)
func TestLeaseResources(t *testing.T) {
if runtime.GOOS == "windows" {
t.Skip()
}
ctx, cancel := testContext(t)
defer cancel()
client, err := newClient(t, address)
if err != nil {
t.Fatal(err)
}
defer client.Close()
var (
ls = client.LeasesService()
cs = client.ContentStore()
imgSrv = client.ImageService()
sn = client.SnapshotService("native")
)
l, err := ls.Create(ctx, leases.WithRandomID())
if err != nil {
t.Fatal(err)
}
defer ls.Delete(ctx, l, leases.SynchronousDelete)
// step 1: download image
imageName := "k8s.gcr.io/pause:3.4.1"
image, err := client.Pull(ctx, imageName, WithPullUnpack, WithPullSnapshotter("native"))
if err != nil {
t.Fatal(err)
}
defer imgSrv.Delete(ctx, imageName)
// both the config and snapshotter should exist
cfgDesc, err := image.Config(ctx)
if err != nil {
t.Fatal(err)
}
if _, err := cs.Info(ctx, cfgDesc.Digest); err != nil {
t.Fatal(err)
}
dgsts, err := image.RootFS(ctx)
if err != nil {
t.Fatal(err)
}
chainID := identity.ChainID(dgsts)
if _, err := sn.Stat(ctx, chainID.String()); err != nil {
t.Fatal(err)
}
// step 2: reference snapshotter with lease
r := leases.Resource{
ID: chainID.String(),
Type: "snapshots/native",
}
if err := ls.AddResource(ctx, l, r); err != nil {
t.Fatal(err)
}
list, err := ls.ListResources(ctx, l)
if err != nil {
t.Fatal(err)
}
if len(list) != 1 || list[0] != r {
t.Fatalf("expected (%v), but got (%v)", []leases.Resource{r}, list)
}
// step 3: remove image and check the status of snapshotter and content
if err := imgSrv.Delete(ctx, imageName, images.SynchronousDelete()); err != nil {
t.Fatal(err)
}
// config should be removed but the snapshotter should exist
if _, err := cs.Info(ctx, cfgDesc.Digest); !errdefs.IsNotFound(err) {
t.Fatalf("expected error(%v), but got(%v)", errdefs.ErrNotFound, err)
}
if _, err := sn.Stat(ctx, chainID.String()); err != nil {
t.Fatal(err)
}
// step 4: remove resource from the lease and check the list API
if err := ls.DeleteResource(ctx, l, r); err != nil {
t.Fatal(err)
}
list, err = ls.ListResources(ctx, l)
if err != nil {
t.Fatal(err)
}
if len(list) != 0 {
t.Fatalf("expected nothing, but got (%v)", list)
}
// step 5: remove the lease to check the status of snapshotter
if err := ls.Delete(ctx, l, leases.SynchronousDelete); err != nil {
t.Fatal(err)
}
if _, err := sn.Stat(ctx, chainID.String()); !errdefs.IsNotFound(err) {
t.Fatalf("expected error(%v), but got(%v)", errdefs.ErrNotFound, err)
}
}