mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
49 lines
1.4 KiB
Go
49 lines
1.4 KiB
Go
![]() |
package main
|
||
|
|
||
|
import (
|
||
|
"net/http"
|
||
|
|
||
|
"github.com/docker/docker/integration-cli/checker"
|
||
|
"github.com/docker/docker/integration-cli/request"
|
||
|
"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 := request.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 = request.ReadBody(body)
|
||
|
c.Assert(err, checker.IsNil)
|
||
|
|
||
|
out = string(buf)
|
||
|
c.Assert(out, checker.Contains, "not supported")
|
||
|
}
|