Function TestProxyWithConfigWebSocketTLS2NonTLS

middleware/proxy_test.go:1125–1167

TestProxyWithConfigWebSocketTLSToNoneTLS tests the proxy with TLS to non-TLS WebSocket connection. (TLS termination)

(t *testing.T)

// TestProxyWithConfigWebSocketTLSToNoneTLS tests the proxy with TLS to non-TLS WebSocket connection. (TLS termination)
func TestProxyWithConfigWebSocketTLS2NonTLS(t *testing.T) {
	/*
		Arrange
	*/

	// Create a WebSocket test server (non-TLS)
	srv := createSimpleWebSocketServer(false)
	defer srv.Close()

	// create proxy server (TLS to non-TLS)
	ts := createSimpleProxyServer(t, srv, true, false)
	defer ts.Close()

	tsURL, _ := url.Parse(ts.URL)
	tsURL.Scheme = "wss"
	tsURL.Path = "/"

	/*
		Act
	*/
	origin, err := url.Parse(ts.URL)
	assert.NoError(t, err)
	config := &websocket.Config{
		Location:  tsURL,
		Origin:    origin,
		TlsConfig: &tls.Config{InsecureSkipVerify: true}, // skip verify for testing
		Version:   websocket.ProtocolVersionHybi13,
	}
	wsConn, err := websocket.DialConfig(config)
	assert.NoError(t, err)
	defer wsConn.Close()

	// Send message
	sendMsg := "Hello, TLS to NoneTLS WebSocket!"
	err = websocket.Message.Send(wsConn, sendMsg)
	assert.NoError(t, err)

	// Read response
	var recvMsg string
	err = websocket.Message.Receive(wsConn, &recvMsg)
	assert.NoError(t, err)
	assert.Equal(t, sendMsg, recvMsg)
}

// TestProxyWebSocketXForwardedFor verifies that for WebSocket Upgrade requests,
// the proxy middleware appends c.RealIP() to any existing X-Forwarded-For chain,