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