package service import ( "fmt" "io/ioutil" "strings" "testing" "github.com/docker/cli/internal/test" "github.com/docker/cli/internal/test/testutil" "github.com/docker/docker/api/types" "github.com/docker/docker/api/types/swarm" "github.com/stretchr/testify/assert" "golang.org/x/net/context" ) func TestRollback(t *testing.T) { testCases := []struct { name string args []string serviceUpdateFunc func(ctx context.Context, serviceID string, version swarm.Version, service swarm.ServiceSpec, options types.ServiceUpdateOptions) (types.ServiceUpdateResponse, error) expectedDockerCliErr string }{ { name: "rollback-service", args: []string{"service-id"}, }, { name: "rollback-service-with-warnings", args: []string{"service-id"}, serviceUpdateFunc: func(ctx context.Context, serviceID string, version swarm.Version, service swarm.ServiceSpec, options types.ServiceUpdateOptions) (types.ServiceUpdateResponse, error) { response := types.ServiceUpdateResponse{} response.Warnings = []string{ "- warning 1", "- warning 2", } return response, nil }, expectedDockerCliErr: "- warning 1\n- warning 2", }, } for _, tc := range testCases { cli := test.NewFakeCli(&fakeClient{ serviceUpdateFunc: tc.serviceUpdateFunc, }) cmd := newRollbackCommand(cli) cmd.SetArgs(tc.args) cmd.Flags().Set("quiet", "true") cmd.SetOutput(ioutil.Discard) assert.NoError(t, cmd.Execute()) assert.Equal(t, strings.TrimSpace(cli.ErrBuffer().String()), tc.expectedDockerCliErr) } } func TestRollbackWithErrors(t *testing.T) { testCases := []struct { name string args []string serviceInspectWithRawFunc func(ctx context.Context, serviceID string, options types.ServiceInspectOptions) (swarm.Service, []byte, error) serviceUpdateFunc func(ctx context.Context, serviceID string, version swarm.Version, service swarm.ServiceSpec, options types.ServiceUpdateOptions) (types.ServiceUpdateResponse, error) expectedError string }{ { name: "not-enough-args", expectedError: "requires exactly 1 argument", }, { name: "too-many-args", args: []string{"service-id-1", "service-id-2"}, expectedError: "requires exactly 1 argument", }, { name: "service-does-not-exists", args: []string{"service-id"}, serviceInspectWithRawFunc: func(ctx context.Context, serviceID string, options types.ServiceInspectOptions) (swarm.Service, []byte, error) { return swarm.Service{}, []byte{}, fmt.Errorf("no such services: %s", serviceID) }, expectedError: "no such services: service-id", }, { name: "service-update-failed", args: []string{"service-id"}, serviceUpdateFunc: func(ctx context.Context, serviceID string, version swarm.Version, service swarm.ServiceSpec, options types.ServiceUpdateOptions) (types.ServiceUpdateResponse, error) { return types.ServiceUpdateResponse{}, fmt.Errorf("no such services: %s", serviceID) }, expectedError: "no such services: service-id", }, } for _, tc := range testCases { cmd := newRollbackCommand( test.NewFakeCli(&fakeClient{ serviceInspectWithRawFunc: tc.serviceInspectWithRawFunc, serviceUpdateFunc: tc.serviceUpdateFunc, })) cmd.SetArgs(tc.args) cmd.Flags().Set("quiet", "true") cmd.SetOutput(ioutil.Discard) testutil.ErrorContains(t, cmd.Execute(), tc.expectedError) } }