0ct0pu5/moby
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request from thaJeztah/plugin_refactor_setupRemotePluginServer

Browse source 
pkg/plugins: don't share plugin server between tests and t.Parallel()
This commit is contained in:
Bjorn Neergaard 2023-07-17 17:51:04 -06:00 • committed by GitHub
commit 0c6b616656
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 58 additions and 48 deletions

98
pkg/plugins/client_test.go
View file

@ -4,6 +4,7 @@ import (
"bytes"
"encoding/json"
"errors"
"fmt"
"io"
"net/http"
"net/http/httptest"
@ -18,24 +19,19 @@ import (
is "gotest.tools/v3/assert/cmp"
)
var (
mux *http.ServeMux
server *httptest.Server
)
func setupRemotePluginServer() string {
func setupRemotePluginServer(t *testing.T) (mux *http.ServeMux, addr string) {
t.Helper()
mux = http.NewServeMux()
server = httptest.NewServer(mux)
return server.URL
}
func teardownRemotePluginServer() {
if server != nil {
server := httptest.NewServer(mux)
t.Logf("started remote plugin server listening on: %s", server.URL)
t.Cleanup(func() {
server.Close()
}
})
return mux, server.URL
}
func TestFailedConnection(t *testing.T) {
t.Parallel()
c, _ := NewClient("tcp://127.0.0.1:1", &tlsconfig.Options{InsecureSkipVerify: true})
_, err := c.callWithRetry("Service.Method", nil, false)
if err == nil {
@ -44,14 +40,14 @@ func TestFailedConnection(t *testing.T) {
}
func TestFailOnce(t *testing.T) {
addr := setupRemotePluginServer()
defer teardownRemotePluginServer()
t.Parallel()
mux, addr := setupRemotePluginServer(t)
failed := false
mux.HandleFunc("/Test.FailOnce", func(w http.ResponseWriter, r *http.Request) {
if !failed {
failed = true
panic("Plugin not ready")
panic("Plugin not ready (intentional panic for test)")
}
})
@ -64,8 +60,8 @@ func TestFailOnce(t *testing.T) {
}
func TestEchoInputOutput(t *testing.T) {
addr := setupRemotePluginServer()
defer teardownRemotePluginServer()
t.Parallel()
mux, addr := setupRemotePluginServer(t)
m := Manifest{[]string{"VolumeDriver", "NetworkDriver"}}
@ -95,47 +91,56 @@ func TestEchoInputOutput(t *testing.T) {
}
func TestBackoff(t *testing.T) {
t.Parallel()
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)},
{expTimeOff: time.Duration(1)},
{retries: 1, expTimeOff: time.Duration(2)},
{retries: 2, expTimeOff: time.Duration(4)},
{retries: 4, expTimeOff: time.Duration(16)},
{retries: 6, expTimeOff: time.Duration(30)},
{retries: 10, expTimeOff: 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)
}
for _, tc := range cases {
tc := tc
t.Run(fmt.Sprintf("retries: %v", tc.retries), func(t *testing.T) {
s := tc.expTimeOff * time.Second
if d := backoff(tc.retries); d != s {
t.Fatalf("Retry %v, expected %v, was %v\n", tc.retries, s, d)
}
})
}
}
func TestAbortRetry(t *testing.T) {
t.Parallel()
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},
{timeOff: time.Duration(1)},
{timeOff: time.Duration(2)},
{timeOff: time.Duration(10)},
{timeOff: time.Duration(30), expAbort: true},
{timeOff: time.Duration(40), expAbort: 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)
}
for _, tc := range cases {
tc := tc
t.Run(fmt.Sprintf("duration: %v", tc.timeOff), func(t *testing.T) {
s := tc.timeOff * time.Second
if a := abort(time.Now(), s); a != tc.expAbort {
t.Fatalf("Duration %v, expected %v, was %v\n", tc.timeOff, s, a)
}
})
}
}
func TestClientScheme(t *testing.T) {
t.Parallel()
cases := map[string]string{
"tcp://127.0.0.1:8080": "http",
"unix:///usr/local/plugins/foo": "http",
@ -146,7 +151,7 @@ func TestClientScheme(t *testing.T) {
for addr, scheme := range cases {
u, err := url.Parse(addr)
if err != nil {
t.Fatal(err)
t.Error(err)
}
s := httpScheme(u)
@ -157,8 +162,8 @@ func TestClientScheme(t *testing.T) {
}
func TestNewClientWithTimeout(t *testing.T) {
addr := setupRemotePluginServer()
defer teardownRemotePluginServer()
t.Parallel()
mux, addr := setupRemotePluginServer(t)
m := Manifest{[]string{"VolumeDriver", "NetworkDriver"}}
@ -178,8 +183,8 @@ func TestNewClientWithTimeout(t *testing.T) {
}
func TestClientStream(t *testing.T) {
addr := setupRemotePluginServer()
defer teardownRemotePluginServer()
t.Parallel()
mux, addr := setupRemotePluginServer(t)
m := Manifest{[]string{"VolumeDriver", "NetworkDriver"}}
var output Manifest
@ -208,8 +213,8 @@ func TestClientStream(t *testing.T) {
}
func TestClientSendFile(t *testing.T) {
addr := setupRemotePluginServer()
defer teardownRemotePluginServer()
t.Parallel()
mux, addr := setupRemotePluginServer(t)
m := Manifest{[]string{"VolumeDriver", "NetworkDriver"}}
var output Manifest
@ -236,6 +241,7 @@ func TestClientSendFile(t *testing.T) {
}
func TestClientWithRequestTimeout(t *testing.T) {
t.Parallel()
type timeoutError interface {
Timeout() bool
}

8
pkg/plugins/plugin_test.go
View file

@ -25,6 +25,7 @@ const (
// regression test for deadlock in handlers
func TestPluginAddHandler(t *testing.T) {
t.Parallel()
// make a plugin which is pre-activated
p := &Plugin{activateWait: sync.NewCond(&sync.Mutex{})}
p.Manifest = &Manifest{Implements: []string{"bananas"}}
@ -36,6 +37,7 @@ func TestPluginAddHandler(t *testing.T) {
}
func TestPluginWaitBadPlugin(t *testing.T) {
t.Parallel()
p := &Plugin{activateWait: sync.NewCond(&sync.Mutex{})}
p.activateErr = errors.New("some junk happened")
testActive(t, p)
@ -57,6 +59,7 @@ func testActive(t *testing.T, p *Plugin) {
}
func TestGet(t *testing.T) {
t.Parallel()
p := &Plugin{name: fruitPlugin, activateWait: sync.NewCond(&sync.Mutex{})}
p.Manifest = &Manifest{Implements: []string{fruitImplements}}
storage.plugins[fruitPlugin] = p
@ -85,8 +88,8 @@ func TestGet(t *testing.T) {
}
func TestPluginWithNoManifest(t *testing.T) {
addr := setupRemotePluginServer()
defer teardownRemotePluginServer()
t.Parallel()
mux, addr := setupRemotePluginServer(t)
m := Manifest{[]string{fruitImplements}}
var buf bytes.Buffer
@ -123,6 +126,7 @@ func TestPluginWithNoManifest(t *testing.T) {
}
func TestGetAll(t *testing.T) {
t.Parallel()
tmpdir, unregister, r := Setup(t)
defer unregister()