|
@@ -2,7 +2,11 @@ package client
|
|
|
|
|
|
import (
|
|
|
"bytes"
|
|
|
+ "encoding/json"
|
|
|
+ "fmt"
|
|
|
"io"
|
|
|
+ "os"
|
|
|
+ "strings"
|
|
|
"testing"
|
|
|
|
|
|
_ "github.com/docker/libnetwork/netutils"
|
|
@@ -15,12 +19,80 @@ type nopCloser struct {
|
|
|
|
|
|
func (nopCloser) Close() error { return nil }
|
|
|
|
|
|
+func TestMain(m *testing.M) {
|
|
|
+ setupMockHTTPCallback()
|
|
|
+ os.Exit(m.Run())
|
|
|
+}
|
|
|
+
|
|
|
+var callbackFunc func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error)
|
|
|
+var mockNwJSON, mockNwListJSON, mockServiceJSON, mockServiceListJSON []byte
|
|
|
+var mockNwName = "test"
|
|
|
+var mockNwID = "23456789"
|
|
|
+var mockServiceName = "testSrv"
|
|
|
+var mockServiceID = "23456789"
|
|
|
+var mockContainerID = "23456789"
|
|
|
+
|
|
|
+func setupMockHTTPCallback() {
|
|
|
+ var list []networkResource
|
|
|
+ nw := networkResource{Name: mockNwName, ID: mockNwID}
|
|
|
+ mockNwJSON, _ = json.Marshal(nw)
|
|
|
+ list = append(list, nw)
|
|
|
+ mockNwListJSON, _ = json.Marshal(list)
|
|
|
+
|
|
|
+ var srvList []endpointResource
|
|
|
+ ep := endpointResource{Name: mockServiceName, ID: mockServiceID, Network: mockNwName}
|
|
|
+ mockServiceJSON, _ = json.Marshal(ep)
|
|
|
+ srvList = append(srvList, ep)
|
|
|
+ mockServiceListJSON, _ = json.Marshal(srvList)
|
|
|
+
|
|
|
+ callbackFunc = func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
+ var rsp string
|
|
|
+ switch method {
|
|
|
+ case "GET":
|
|
|
+ if strings.Contains(path, fmt.Sprintf("networks?name=%s", mockNwName)) {
|
|
|
+ rsp = string(mockNwListJSON)
|
|
|
+ } else if strings.Contains(path, "networks?name=") {
|
|
|
+ rsp = "[]"
|
|
|
+ } else if strings.Contains(path, fmt.Sprintf("networks?partial-id=%s", mockNwID)) {
|
|
|
+ rsp = string(mockNwListJSON)
|
|
|
+ } else if strings.Contains(path, "networks?partial-id=") {
|
|
|
+ rsp = "[]"
|
|
|
+ } else if strings.HasSuffix(path, "networks") {
|
|
|
+ rsp = string(mockNwListJSON)
|
|
|
+ } else if strings.HasSuffix(path, "networks/"+mockNwID) {
|
|
|
+ rsp = string(mockNwJSON)
|
|
|
+ } else if strings.Contains(path, fmt.Sprintf("endpoints?name=%s", mockServiceName)) {
|
|
|
+ rsp = string(mockServiceListJSON)
|
|
|
+ } else if strings.Contains(path, "endpoints?name=") {
|
|
|
+ rsp = "[]"
|
|
|
+ } else if strings.Contains(path, fmt.Sprintf("endpoints?partial-id=%s", mockServiceID)) {
|
|
|
+ rsp = string(mockServiceListJSON)
|
|
|
+ } else if strings.Contains(path, "endpoints?partial-id=") {
|
|
|
+ rsp = "[]"
|
|
|
+ } else if strings.HasSuffix(path, "endpoints") {
|
|
|
+ rsp = string(mockServiceListJSON)
|
|
|
+ } else if strings.HasSuffix(path, "endpoints/"+mockServiceID) {
|
|
|
+ rsp = string(mockServiceJSON)
|
|
|
+ }
|
|
|
+ case "POST":
|
|
|
+ if strings.HasSuffix(path, "networks") {
|
|
|
+ rsp = mockNwID
|
|
|
+ } else if strings.HasSuffix(path, "endpoints") {
|
|
|
+ rsp = mockServiceID
|
|
|
+ } else if strings.HasSuffix(path, "containers") {
|
|
|
+ rsp = mockContainerID
|
|
|
+ }
|
|
|
+ case "PUT":
|
|
|
+ case "DELETE":
|
|
|
+ rsp = ""
|
|
|
+ }
|
|
|
+ return nopCloser{bytes.NewBufferString(rsp)}, 200, nil
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
func TestClientDummyCommand(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
- cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
- return nopCloser{bytes.NewBufferString("")}, 200, nil
|
|
|
- }
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
err := cli.Cmd("docker", "dummy")
|
|
|
if err == nil {
|
|
@@ -30,10 +102,7 @@ func TestClientDummyCommand(t *testing.T) {
|
|
|
|
|
|
func TestClientNetworkInvalidCommand(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
- cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
- return nopCloser{bytes.NewBufferString("")}, 200, nil
|
|
|
- }
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
err := cli.Cmd("docker", "network", "invalid")
|
|
|
if err == nil {
|
|
@@ -43,12 +112,9 @@ func TestClientNetworkInvalidCommand(t *testing.T) {
|
|
|
|
|
|
func TestClientNetworkCreate(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
- cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
- return nopCloser{bytes.NewBufferString("")}, 200, nil
|
|
|
- }
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
- err := cli.Cmd("docker", "network", "create", "test")
|
|
|
+ err := cli.Cmd("docker", "network", "create", mockNwName)
|
|
|
if err != nil {
|
|
|
t.Fatal(err.Error())
|
|
|
}
|
|
@@ -56,17 +122,14 @@ func TestClientNetworkCreate(t *testing.T) {
|
|
|
|
|
|
func TestClientNetworkCreateWithDriver(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
- cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
- return nopCloser{bytes.NewBufferString("")}, 200, nil
|
|
|
- }
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
- err := cli.Cmd("docker", "network", "create", "-f=dummy", "test")
|
|
|
+ err := cli.Cmd("docker", "network", "create", "-f=dummy", mockNwName)
|
|
|
if err == nil {
|
|
|
t.Fatalf("Passing incorrect flags to the create command must fail")
|
|
|
}
|
|
|
|
|
|
- err = cli.Cmd("docker", "network", "create", "-d=dummy", "test")
|
|
|
+ err = cli.Cmd("docker", "network", "create", "-d=dummy", mockNwName)
|
|
|
if err != nil {
|
|
|
t.Fatalf(err.Error())
|
|
|
}
|
|
@@ -74,12 +137,9 @@ func TestClientNetworkCreateWithDriver(t *testing.T) {
|
|
|
|
|
|
func TestClientNetworkRm(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
- cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
- return nopCloser{bytes.NewBufferString("")}, 200, nil
|
|
|
- }
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
- err := cli.Cmd("docker", "network", "rm", "test")
|
|
|
+ err := cli.Cmd("docker", "network", "rm", mockNwName)
|
|
|
if err != nil {
|
|
|
t.Fatal(err.Error())
|
|
|
}
|
|
@@ -87,47 +147,141 @@ func TestClientNetworkRm(t *testing.T) {
|
|
|
|
|
|
func TestClientNetworkLs(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
- networks := "db,web,test"
|
|
|
- cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
- return nopCloser{bytes.NewBufferString(networks)}, 200, nil
|
|
|
- }
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
err := cli.Cmd("docker", "network", "ls")
|
|
|
if err != nil {
|
|
|
t.Fatal(err.Error())
|
|
|
}
|
|
|
- if out.String() != networks {
|
|
|
- t.Fatal("Network List command fail to return the intended list")
|
|
|
+ if out.String() != string(mockNwListJSON) {
|
|
|
+ t.Fatal("Network List command fail to return the expected list")
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func TestClientNetworkInfo(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
- info := "dummy info"
|
|
|
- cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
- return nopCloser{bytes.NewBufferString(info)}, 200, nil
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "info", mockNwName)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
+ }
|
|
|
+ if out.String() != string(mockNwJSON) {
|
|
|
+ t.Fatal("Network info command fail to return the expected object")
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkInfoById(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "info", mockNwID)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
+ }
|
|
|
+ if out.String() != string(mockNwJSON) {
|
|
|
+ t.Fatal("Network info command fail to return the expected object")
|
|
|
}
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceInvalidCommand(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
- err := cli.Cmd("docker", "network", "info", "test")
|
|
|
+ err := cli.Cmd("docker", "network", "service", "invalid")
|
|
|
+ if err == nil {
|
|
|
+ t.Fatalf("Passing invalid commands must fail")
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceCreate(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "service", "create", mockServiceName, mockNwName)
|
|
|
if err != nil {
|
|
|
t.Fatal(err.Error())
|
|
|
}
|
|
|
- if out.String() != info {
|
|
|
- t.Fatal("Network List command fail to return the intended list")
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceRm(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "service", "rm", mockServiceName, mockNwName)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceLs(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "service", "ls", mockNwName)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
+ }
|
|
|
+ if out.String() != string(mockServiceListJSON) {
|
|
|
+ t.Fatal("Network service ls command fail to return the expected list")
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceInfo(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "service", "info", mockServiceName, mockNwName)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
+ }
|
|
|
+ if out.String() != string(mockServiceJSON) {
|
|
|
+ t.Fatal("Network info command fail to return the expected object")
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceInfoById(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "service", "info", mockServiceID, mockNwID)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
+ }
|
|
|
+ if out.String() != string(mockServiceJSON) {
|
|
|
+ t.Fatal("Network info command fail to return the expected object")
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceJoin(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "service", "join", mockContainerID, mockServiceName, mockNwName)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+func TestClientNetworkServiceLeave(t *testing.T) {
|
|
|
+ var out, errOut bytes.Buffer
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
+
|
|
|
+ err := cli.Cmd("docker", "network", "service", "leave", mockContainerID, mockServiceName, mockNwName)
|
|
|
+ if err != nil {
|
|
|
+ t.Fatal(err.Error())
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// Docker Flag processing in flag.go uses os.Exit() frequently, even for --help
|
|
|
// TODO : Handle the --help test-case in the IT when CLI is available
|
|
|
/*
|
|
|
-func TestClientNetworkCreateHelp(t *testing.T) {
|
|
|
+func TestClientNetworkServiceCreateHelp(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
return nil, 0, nil
|
|
|
}
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
err := cli.Cmd("docker", "network", "create", "--help")
|
|
|
if err != nil {
|
|
@@ -139,12 +293,12 @@ func TestClientNetworkCreateHelp(t *testing.T) {
|
|
|
// Docker flag processing in flag.go uses os.Exit(1) for incorrect parameter case.
|
|
|
// TODO : Handle the missing argument case in the IT when CLI is available
|
|
|
/*
|
|
|
-func TestClientNetworkCreateMissingArgument(t *testing.T) {
|
|
|
+func TestClientNetworkServiceCreateMissingArgument(t *testing.T) {
|
|
|
var out, errOut bytes.Buffer
|
|
|
cFunc := func(method, path string, data interface{}, headers map[string][]string) (io.ReadCloser, int, error) {
|
|
|
return nil, 0, nil
|
|
|
}
|
|
|
- cli := NewNetworkCli(&out, &errOut, cFunc)
|
|
|
+ cli := NewNetworkCli(&out, &errOut, callbackFunc)
|
|
|
|
|
|
err := cli.Cmd("docker", "network", "create")
|
|
|
if err != nil {
|