mirror of https://github.com/docker/cli.git
120 lines
3.1 KiB
Go
120 lines
3.1 KiB
Go
package network
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/docker/cli/internal/test"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/network"
|
|
"github.com/docker/docker/errdefs"
|
|
"github.com/pkg/errors"
|
|
"gotest.tools/v3/assert"
|
|
is "gotest.tools/v3/assert/cmp"
|
|
)
|
|
|
|
func TestNetworkRemoveForce(t *testing.T) {
|
|
tests := []struct {
|
|
doc string
|
|
args []string
|
|
expectedErr string
|
|
}{
|
|
{
|
|
doc: "existing network",
|
|
args: []string{"existing-network"},
|
|
},
|
|
{
|
|
doc: "existing network (forced)",
|
|
args: []string{"--force", "existing-network"},
|
|
},
|
|
{
|
|
doc: "non-existing network",
|
|
args: []string{"no-such-network"},
|
|
expectedErr: "no such network: no-such-network",
|
|
},
|
|
{
|
|
doc: "non-existing network (forced)",
|
|
args: []string{"--force", "no-such-network"},
|
|
},
|
|
{
|
|
doc: "in-use network",
|
|
args: []string{"in-use-network"},
|
|
expectedErr: "network is in use",
|
|
},
|
|
{
|
|
doc: "in-use network (forced)",
|
|
args: []string{"--force", "in-use-network"},
|
|
expectedErr: "network is in use",
|
|
},
|
|
{
|
|
doc: "multiple networks",
|
|
args: []string{"existing-network", "no-such-network"},
|
|
expectedErr: "no such network: no-such-network",
|
|
},
|
|
{
|
|
doc: "multiple networks (forced)",
|
|
args: []string{"--force", "existing-network", "no-such-network"},
|
|
},
|
|
{
|
|
doc: "multiple networks 2 (forced)",
|
|
args: []string{"--force", "existing-network", "no-such-network", "in-use-network"},
|
|
expectedErr: "network is in use",
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
tc := tc
|
|
t.Run(tc.doc, func(t *testing.T) {
|
|
fakeCli := test.NewFakeCli(&fakeClient{
|
|
networkRemoveFunc: func(ctx context.Context, networkID string) error {
|
|
switch networkID {
|
|
case "no-such-network":
|
|
return errdefs.NotFound(errors.New("no such network: no-such-network"))
|
|
case "in-use-network":
|
|
return errdefs.Forbidden(errors.New("network is in use"))
|
|
case "existing-network":
|
|
return nil
|
|
default:
|
|
return nil
|
|
}
|
|
},
|
|
})
|
|
|
|
cmd := newRemoveCommand(fakeCli)
|
|
cmd.SetOut(io.Discard)
|
|
cmd.SetErr(fakeCli.ErrBuffer())
|
|
cmd.SetArgs(tc.args)
|
|
|
|
err := cmd.Execute()
|
|
if tc.expectedErr == "" {
|
|
assert.NilError(t, err)
|
|
} else {
|
|
assert.Check(t, is.Contains(fakeCli.ErrBuffer().String(), tc.expectedErr))
|
|
assert.ErrorContains(t, err, "Code: 1")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNetworkRemovePromptTermination(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
t.Cleanup(cancel)
|
|
|
|
cli := test.NewFakeCli(&fakeClient{
|
|
networkRemoveFunc: func(ctx context.Context, networkID string) error {
|
|
return errors.New("fakeClient networkRemoveFunc should not be called")
|
|
},
|
|
networkInspectFunc: func(ctx context.Context, networkID string, options network.InspectOptions) (types.NetworkResource, []byte, error) {
|
|
return types.NetworkResource{
|
|
ID: "existing-network",
|
|
Name: "existing-network",
|
|
Ingress: true,
|
|
}, nil, nil
|
|
},
|
|
})
|
|
cmd := newRemoveCommand(cli)
|
|
cmd.SetArgs([]string{"existing-network"})
|
|
test.TerminatePrompt(ctx, t, cmd, cli)
|
|
}
|