moby/integration-cli/docker_api_session_test.go

package main

import (
	"net/http"

	"github.com/docker/docker/integration-cli/checker"
	"github.com/docker/docker/integration-cli/request"
	"github.com/docker/docker/pkg/testutil"
	"github.com/go-check/check"
)

func (s *DockerSuite) TestSessionCreate(c *check.C) {
	testRequires(c, ExperimentalDaemon)

	res, body, err := request.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
	})
	c.Assert(err, checker.IsNil)
	c.Assert(res.StatusCode, checker.Equals, http.StatusSwitchingProtocols)
	c.Assert(res.Header.Get("Upgrade"), checker.Equals, "h2c")
	c.Assert(body.Close(), checker.IsNil)
}

func (s *DockerSuite) TestSessionCreateWithBadUpgrade(c *check.C) {
	testRequires(c, ExperimentalDaemon)

	res, body, err := request.Post("/session")
	c.Assert(err, checker.IsNil)
	c.Assert(res.StatusCode, checker.Equals, http.StatusBadRequest)
	buf, err := testutil.ReadBody(body)
	c.Assert(err, checker.IsNil)

	out := string(buf)
	c.Assert(out, checker.Contains, "no upgrade")

	res, body, err = request.Post("/session", func(r *http.Request) error {
		r.Header.Set("Upgrade", "foo")
		return nil
	})
	c.Assert(err, checker.IsNil)
	c.Assert(res.StatusCode, checker.Equals, http.StatusBadRequest)
	buf, err = testutil.ReadBody(body)
	c.Assert(err, checker.IsNil)

	out = string(buf)
	c.Assert(out, checker.Contains, "not supported")
}