6c258cf40d
* Add QEMU to run different architectures * Update dependencies in `go.mod` * Add `--container-architecture` flag to specify custom image architecture Co-authored-by: Casey Lee <cplee@nektos.com>
67 lines
2 KiB
Go
67 lines
2 KiB
Go
package container
|
|
|
|
import (
|
|
"context"
|
|
"io/ioutil"
|
|
"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, "alpine:latest", types.ImagePullOptions{
|
|
Platform: "linux/amd64",
|
|
})
|
|
assert.Nil(t, err)
|
|
defer readerDefault.Close()
|
|
_, err = ioutil.ReadAll(readerDefault)
|
|
assert.Nil(t, err)
|
|
|
|
imageDefaultArchExists, err := ImageExistsLocally(ctx, "alpine:latest", "linux/amd64")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, true, imageDefaultArchExists)
|
|
|
|
// Validate if another architecture platform can be pulled
|
|
readerArm64, err := cli.ImagePull(ctx, "alpine:latest", types.ImagePullOptions{
|
|
Platform: "linux/arm64",
|
|
})
|
|
assert.Nil(t, err)
|
|
defer readerArm64.Close()
|
|
_, err = ioutil.ReadAll(readerArm64)
|
|
assert.Nil(t, err)
|
|
|
|
imageArm64Exists, err := ImageExistsLocally(ctx, "alpine:latest", "linux/arm64")
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, true, imageArm64Exists)
|
|
}
|