123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112 |
- package client
- import (
- "bytes"
- "encoding/json"
- "fmt"
- "io/ioutil"
- "net/http"
- "strings"
- "testing"
- "github.com/docker/docker/api/types"
- "github.com/docker/docker/api/types/filters"
- "github.com/stretchr/testify/assert"
- "golang.org/x/net/context"
- )
- func TestNetworksPruneError(t *testing.T) {
- client := &Client{
- client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
- version: "1.25",
- }
- filters := filters.NewArgs()
- _, err := client.NetworksPrune(context.Background(), filters)
- if err == nil || err.Error() != "Error response from daemon: Server error" {
- t.Fatalf("expected a Server Error, got %v", err)
- }
- }
- func TestNetworksPrune(t *testing.T) {
- expectedURL := "/v1.25/networks/prune"
- danglingFilters := filters.NewArgs()
- danglingFilters.Add("dangling", "true")
- noDanglingFilters := filters.NewArgs()
- noDanglingFilters.Add("dangling", "false")
- labelFilters := filters.NewArgs()
- labelFilters.Add("dangling", "true")
- labelFilters.Add("label", "label1=foo")
- labelFilters.Add("label", "label2!=bar")
- listCases := []struct {
- filters filters.Args
- expectedQueryParams map[string]string
- }{
- {
- filters: filters.Args{},
- expectedQueryParams: map[string]string{
- "until": "",
- "filter": "",
- "filters": "",
- },
- },
- {
- filters: danglingFilters,
- expectedQueryParams: map[string]string{
- "until": "",
- "filter": "",
- "filters": `{"dangling":{"true":true}}`,
- },
- },
- {
- filters: noDanglingFilters,
- expectedQueryParams: map[string]string{
- "until": "",
- "filter": "",
- "filters": `{"dangling":{"false":true}}`,
- },
- },
- {
- filters: labelFilters,
- expectedQueryParams: map[string]string{
- "until": "",
- "filter": "",
- "filters": `{"dangling":{"true":true},"label":{"label1=foo":true,"label2!=bar":true}}`,
- },
- },
- }
- for _, listCase := range listCases {
- 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)
- }
- query := req.URL.Query()
- for key, expected := range listCase.expectedQueryParams {
- actual := query.Get(key)
- assert.Equal(t, expected, actual)
- }
- content, err := json.Marshal(types.NetworksPruneReport{
- NetworksDeleted: []string{"network_id1", "network_id2"},
- })
- if err != nil {
- return nil, err
- }
- return &http.Response{
- StatusCode: http.StatusOK,
- Body: ioutil.NopCloser(bytes.NewReader(content)),
- }, nil
- }),
- version: "1.25",
- }
- report, err := client.NetworksPrune(context.Background(), listCase.filters)
- assert.NoError(t, err)
- assert.Len(t, report.NetworksDeleted, 2)
- }
- }
|