Function wsFlakyEndpoint

connection/connection_test.go:179–197

(w ResponseWriter, r *http.Request)

}

func wsFlakyEndpoint(w ResponseWriter, r *http.Request) error {
	resp := &http.Response{
		StatusCode: http.StatusSwitchingProtocols,
	}
	if err := w.WriteRespHeaders(resp.StatusCode, resp.Header); err != nil {
		return err
	}
	wsCtx, cancel := context.WithCancel(r.Context())

	wsConn := websocket.NewConn(wsCtx, NewHTTPResponseReadWriterAcker(w, w.(http.Flusher), r), &log)

	rInt, _ := rand.Int(rand.Reader, big.NewInt(50))
	closedAfter := time.Millisecond * time.Duration(rInt.Int64())
	originConn := &flakyConn{closeAt: time.Now().Add(closedAfter)}
	stream.Pipe(wsConn, originConn, &log)
	cancel()
	wsConn.Close()
	return nil
}

func originRespEndpoint(w ResponseWriter, status int, data []byte) {
	resp := &http.Response{