package client
 
 import (
 	"bytes"
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
 	"net/http"
 	"strings"
 	"testing"
 
 	"github.com/docker/docker/api/types"

 	"github.com/stretchr/testify/assert"

 	"golang.org/x/net/context"
 )
 
 func TestImageRemoveError(t *testing.T) {
 	client := &Client{

 		client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),

 	}
 
 	_, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{})

 	assert.EqualError(t, err, "Error response from daemon: Server error")

 }
 

 func TestImageRemoveImageNotFound(t *testing.T) {
 	client := &Client{

 		client: newMockClient(errorMock(http.StatusNotFound, "missing")),

 	}
 
 	_, err := client.ImageRemove(context.Background(), "unknown", types.ImageRemoveOptions{})

 	assert.EqualError(t, err, "Error: No such image: unknown")
 	assert.True(t, IsErrNotFound(err))

 }
 

 func TestImageRemove(t *testing.T) {
 	expectedURL := "/images/image_id"
 	removeCases := []struct {
 		force               bool
 		pruneChildren       bool
 		expectedQueryParams map[string]string
 	}{
 		{
 			force:         false,
 			pruneChildren: false,
 			expectedQueryParams: map[string]string{
 				"force":   "",
 				"noprune": "1",
 			},
 		}, {
 			force:         true,
 			pruneChildren: true,
 			expectedQueryParams: map[string]string{
 				"force":   "1",
 				"noprune": "",
 			},
 		},
 	}
 	for _, removeCase := range removeCases {
 		client := &Client{

 			client: newMockClient(func(req *http.Request) (*http.Response, error) {

 				if !strings.HasPrefix(req.URL.Path, expectedURL) {
 					return nil, fmt.Errorf("expected URL '%s', got '%s'", expectedURL, req.URL)
 				}
 				if req.Method != "DELETE" {
 					return nil, fmt.Errorf("expected DELETE method, got %s", req.Method)
 				}
 				query := req.URL.Query()
 				for key, expected := range removeCase.expectedQueryParams {
 					actual := query.Get(key)
 					if actual != expected {
 						return nil, fmt.Errorf("%s not set in URL query properly. Expected '%s', got %s", key, expected, actual)
 					}
 				}

 				b, err := json.Marshal([]types.ImageDeleteResponseItem{

 					{
 						Untagged: "image_id1",
 					},
 					{
 						Deleted: "image_id",
 					},
 				})
 				if err != nil {
 					return nil, err
 				}
 
 				return &http.Response{
 					StatusCode: http.StatusOK,
 					Body:       ioutil.NopCloser(bytes.NewReader(b)),
 				}, nil
 			}),
 		}
 		imageDeletes, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{
 			Force:         removeCase.force,
 			PruneChildren: removeCase.pruneChildren,
 		})
 		if err != nil {
 			t.Fatal(err)
 		}
 		if len(imageDeletes) != 2 {
 			t.Fatalf("expected 2 deleted images, got %v", imageDeletes)
 		}
 	}
 }