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