github.com/zhouyu0/docker-note@v0.0.0-20190722021225-b8d3825084db/client/network_remove_test.go (about) 1 package client // import "github.com/docker/docker/client" 2 3 import ( 4 "bytes" 5 "context" 6 "fmt" 7 "io/ioutil" 8 "net/http" 9 "strings" 10 "testing" 11 ) 12 13 func TestNetworkRemoveError(t *testing.T) { 14 client := &Client{ 15 client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), 16 } 17 18 err := client.NetworkRemove(context.Background(), "network_id") 19 if err == nil || err.Error() != "Error response from daemon: Server error" { 20 t.Fatalf("expected a Server Error, got %v", err) 21 } 22 } 23 24 func TestNetworkRemove(t *testing.T) { 25 expectedURL := "/networks/network_id" 26 27 client := &Client{ 28 client: newMockClient(func(req *http.Request) (*http.Response, error) { 29 if !strings.HasPrefix(req.URL.Path, expectedURL) { 30 return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL) 31 } 32 if req.Method != "DELETE" { 33 return nil, fmt.Errorf("expected DELETE method, got %s", req.Method) 34 } 35 return &http.Response{ 36 StatusCode: http.StatusOK, 37 Body: ioutil.NopCloser(bytes.NewReader([]byte("body"))), 38 }, nil 39 }), 40 } 41 42 err := client.NetworkRemove(context.Background(), "network_id") 43 if err != nil { 44 t.Fatal(err) 45 } 46 }