github.com/rita33cool1/iot-system-gateway@v0.0.0-20200911033302-e65bde238cc5/docker-engine/integration/session/session_test.go (about) 1 package session // import "github.com/docker/docker/integration/session" 2 3 import ( 4 "net/http" 5 "testing" 6 7 req "github.com/docker/docker/integration-cli/request" 8 "github.com/gotestyourself/gotestyourself/assert" 9 is "github.com/gotestyourself/gotestyourself/assert/cmp" 10 "github.com/gotestyourself/gotestyourself/skip" 11 ) 12 13 func TestSessionCreate(t *testing.T) { 14 skip.If(t, !testEnv.DaemonInfo.ExperimentalBuild) 15 16 defer setupTest(t)() 17 18 res, body, err := req.Post("/session", func(r *http.Request) error { 19 r.Header.Set("X-Docker-Expose-Session-Uuid", "testsessioncreate") // so we don't block default name if something else is using it 20 r.Header.Set("Upgrade", "h2c") 21 return nil 22 }) 23 assert.NilError(t, err) 24 assert.NilError(t, body.Close()) 25 assert.Check(t, is.DeepEqual(res.StatusCode, http.StatusSwitchingProtocols)) 26 assert.Check(t, is.Equal(res.Header.Get("Upgrade"), "h2c")) 27 } 28 29 func TestSessionCreateWithBadUpgrade(t *testing.T) { 30 skip.If(t, !testEnv.DaemonInfo.ExperimentalBuild) 31 32 res, body, err := req.Post("/session") 33 assert.NilError(t, err) 34 assert.Check(t, is.DeepEqual(res.StatusCode, http.StatusBadRequest)) 35 buf, err := req.ReadBody(body) 36 assert.NilError(t, err) 37 assert.Check(t, is.Contains(string(buf), "no upgrade")) 38 39 res, body, err = req.Post("/session", func(r *http.Request) error { 40 r.Header.Set("Upgrade", "foo") 41 return nil 42 }) 43 assert.NilError(t, err) 44 assert.Check(t, is.DeepEqual(res.StatusCode, http.StatusBadRequest)) 45 buf, err = req.ReadBody(body) 46 assert.NilError(t, err) 47 assert.Check(t, is.Contains(string(buf), "not supported")) 48 }