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 }