mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-20 22:00:23 -04:00
105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
![]() |
package client
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"encoding/json"
|
||
|
"fmt"
|
||
|
"io/ioutil"
|
||
|
"net/http"
|
||
|
"strings"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/docker/docker/api/types"
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
"golang.org/x/net/context"
|
||
|
)
|
||
|
|
||
|
func TestImageRemoveError(t *testing.T) {
|
||
|
client := &Client{
|
||
|
client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
|
||
|
}
|
||
|
|
||
|
_, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{})
|
||
|
assert.EqualError(t, err, "Error response from daemon: Server error")
|
||
|
}
|
||
|
|
||
|
func TestImageRemoveImageNotFound(t *testing.T) {
|
||
|
client := &Client{
|
||
|
client: newMockClient(errorMock(http.StatusNotFound, "missing")),
|
||
|
}
|
||
|
|
||
|
_, err := client.ImageRemove(context.Background(), "unknown", types.ImageRemoveOptions{})
|
||
|
assert.EqualError(t, err, "Error: No such image: unknown")
|
||
|
assert.True(t, IsErrNotFound(err))
|
||
|
}
|
||
|
|
||
|
func TestImageRemove(t *testing.T) {
|
||
|
expectedURL := "/images/image_id"
|
||
|
removeCases := []struct {
|
||
|
force bool
|
||
|
pruneChildren bool
|
||
|
expectedQueryParams map[string]string
|
||
|
}{
|
||
|
{
|
||
|
force: false,
|
||
|
pruneChildren: false,
|
||
|
expectedQueryParams: map[string]string{
|
||
|
"force": "",
|
||
|
"noprune": "1",
|
||
|
},
|
||
|
}, {
|
||
|
force: true,
|
||
|
pruneChildren: true,
|
||
|
expectedQueryParams: map[string]string{
|
||
|
"force": "1",
|
||
|
"noprune": "",
|
||
|
},
|
||
|
},
|
||
|
}
|
||
|
for _, removeCase := range removeCases {
|
||
|
client := &Client{
|
||
|
client: newMockClient(func(req *http.Request) (*http.Response, error) {
|
||
|
if !strings.HasPrefix(req.URL.Path, expectedURL) {
|
||
|
return nil, fmt.Errorf("expected URL '%s', got '%s'", expectedURL, req.URL)
|
||
|
}
|
||
|
if req.Method != "DELETE" {
|
||
|
return nil, fmt.Errorf("expected DELETE method, got %s", req.Method)
|
||
|
}
|
||
|
query := req.URL.Query()
|
||
|
for key, expected := range removeCase.expectedQueryParams {
|
||
|
actual := query.Get(key)
|
||
|
if actual != expected {
|
||
|
return nil, fmt.Errorf("%s not set in URL query properly. Expected '%s', got %s", key, expected, actual)
|
||
|
}
|
||
|
}
|
||
|
b, err := json.Marshal([]types.ImageDeleteResponseItem{
|
||
|
{
|
||
|
Untagged: "image_id1",
|
||
|
},
|
||
|
{
|
||
|
Deleted: "image_id",
|
||
|
},
|
||
|
})
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
return &http.Response{
|
||
|
StatusCode: http.StatusOK,
|
||
|
Body: ioutil.NopCloser(bytes.NewReader(b)),
|
||
|
}, nil
|
||
|
}),
|
||
|
}
|
||
|
imageDeletes, err := client.ImageRemove(context.Background(), "image_id", types.ImageRemoveOptions{
|
||
|
Force: removeCase.force,
|
||
|
PruneChildren: removeCase.pruneChildren,
|
||
|
})
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
if len(imageDeletes) != 2 {
|
||
|
t.Fatalf("expected 2 deleted images, got %v", imageDeletes)
|
||
|
}
|
||
|
}
|
||
|
}
|