package node import ( "bytes" "io/ioutil" "testing" "github.com/docker/docker/cli/internal/test" "github.com/docker/docker/pkg/testutil/assert" "github.com/pkg/errors" ) func TestNodeRemoveErrors(t *testing.T) { testCases := []struct { args []string nodeRemoveFunc func() error expectedError string }{ { expectedError: "requires at least 1 argument", }, { args: []string{"nodeID"}, nodeRemoveFunc: func() error { return errors.Errorf("error removing the node") }, expectedError: "error removing the node", }, } for _, tc := range testCases { buf := new(bytes.Buffer) cmd := newRemoveCommand( test.NewFakeCli(&fakeClient{ nodeRemoveFunc: tc.nodeRemoveFunc, }, buf)) cmd.SetArgs(tc.args) cmd.SetOutput(ioutil.Discard) assert.Error(t, cmd.Execute(), tc.expectedError) } } func TestNodeRemoveMultiple(t *testing.T) { buf := new(bytes.Buffer) cmd := newRemoveCommand(test.NewFakeCli(&fakeClient{}, buf)) cmd.SetArgs([]string{"nodeID1", "nodeID2"}) assert.NilError(t, cmd.Execute()) }