github.com/xdlianrong208/docker-ce-comments@v17.12.1-ce-rc2+incompatible/components/engine/client/service_remove_test.go (about) 1 package client 2 3 import ( 4 "bytes" 5 "fmt" 6 "io/ioutil" 7 "net/http" 8 "strings" 9 "testing" 10 11 "github.com/stretchr/testify/assert" 12 "golang.org/x/net/context" 13 ) 14 15 func TestServiceRemoveError(t *testing.T) { 16 client := &Client{ 17 client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), 18 } 19 20 err := client.ServiceRemove(context.Background(), "service_id") 21 assert.EqualError(t, err, "Error response from daemon: Server error") 22 } 23 24 func TestServiceRemoveNotFoundError(t *testing.T) { 25 client := &Client{ 26 client: newMockClient(errorMock(http.StatusNotFound, "missing")), 27 } 28 29 err := client.ServiceRemove(context.Background(), "service_id") 30 assert.EqualError(t, err, "Error: No such service: service_id") 31 assert.True(t, IsErrNotFound(err)) 32 } 33 34 func TestServiceRemove(t *testing.T) { 35 expectedURL := "/services/service_id" 36 37 client := &Client{ 38 client: newMockClient(func(req *http.Request) (*http.Response, error) { 39 if !strings.HasPrefix(req.URL.Path, expectedURL) { 40 return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL) 41 } 42 if req.Method != "DELETE" { 43 return nil, fmt.Errorf("expected DELETE method, got %s", req.Method) 44 } 45 return &http.Response{ 46 StatusCode: http.StatusOK, 47 Body: ioutil.NopCloser(bytes.NewReader([]byte("body"))), 48 }, nil 49 }), 50 } 51 52 err := client.ServiceRemove(context.Background(), "service_id") 53 if err != nil { 54 t.Fatal(err) 55 } 56 }