mirror of https://github.com/docker/cli.git
74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
|
package manifest
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
manifesttypes "github.com/docker/cli/cli/manifest/types"
|
||
|
|
"github.com/docker/cli/internal/test"
|
||
|
|
"github.com/docker/cli/internal/test/testutil"
|
||
|
|
"github.com/docker/distribution/reference"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"golang.org/x/net/context"
|
||
|
|
)
|
||
|
|
|
||
|
|
func newFakeRegistryClient(t *testing.T) *fakeRegistryClient {
|
||
|
|
require.NoError(t, nil)
|
||
|
|
|
||
|
|
return &fakeRegistryClient{
|
||
|
|
getManifestFunc: func(_ context.Context, _ reference.Named) (manifesttypes.ImageManifest, error) {
|
||
|
|
return manifesttypes.ImageManifest{}, errors.New("")
|
||
|
|
},
|
||
|
|
getManifestListFunc: func(_ context.Context, _ reference.Named) ([]manifesttypes.ImageManifest, error) {
|
||
|
|
return nil, errors.Errorf("")
|
||
|
|
},
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestManifestPushErrors(t *testing.T) {
|
||
|
|
testCases := []struct {
|
||
|
|
args []string
|
||
|
|
expectedError string
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
args: []string{"one-arg", "extra-arg"},
|
||
|
|
expectedError: "requires exactly 1 argument",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
args: []string{"th!si'sa/fa!ke/li$t/-name"},
|
||
|
|
expectedError: "invalid reference format",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tc := range testCases {
|
||
|
|
cli := test.NewFakeCli(nil)
|
||
|
|
cmd := newPushListCommand(cli)
|
||
|
|
cmd.SetArgs(tc.args)
|
||
|
|
cmd.SetOutput(ioutil.Discard)
|
||
|
|
testutil.ErrorContains(t, cmd.Execute(), tc.expectedError)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// store a one-image manifest list and puah it
|
||
|
|
func TestManifestPush(t *testing.T) {
|
||
|
|
store, sCleanup := newTempManifestStore(t)
|
||
|
|
defer sCleanup()
|
||
|
|
|
||
|
|
registry := newFakeRegistryClient(t)
|
||
|
|
|
||
|
|
cli := test.NewFakeCli(nil)
|
||
|
|
cli.SetManifestStore(store)
|
||
|
|
cli.SetRegistryClient(registry)
|
||
|
|
|
||
|
|
namedRef := ref(t, "alpine:3.0")
|
||
|
|
imageManifest := fullImageManifest(t, namedRef)
|
||
|
|
err := store.Save(ref(t, "list:v1"), namedRef, imageManifest)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
cmd := newPushListCommand(cli)
|
||
|
|
cmd.SetArgs([]string{"example.com/list:v1"})
|
||
|
|
err = cmd.Execute()
|
||
|
|
require.NoError(t, err)
|
||
|
|
}
|