mirror of https://github.com/docker/cli.git
44 lines
1.0 KiB
Go
44 lines
1.0 KiB
Go
|
package containerizedengine
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/containerd/containerd"
|
||
|
|
"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, &testOutStream{}, &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, &testOutStream{}, &types.AuthConfig{})
|
||
|
|
assert.NilError(t, err)
|
||
|
|
}
|