swarm_join_test.go 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051
  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. "github.com/docker/docker/api/types/swarm"
  11. )
  12. func TestSwarmJoinError(t *testing.T) {
  13. client := &Client{
  14. client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  15. }
  16. err := client.SwarmJoin(context.Background(), swarm.JoinRequest{})
  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 TestSwarmJoin(t *testing.T) {
  22. expectedURL := "/swarm/join"
  23. client := &Client{
  24. client: newMockClient(func(req *http.Request) (*http.Response, error) {
  25. if !strings.HasPrefix(req.URL.Path, expectedURL) {
  26. return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  27. }
  28. if req.Method != "POST" {
  29. return nil, fmt.Errorf("expected POST method, got %s", req.Method)
  30. }
  31. return &http.Response{
  32. StatusCode: http.StatusOK,
  33. Body: ioutil.NopCloser(bytes.NewReader([]byte(""))),
  34. }, nil
  35. }),
  36. }
  37. err := client.SwarmJoin(context.Background(), swarm.JoinRequest{
  38. ListenAddr: "0.0.0.0:2377",
  39. })
  40. if err != nil {
  41. t.Fatal(err)
  42. }
  43. }