mirror of https://github.com/docker/cli.git
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package containerizedengine
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/containerd/containerd"
|
|
"github.com/docker/cli/cli/streams"
|
|
"github.com/docker/docker/api/types"
|
|
"gotest.tools/assert"
|
|
)
|
|
|
|
func TestPullWithAuthPullFail(t *testing.T) {
|
|
ctx := context.Background()
|
|
client := baseClient{
|
|
cclient: &fakeContainerdClient{
|
|
pullFunc: func(ctx context.Context, ref string, opts ...containerd.RemoteOpt) (containerd.Image, error) {
|
|
return nil, fmt.Errorf("pull failure")
|
|
|
|
},
|
|
},
|
|
}
|
|
imageName := "testnamegoeshere"
|
|
|
|
_, err := client.pullWithAuth(ctx, imageName, streams.NewOut(&bytes.Buffer{}), &types.AuthConfig{})
|
|
assert.ErrorContains(t, err, "pull failure")
|
|
}
|
|
|
|
func TestPullWithAuthPullPass(t *testing.T) {
|
|
ctx := context.Background()
|
|
client := baseClient{
|
|
cclient: &fakeContainerdClient{
|
|
pullFunc: func(ctx context.Context, ref string, opts ...containerd.RemoteOpt) (containerd.Image, error) {
|
|
return nil, nil
|
|
|
|
},
|
|
},
|
|
}
|
|
imageName := "testnamegoeshere"
|
|
|
|
_, err := client.pullWithAuth(ctx, imageName, streams.NewOut(&bytes.Buffer{}), &types.AuthConfig{})
|
|
assert.NilError(t, err)
|
|
}
|