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  }