Method RoundTrip

testutil/websocket.go:36–63

(request *http.Request)

}

func (i InMemWebsocketRoundTripper) RoundTrip(request *http.Request) (*http.Response, error) {
	i.Logger.Debug(context.Background(), "round trip start")
	defer i.Logger.Debug(context.Background(), "round trip end")
	newCtx := i.CtxMutator(request.Context())
	request = request.WithContext(newCtx)
	serverP, clientP := net.Pipe()
	var _ io.ReadWriteCloser = clientP // compile time check that response body is OK for websocket
	response := &http.Response{
		Header: make(http.Header),
		Body:   clientP,
	}
	rw := newInMemWebsocketResponseWriter(response, serverP)
	go func() {
		i.Handler.ServeHTTP(rw, request)
		if !rw.hijacked {
			i.Logger.Debug(context.Background(), "closing connection after handler did not hijack")
			// If the handler didn't hijack the connection, we should close it when the handler finishes.
			// This prevents a 3s delay in websocket.Dial() reading the non-upgraded response.
			_ = serverP.Close()
		}
	}()
	select {
	case <-newCtx.Done():
		return nil, newCtx.Err()
	case <-rw.gotHeaders:
		return response, nil
	}
}

func newInMemWebsocketResponseWriter(resp *http.Response, conn net.Conn) *inMemWebsocketResponseWriter {
	r := bufio.NewReader(conn)