node_remove_test.go 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. package client
  2. import (
  3. "bytes"
  4. "fmt"
  5. "io/ioutil"
  6. "net/http"
  7. "strings"
  8. "testing"
  9. "github.com/docker/docker/api/types"
  10. "golang.org/x/net/context"
  11. )
  12. func TestNodeRemoveError(t *testing.T) {
  13. client := &Client{
  14. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  15. }
  16. err := client.NodeRemove(context.Background(), "node_id", types.NodeRemoveOptions{Force: false})
  17. if err == nil || err.Error() != "Error response from daemon: Server error" {
  18. t.Fatalf("expected a Server Error, got %v", err)
  19. }
  20. }
  21. func TestNodeRemove(t *testing.T) {
  22. expectedURL := "/nodes/node_id"
  23. removeCases := []struct {
  24. force bool
  25. expectedForce string
  26. }{
  27. {
  28. expectedForce: "",
  29. },
  30. {
  31. force: true,
  32. expectedForce: "1",
  33. },
  34. }
  35. for _, removeCase := range removeCases {
  36. client := &Client{
  37. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  38. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  39. return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  40. }
  41. if req.Method != "DELETE" {
  42. return nil, fmt.Errorf("expected DELETE method, got %s", req.Method)
  43. }
  44. force := req.URL.Query().Get("force")
  45. if force != removeCase.expectedForce {
  46. return nil, fmt.Errorf("force not set in URL query properly. expected '%s', got %s", removeCase.expectedForce, force)
  47. }
  48. return &http.Response{
  49. StatusCode: http.StatusOK,
  50. Body: ioutil.NopCloser(bytes.NewReader([]byte("body"))),
  51. }, nil
  52. }),
  53. }
  54. err := client.NodeRemove(context.Background(), "node_id", types.NodeRemoveOptions{Force: removeCase.force})
  55. if err != nil {
  56. t.Fatal(err)
  57. }
  58. }
  59. }