mirror of https://code.forgejo.org/forgejo/act
68 lines
2.0 KiB
Go
68 lines
2.0 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)
|
|
}
|