github.com/zhouyu0/docker-note@v0.0.0-20190722021225-b8d3825084db/client/image_remove_test.go (about)

     1  package client // import "github.com/docker/docker/client"
     2  
     3  import (
     4  	"bytes"
     5  	"context"
     6  	"encoding/json"
     7  	"fmt"
     8  	"io/ioutil"
     9  	"net/http"
    10  	"strings"
    11  	"testing"
    12  
    13  	"github.com/docker/docker/api/types"
    14  	"gotest.tools/assert"
    15  	is "gotest.tools/assert/cmp"
    16  )
    17  
    18  func TestImageRemoveError(t *testing.T) {
    19  	client := &Client{
    20  		client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
    21  	}
    22  
    23  	_, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{})
    24  	assert.Check(t, is.Error(err, "Error response from daemon: Server error"))
    25  }
    26  
    27  func TestImageRemoveImageNotFound(t *testing.T) {
    28  	client := &Client{
    29  		client: newMockClient(errorMock(http.StatusNotFound, "missing")),
    30  	}
    31  
    32  	_, err := client.ImageRemove(context.Background(), "unknown", types.ImageRemoveOptions{})
    33  	assert.Check(t, is.Error(err, "Error: No such image: unknown"))
    34  	assert.Check(t, IsErrNotFound(err))
    35  }
    36  
    37  func TestImageRemove(t *testing.T) {
    38  	expectedURL := "/images/image_id"
    39  	removeCases := []struct {
    40  		force               bool
    41  		pruneChildren       bool
    42  		expectedQueryParams map[string]string
    43  	}{
    44  		{
    45  			force:         false,
    46  			pruneChildren: false,
    47  			expectedQueryParams: map[string]string{
    48  				"force":   "",
    49  				"noprune": "1",
    50  			},
    51  		}, {
    52  			force:         true,
    53  			pruneChildren: true,
    54  			expectedQueryParams: map[string]string{
    55  				"force":   "1",
    56  				"noprune": "",
    57  			},
    58  		},
    59  	}
    60  	for _, removeCase := range removeCases {
    61  		client := &Client{
    62  			client: newMockClient(func(req *http.Request) (*http.Response, error) {
    63  				if !strings.HasPrefix(req.URL.Path, expectedURL) {
    64  					return nil, fmt.Errorf("expected URL '%s', got '%s'", expectedURL, req.URL)
    65  				}
    66  				if req.Method != "DELETE" {
    67  					return nil, fmt.Errorf("expected DELETE method, got %s", req.Method)
    68  				}
    69  				query := req.URL.Query()
    70  				for key, expected := range removeCase.expectedQueryParams {
    71  					actual := query.Get(key)
    72  					if actual != expected {
    73  						return nil, fmt.Errorf("%s not set in URL query properly. Expected '%s', got %s", key, expected, actual)
    74  					}
    75  				}
    76  				b, err := json.Marshal([]types.ImageDeleteResponseItem{
    77  					{
    78  						Untagged: "image_id1",
    79  					},
    80  					{
    81  						Deleted: "image_id",
    82  					},
    83  				})
    84  				if err != nil {
    85  					return nil, err
    86  				}
    87  
    88  				return &http.Response{
    89  					StatusCode: http.StatusOK,
    90  					Body:       ioutil.NopCloser(bytes.NewReader(b)),
    91  				}, nil
    92  			}),
    93  		}
    94  		imageDeletes, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{
    95  			Force:         removeCase.force,
    96  			PruneChildren: removeCase.pruneChildren,
    97  		})
    98  		if err != nil {
    99  			t.Fatal(err)
   100  		}
   101  		if len(imageDeletes) != 2 {
   102  			t.Fatalf("expected 2 deleted images, got %v", imageDeletes)
   103  		}
   104  	}
   105  }