moby/integration/system/ping_test.go
Sebastiaan van Stijn a4ac991d02
Merge pull request #43657 from thaJeztah/default_builder_version
api: set default "Builder-Version" to "2" (BuildKit) on Linux
2022-08-18 20:14:47 +02:00

144 lines
3.8 KiB
Go

package system // import "github.com/docker/docker/integration/system"
import (
"context"
"net/http"
"os"
"path/filepath"
"runtime"
"strings"
"testing"
"github.com/docker/docker/api/types"
"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 TestPingBuilderHeader(t *testing.T) {
skip.If(t, testEnv.IsRemoteDaemon)
skip.If(t, testEnv.DaemonInfo.OSType == "windows", "cannot spin up additional daemons on windows")
defer setupTest(t)()
d := daemon.New(t)
client := d.NewClientT(t)
defer client.Close()
ctx := context.TODO()
t.Run("default config", func(t *testing.T) {
d.Start(t)
defer d.Stop(t)
var expected = types.BuilderBuildKit
if runtime.GOOS == "windows" {
expected = types.BuilderV1
}
p, err := client.Ping(ctx)
assert.NilError(t, err)
assert.Equal(t, p.BuilderVersion, expected)
})
t.Run("buildkit disabled", func(t *testing.T) {
cfg := filepath.Join(d.RootDir(), "daemon.json")
err := os.WriteFile(cfg, []byte(`{"features": { "buildkit": false }}`), 0644)
assert.NilError(t, err)
d.Start(t, "--config-file", cfg)
defer d.Stop(t)
var expected = types.BuilderV1
p, err := client.Ping(ctx)
assert.NilError(t, err)
assert.Equal(t, p.BuilderVersion, expected)
})
}
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, ", ")
}