github.com/zhouyu0/docker-note@v0.0.0-20190722021225-b8d3825084db/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/internal/test/request"
     8  	"gotest.tools/assert"
     9  	is "gotest.tools/assert/cmp"
    10  	"gotest.tools/skip"
    11  )
    12  
    13  func TestSessionCreate(t *testing.T) {
    14  	skip.If(t, !testEnv.DaemonInfo.ExperimentalBuild)
    15  	skip.If(t, testEnv.OSType == "windows", "FIXME")
    16  
    17  	defer setupTest(t)()
    18  
    19  	res, body, err := req.Post("/session", req.With(func(r *http.Request) error {
    20  		r.Header.Set("X-Docker-Expose-Session-Uuid", "testsessioncreate") // so we don't block default name if something else is using it
    21  		r.Header.Set("Upgrade", "h2c")
    22  		return nil
    23  	}))
    24  	assert.NilError(t, err)
    25  	assert.NilError(t, body.Close())
    26  	assert.Check(t, is.DeepEqual(res.StatusCode, http.StatusSwitchingProtocols))
    27  	assert.Check(t, is.Equal(res.Header.Get("Upgrade"), "h2c"))
    28  }
    29  
    30  func TestSessionCreateWithBadUpgrade(t *testing.T) {
    31  	skip.If(t, !testEnv.DaemonInfo.ExperimentalBuild)
    32  	skip.If(t, testEnv.OSType == "windows", "FIXME")
    33  
    34  	res, body, err := req.Post("/session")
    35  	assert.NilError(t, err)
    36  	assert.Check(t, is.DeepEqual(res.StatusCode, http.StatusBadRequest))
    37  	buf, err := req.ReadBody(body)
    38  	assert.NilError(t, err)
    39  	assert.Check(t, is.Contains(string(buf), "no upgrade"))
    40  
    41  	res, body, err = req.Post("/session", req.With(func(r *http.Request) error {
    42  		r.Header.Set("Upgrade", "foo")
    43  		return nil
    44  	}))
    45  	assert.NilError(t, err)
    46  	assert.Check(t, is.DeepEqual(res.StatusCode, http.StatusBadRequest))
    47  	buf, err = req.ReadBody(body)
    48  	assert.NilError(t, err)
    49  	assert.Check(t, is.Contains(string(buf), "not supported"))
    50  }