mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
a3b1b66bb3
I noticed I made a mistake in the first ping ("before swarm init"), which
was not specifying the daemon's socket path and because of that testing
against the main integration daemon (not the locally spun up daemon).
While fixing that, I wondered why the test didn't actually use the client
for the requests (to also verify the client converted the response), so
I rewrote the test to use `client.Ping()` and to verify the ping response
has the expected values set.
Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package system // import "github.com/docker/docker/integration/system"
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/docker/docker/api/types/swarm"
|
|
"github.com/docker/docker/api/types/versions"
|
|
"github.com/docker/docker/testutil/daemon"
|
|
"github.com/docker/docker/testutil/request"
|
|
"gotest.tools/v3/assert"
|
|
"gotest.tools/v3/skip"
|
|
)
|
|
|
|
func TestPingCacheHeaders(t *testing.T) {
|
|
skip.If(t, versions.LessThan(testEnv.DaemonAPIVersion(), "1.40"), "skip test from new feature")
|
|
defer setupTest(t)()
|
|
|
|
res, _, err := request.Get("/_ping")
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, res.StatusCode, http.StatusOK)
|
|
|
|
assert.Equal(t, hdr(res, "Cache-Control"), "no-cache, no-store, must-revalidate")
|
|
assert.Equal(t, hdr(res, "Pragma"), "no-cache")
|
|
}
|
|
|
|
func TestPingGet(t *testing.T) {
|
|
defer setupTest(t)()
|
|
|
|
res, body, err := request.Get("/_ping")
|
|
assert.NilError(t, err)
|
|
|
|
b, err := request.ReadBody(body)
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, string(b), "OK")
|
|
assert.Equal(t, res.StatusCode, http.StatusOK)
|
|
assert.Check(t, hdr(res, "API-Version") != "")
|
|
}
|
|
|
|
func TestPingHead(t *testing.T) {
|
|
skip.If(t, versions.LessThan(testEnv.DaemonAPIVersion(), "1.40"), "skip test from new feature")
|
|
defer setupTest(t)()
|
|
|
|
res, body, err := request.Head("/_ping")
|
|
assert.NilError(t, err)
|
|
|
|
b, err := request.ReadBody(body)
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, 0, len(b))
|
|
assert.Equal(t, res.StatusCode, http.StatusOK)
|
|
assert.Check(t, hdr(res, "API-Version") != "")
|
|
}
|
|
|
|
func TestPingSwarmHeader(t *testing.T) {
|
|
skip.If(t, testEnv.IsRemoteDaemon)
|
|
skip.If(t, testEnv.DaemonInfo.OSType == "windows")
|
|
|
|
defer setupTest(t)()
|
|
d := daemon.New(t)
|
|
d.Start(t)
|
|
defer d.Stop(t)
|
|
client := d.NewClientT(t)
|
|
defer client.Close()
|
|
ctx := context.TODO()
|
|
|
|
t.Run("before swarm init", func(t *testing.T) {
|
|
p, err := client.Ping(ctx)
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, p.SwarmStatus.NodeState, swarm.LocalNodeStateInactive)
|
|
assert.Equal(t, p.SwarmStatus.ControlAvailable, false)
|
|
})
|
|
|
|
_, err := client.SwarmInit(ctx, swarm.InitRequest{ListenAddr: "127.0.0.1", AdvertiseAddr: "127.0.0.1:2377"})
|
|
assert.NilError(t, err)
|
|
|
|
t.Run("after swarm init", func(t *testing.T) {
|
|
p, err := client.Ping(ctx)
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, p.SwarmStatus.NodeState, swarm.LocalNodeStateActive)
|
|
assert.Equal(t, p.SwarmStatus.ControlAvailable, true)
|
|
})
|
|
|
|
err = client.SwarmLeave(ctx, true)
|
|
assert.NilError(t, err)
|
|
|
|
t.Run("after swarm leave", func(t *testing.T) {
|
|
p, err := client.Ping(ctx)
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, p.SwarmStatus.NodeState, swarm.LocalNodeStateInactive)
|
|
assert.Equal(t, p.SwarmStatus.ControlAvailable, false)
|
|
})
|
|
}
|
|
|
|
func hdr(res *http.Response, name string) string {
|
|
val, ok := res.Header[http.CanonicalHeaderKey(name)]
|
|
if !ok || len(val) == 0 {
|
|
return ""
|
|
}
|
|
return strings.Join(val, ", ")
|
|
}
|