2f1c5a19f1
The io/ioutil package has been deprecated as of Go 1.16 [1]. This commit replaces the existing io/ioutil functions with their new definitions in io and os packages. [1]: https://golang.org/doc/go1.16#ioutil Signed-off-by: Eng Zer Jun <engzerjun@gmail.com> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
67 lines
2 KiB
Go
67 lines
2 KiB
Go
package container
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/client"
|
|
log "github.com/sirupsen/logrus"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func init() {
|
|
log.SetLevel(log.DebugLevel)
|
|
}
|
|
|
|
func TestImageExistsLocally(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("skipping integration test")
|
|
}
|
|
ctx := context.Background()
|
|
// to help make this test reliable and not flaky, we need to have
|
|
// an image that will exist, and onew that won't exist
|
|
|
|
// Test if image exists with specific tag
|
|
invalidImageTag, err := ImageExistsLocally(ctx, "library/alpine:this-random-tag-will-never-exist", "linux/amd64")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, false, invalidImageTag)
|
|
|
|
// Test if image exists with specific architecture (image platform)
|
|
invalidImagePlatform, err := ImageExistsLocally(ctx, "alpine:latest", "windows/amd64")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, false, invalidImagePlatform)
|
|
|
|
// pull an image
|
|
cli, err := client.NewClientWithOpts(client.FromEnv)
|
|
assert.Nil(t, err)
|
|
cli.NegotiateAPIVersion(context.Background())
|
|
|
|
// Chose alpine latest because it's so small
|
|
// maybe we should build an image instead so that tests aren't reliable on dockerhub
|
|
readerDefault, err := cli.ImagePull(ctx, "node:16-buster-slim", types.ImagePullOptions{
|
|
Platform: "linux/amd64",
|
|
})
|
|
assert.Nil(t, err)
|
|
defer readerDefault.Close()
|
|
_, err = io.ReadAll(readerDefault)
|
|
assert.Nil(t, err)
|
|
|
|
imageDefaultArchExists, err := ImageExistsLocally(ctx, "node:16-buster-slim", "linux/amd64")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, true, imageDefaultArchExists)
|
|
|
|
// Validate if another architecture platform can be pulled
|
|
readerArm64, err := cli.ImagePull(ctx, "node:16-buster-slim", types.ImagePullOptions{
|
|
Platform: "linux/arm64",
|
|
})
|
|
assert.Nil(t, err)
|
|
defer readerArm64.Close()
|
|
_, err = io.ReadAll(readerArm64)
|
|
assert.Nil(t, err)
|
|
|
|
imageArm64Exists, err := ImageExistsLocally(ctx, "node:16-buster-slim", "linux/arm64")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, true, imageArm64Exists)
|
|
}
|