Function TestNewRemoveCommandSuccess

cli/command/image/remove_test.go:79–144

(t *testing.T)

}

func TestNewRemoveCommandSuccess(t *testing.T) {
	testCases := []struct {
		name            string
		args            []string
		imageRemoveFunc func(img string, options client.ImageRemoveOptions) (client.ImageRemoveResult, error)
		expectedStderr  string
	}{
		{
			name: "Image Deleted",
			args: []string{"image1"},
			imageRemoveFunc: func(img string, options client.ImageRemoveOptions) (client.ImageRemoveResult, error) {
				assert.Check(t, is.Equal("image1", img))
				return client.ImageRemoveResult{
					Items: []image.DeleteResponse{{Deleted: img}},
				}, nil
			},
		},
		{
			name: "Image not found with force option",
			args: []string{"-f", "image1"},
			imageRemoveFunc: func(img string, options client.ImageRemoveOptions) (client.ImageRemoveResult, error) {
				assert.Check(t, is.Equal("image1", img))
				assert.Check(t, is.Equal(true, options.Force))
				return client.ImageRemoveResult{}, notFound{"image1"}
			},
			expectedStderr: "Error: No such image: image1\n",
		},

		{
			name: "Image Untagged",
			args: []string{"image1"},
			imageRemoveFunc: func(img string, options client.ImageRemoveOptions) (client.ImageRemoveResult, error) {
				assert.Check(t, is.Equal("image1", img))
				return client.ImageRemoveResult{
					Items: []image.DeleteResponse{{Untagged: img}},
				}, nil
			},
		},
		{
			name: "Image Deleted and Untagged",
			args: []string{"image1", "image2"},
			imageRemoveFunc: func(img string, options client.ImageRemoveOptions) (client.ImageRemoveResult, error) {
				if img == "image1" {
					return client.ImageRemoveResult{
						Items: []image.DeleteResponse{{Untagged: img}},
					}, nil
				}
				return client.ImageRemoveResult{
					Items: []image.DeleteResponse{{Deleted: img}},
				}, nil
			},
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			cli := test.NewFakeCli(&fakeClient{imageRemoveFunc: tc.imageRemoveFunc})
			cmd := newRemoveCommand(cli)
			cmd.SetOut(io.Discard)