Function TestProxyWithConfigWebSocketNonTLS2TLS

middleware/proxy_test.go:1084–1122

TestProxyWithConfigWebSocketNonTLS2TLS tests the proxy with non-TLS to TLS WebSocket connection.

(t *testing.T)

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

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

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

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

	/*
		Act
	*/
	// Connect to the proxy WebSocket
	wsConn, err := websocket.Dial(tsURL.String(), "", "http://localhost/")
	assert.NoError(t, err)
	defer wsConn.Close()

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

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

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