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