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