83e643da0d
Signed by all authors: Signed-off-by: Michael Crosby <crosbymichael@gmail.com> Signed-off-by: Arnaud Porterie <arnaud.porterie@docker.com> Signed-off-by: David Calavera <david.calavera@gmail.com> Signed-off-by: Jeff Lindsay <progrium@gmail.com> Signed-off-by: Alexander Morozov <lk4d4@docker.com> Signed-off-by: Luke Marsden <luke@clusterhq.com> Signed-off-by: David Calavera <david.calavera@gmail.com>
105 lines
2 KiB
Go
105 lines
2 KiB
Go
package plugins
|
|
|
|
import (
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
mux *http.ServeMux
|
|
server *httptest.Server
|
|
)
|
|
|
|
func setupRemotePluginServer() string {
|
|
mux = http.NewServeMux()
|
|
server = httptest.NewServer(mux)
|
|
return server.URL
|
|
}
|
|
|
|
func teardownRemotePluginServer() {
|
|
if server != nil {
|
|
server.Close()
|
|
}
|
|
}
|
|
|
|
func TestFailedConnection(t *testing.T) {
|
|
c := NewClient("tcp://127.0.0.1:1")
|
|
err := c.callWithRetry("Service.Method", nil, nil, false)
|
|
if err == nil {
|
|
t.Fatal("Unexpected successful connection")
|
|
}
|
|
}
|
|
|
|
func TestEchoInputOutput(t *testing.T) {
|
|
addr := setupRemotePluginServer()
|
|
defer teardownRemotePluginServer()
|
|
|
|
m := Manifest{[]string{"VolumeDriver", "NetworkDriver"}}
|
|
|
|
mux.HandleFunc("/Test.Echo", func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != "POST" {
|
|
t.Fatalf("Expected POST, got %s\n", r.Method)
|
|
}
|
|
|
|
header := w.Header()
|
|
header.Set("Content-Type", versionMimetype)
|
|
|
|
io.Copy(w, r.Body)
|
|
})
|
|
|
|
c := NewClient(addr)
|
|
var output Manifest
|
|
err := c.Call("Test.Echo", m, &output)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(output, m) {
|
|
t.Fatalf("Expected %v, was %v\n", m, output)
|
|
}
|
|
}
|
|
|
|
func TestBackoff(t *testing.T) {
|
|
cases := []struct {
|
|
retries int
|
|
expTimeOff time.Duration
|
|
}{
|
|
{0, time.Duration(1)},
|
|
{1, time.Duration(2)},
|
|
{2, time.Duration(4)},
|
|
{4, time.Duration(16)},
|
|
{6, time.Duration(30)},
|
|
{10, time.Duration(30)},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
s := c.expTimeOff * time.Second
|
|
if d := backoff(c.retries); d != s {
|
|
t.Fatalf("Retry %v, expected %v, was %v\n", c.retries, s, d)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAbortRetry(t *testing.T) {
|
|
cases := []struct {
|
|
timeOff time.Duration
|
|
expAbort bool
|
|
}{
|
|
{time.Duration(1), false},
|
|
{time.Duration(2), false},
|
|
{time.Duration(10), false},
|
|
{time.Duration(30), true},
|
|
{time.Duration(40), true},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
s := c.timeOff * time.Second
|
|
if a := abort(time.Now(), s); a != c.expAbort {
|
|
t.Fatalf("Duration %v, expected %v, was %v\n", c.timeOff, s, a)
|
|
}
|
|
}
|
|
}
|