github.com/zhouyu0/docker-note@v0.0.0-20190722021225-b8d3825084db/client/node_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 "github.com/docker/docker/api/types" 13 ) 14 15 func TestNodeRemoveError(t *testing.T) { 16 client := &Client{ 17 client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")), 18 } 19 20 err := client.NodeRemove(context.Background(), "node_id", types.NodeRemoveOptions{Force: false}) 21 if err == nil || err.Error() != "Error response from daemon: Server error" { 22 t.Fatalf("expected a Server Error, got %v", err) 23 } 24 } 25 26 func TestNodeRemove(t *testing.T) { 27 expectedURL := "/nodes/node_id" 28 29 removeCases := []struct { 30 force bool 31 expectedForce string 32 }{ 33 { 34 expectedForce: "", 35 }, 36 { 37 force: true, 38 expectedForce: "1", 39 }, 40 } 41 42 for _, removeCase := range removeCases { 43 client := &Client{ 44 client: newMockClient(func(req *http.Request) (*http.Response, error) { 45 if !strings.HasPrefix(req.URL.Path, expectedURL) { 46 return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL) 47 } 48 if req.Method != "DELETE" { 49 return nil, fmt.Errorf("expected DELETE method, got %s", req.Method) 50 } 51 force := req.URL.Query().Get("force") 52 if force != removeCase.expectedForce { 53 return nil, fmt.Errorf("force not set in URL query properly. expected '%s', got %s", removeCase.expectedForce, force) 54 } 55 56 return &http.Response{ 57 StatusCode: http.StatusOK, 58 Body: ioutil.NopCloser(bytes.NewReader([]byte("body"))), 59 }, nil 60 }), 61 } 62 63 err := client.NodeRemove(context.Background(), "node_id", types.NodeRemoveOptions{Force: removeCase.force}) 64 if err != nil { 65 t.Fatal(err) 66 } 67 } 68 }