swarm_leave_test.go 1.5 KB

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