client/swarm_init_test.go

   1 package client
   2
   3 import (
   4         "bytes"
   5         "fmt"
   6         "io/ioutil"
   7         "net/http"
   8         "strings"
   9         "testing"
  10
  11         "golang.org/x/net/context"
  12
  13         "github.com/docker/docker/api/types/swarm"
  14 )
  15
  16 func TestSwarmInitError(t *testing.T) {
  17         client := &Client{
  18                 client: newMockClient(errorMock(http.StatusInternalServerError, "Server error")),
  19         }
  20
  21         _, err := client.SwarmInit(context.Background(), swarm.InitRequest{})
  22         if err == nil || err.Error() != "Error response from daemon: Server error" {
  23                 t.Fatalf("expected a Server Error, got %v", err)
  24         }
  25 }
  26
  27 func TestSwarmInit(t *testing.T) {
  28         expectedURL := "/swarm/init"
  29
  30         client := &Client{
  31                 client: newMockClient(func(req *http.Request) (*http.Response, error) {
  32                         if !strings.HasPrefix(req.URL.Path, expectedURL) {
  33                                 return nil, fmt.Errorf("Expected URL '%s', got '%s'", expectedURL, req.URL)
  34                         }
  35                         if req.Method != "POST" {
  36                                 return nil, fmt.Errorf("expected POST method, got %s", req.Method)
  37                         }
  38                         return &http.Response{
  39                                 StatusCode: http.StatusOK,
  40                                 Body:       ioutil.NopCloser(bytes.NewReader([]byte(`"body"`))),
  41                         }, nil
  42                 }),
  43         }
  44
  45         resp, err := client.SwarmInit(context.Background(), swarm.InitRequest{
  46                 ListenAddr: "0.0.0.0:2377",
  47         })
  48         if err != nil {
  49                 t.Fatal(err)
  50         }
  51         if resp != "body" {
  52                 t.Fatalf("Expected 'body', got %s", resp)
  53         }
  54 }