github.com/zhouyu0/docker-note@v0.0.0-20190722021225-b8d3825084db/client/service_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 "gotest.tools/assert" 13 is "gotest.tools/assert/cmp" 14 ) 15 16 func TestServiceRemoveError(t *testing.T) { 17 client := &Client{ 18 client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), 19 } 20 21 err := client.ServiceRemove(context.Background(), "service_id") 22 assert.Check(t, is.Error(err, "Error response from daemon: Server error")) 23 } 24 25 func TestServiceRemoveNotFoundError(t *testing.T) { 26 client := &Client{ 27 client: newMockClient(errorMock(http.StatusNotFound, "missing")), 28 } 29 30 err := client.ServiceRemove(context.Background(), "service_id") 31 assert.Check(t, is.Error(err, "Error: No such service: service_id")) 32 assert.Check(t, IsErrNotFound(err)) 33 } 34 35 func TestServiceRemove(t *testing.T) { 36 expectedURL := "/services/service_id" 37 38 client := &Client{ 39 client: newMockClient(func(req *http.Request) (*http.Response, error) { 40 if !strings.HasPrefix(req.URL.Path, expectedURL) { 41 return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL) 42 } 43 if req.Method != "DELETE" { 44 return nil, fmt.Errorf("expected DELETE method, got %s", req.Method) 45 } 46 return &http.Response{ 47 StatusCode: http.StatusOK, 48 Body: ioutil.NopCloser(bytes.NewReader([]byte("body"))), 49 }, nil 50 }), 51 } 52 53 err := client.ServiceRemove(context.Background(), "service_id") 54 if err != nil { 55 t.Fatal(err) 56 } 57 }