Function createSimpleProxyServer

middleware/proxy_test.go:963–996

(t *testing.T, srv *httptest.Server, serveTLS bool, toTLS bool)

}

func createSimpleProxyServer(t *testing.T, srv *httptest.Server, serveTLS bool, toTLS bool) *httptest.Server {
	e := echo.New()

	if toTLS {
		// proxy to tls target
		tgtURL, _ := url.Parse(srv.URL)
		tgtURL.Scheme = "wss"
		balancer := NewRandomBalancer([]*ProxyTarget{{URL: tgtURL}})

		defaultTransport, ok := http.DefaultTransport.(*http.Transport)
		if !ok {
			t.Fatal("Default transport is not of type *http.Transport")
		}
		transport := defaultTransport.Clone()
		transport.TLSClientConfig = &tls.Config{
			InsecureSkipVerify: true,
		}
		e.Use(ProxyWithConfig(ProxyConfig{Balancer: balancer, Transport: transport}))
	} else {
		// proxy to non-TLS target
		tgtURL, _ := url.Parse(srv.URL)
		balancer := NewRandomBalancer([]*ProxyTarget{{URL: tgtURL}})
		e.Use(ProxyWithConfig(ProxyConfig{Balancer: balancer}))
	}

	if serveTLS {
		// serve proxy server with TLS
		ts := httptest.NewTLSServer(e)
		return ts
	}
	// serve proxy server without TLS
	ts := httptest.NewServer(e)
	return ts
}

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