reg/vendor/github.com/docker/docker-ce/components/engine/integration/session/session_test.go

package session // import "github.com/docker/docker/integration/session"

import (
	"net/http"
	"testing"

	req "github.com/docker/docker/integration-cli/request"
	"github.com/gotestyourself/gotestyourself/skip"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestSessionCreate(t *testing.T) {
	skip.If(t, !testEnv.DaemonInfo.ExperimentalBuild)

	defer setupTest(t)()

	res, body, err := req.Post("/session", func(r *http.Request) error {
		r.Header.Set("X-Docker-Expose-Session-Uuid", "testsessioncreate") // so we don't block default name if something else is using it
		r.Header.Set("Upgrade", "h2c")
		return nil
	})
	require.NoError(t, err)
	require.NoError(t, body.Close())
	assert.Equal(t, res.StatusCode, http.StatusSwitchingProtocols)
	assert.Equal(t, res.Header.Get("Upgrade"), "h2c")
}

func TestSessionCreateWithBadUpgrade(t *testing.T) {
	skip.If(t, !testEnv.DaemonInfo.ExperimentalBuild)

	res, body, err := req.Post("/session")
	require.NoError(t, err)
	assert.Equal(t, res.StatusCode, http.StatusBadRequest)
	buf, err := req.ReadBody(body)
	require.NoError(t, err)
	assert.Contains(t, string(buf), "no upgrade")

	res, body, err = req.Post("/session", func(r *http.Request) error {
		r.Header.Set("Upgrade", "foo")
		return nil
	})
	require.NoError(t, err)
	assert.Equal(t, res.StatusCode, http.StatusBadRequest)
	buf, err = req.ReadBody(body)
	require.NoError(t, err)
	assert.Contains(t, string(buf), "not supported")
}