Function TestForwardToProxy

proxy/proxy_test.go:2271–2318

(t *testing.T)

}

func TestForwardToProxy(t *testing.T) {
	for _, ti := range []struct {
		tls                *tls.ConnectionState
		outgoingURL        string
		incomingHost       string
		expectedProxyURL   string
		expectedRequestURL string
	}{{
		tls:                nil,
		outgoingURL:        "http://proxy.example.com/anything?key=val",
		incomingHost:       "example.com",
		expectedProxyURL:   "http://proxy.example.com",
		expectedRequestURL: "http://example.com/anything?key=val",
	}, {
		tls:                nil,
		outgoingURL:        "https://proxy.example.com/anything?key=val",
		incomingHost:       "example.com",
		expectedProxyURL:   "https://proxy.example.com",
		expectedRequestURL: "http://example.com/anything?key=val",
	}, {
		tls:                &tls.ConnectionState{},
		outgoingURL:        "http://proxy.example.com/anything?key=val",
		incomingHost:       "example.com",
		expectedProxyURL:   "http://proxy.example.com",
		expectedRequestURL: "https://example.com/anything?key=val",
	}} {
		outgoingURL, _ := url.Parse(ti.outgoingURL)

		outgoing := &http.Request{
			URL:    outgoingURL,
			Header: make(http.Header),
		}

		incoming := &http.Request{
			Host: ti.incomingHost,
			TLS:  ti.tls,
		}

		outgoing = forwardToProxy(incoming, outgoing)

		assert.Equal(t, ti.expectedRequestURL, outgoing.URL.String())

		proxyURL, err := proxyFromContext(outgoing)

		assert.NoError(t, err)
		assert.Equal(t, ti.expectedProxyURL, proxyURL.String())
	}
}

func TestProxyFromEmptyContext(t *testing.T) {
	proxyURL, err := proxyFromContext(&http.Request{})