Function TestProxyWSUpstream

proxy/ws_integration_test.go:18–94

(t *testing.T)

)

func TestProxyWSUpstream(t *testing.T) {
	wsServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		switch r.RequestURI {
		case "/ws", "/wss", "/insecure", "/strip", "/foo/bar/baz", "/new/world":
			websocket.Handler(wsEchoHandler).ServeHTTP(w, r)
		default:
			w.WriteHeader(404)
		}
	}))
	defer wsServer.Close()
	t.Log("Started WS server: ", wsServer.URL)

	wssServer := httptest.NewUnstartedServer(websocket.Handler(wsEchoHandler))
	wssServer.TLS = tlsServerConfig()
	wssServer.StartTLS()
	defer wssServer.Close()
	t.Log("Started WSS server: ", wssServer.URL)

	routes := "route add ws /ws  " + wsServer.URL + "\n"
	routes += "route add ws /wss " + wssServer.URL + ` opts "proto=https"` + "\n"
	routes += "route add ws /insecure " + wssServer.URL + ` opts "proto=https tlsskipverify=true"` + "\n"
	routes += "route add ws /foo/strip  " + wsServer.URL + ` opts "strip=/foo"` + "\n"
	routes += "route add ws /bar/baz  " + wsServer.URL + ` opts "prepend=/foo"` + "\n"
	routes += "route add ws /old/world  " + wsServer.URL + ` opts "prepend=/new strip=/old"` + "\n"

	httpProxy := httptest.NewServer(&HTTPProxy{
		Config:            config.Proxy{NoRouteStatus: 404, GZIPContentTypes: regexp.MustCompile(".*")},
		Transport:         &http.Transport{TLSClientConfig: tlsClientConfig()},
		InsecureTransport: &http.Transport{TLSClientConfig: tlsInsecureConfig()},
		Lookup: func(r *http.Request) *route.Target {
			tbl, _ := route.NewTable(bytes.NewBufferString(routes))
			return tbl.Lookup(r, route.Picker["rr"], route.Matcher["prefix"], globCache, globEnabled)
		},
	})
	defer httpProxy.Close()
	t.Log("Started HTTP proxy: ", httpProxy.URL)

	httpsProxy := httptest.NewUnstartedServer(&HTTPProxy{
		Config:            config.Proxy{NoRouteStatus: 404},
		Transport:         &http.Transport{TLSClientConfig: tlsClientConfig()},
		InsecureTransport: &http.Transport{TLSClientConfig: tlsInsecureConfig()},
		Lookup: func(r *http.Request) *route.Target {
			tbl, _ := route.NewTable(bytes.NewBufferString(routes))
			return tbl.Lookup(r, route.Picker["rr"], route.Matcher["prefix"], globCache, globEnabled)
		},
	})
	httpsProxy.TLS = tlsServerConfig()
	httpsProxy.StartTLS()
	defer httpsProxy.Close()
	t.Log("Started HTTPS proxy: ", httpsProxy.URL)

	wsServerURL := wsServer.URL[len("http://"):]
	wssServerURL := wssServer.URL[len("https://"):]
	httpProxyURL := httpProxy.URL[len("http://"):]
	httpsProxyURL := httpsProxy.URL[len("https://"):]

	t.Run("ws-ws direct", func(t *testing.T) { testWSEcho(t, "ws://"+wsServerURL+"/ws", nil) })
	t.Run("wss-wss direct", func(t *testing.T) { testWSEcho(t, "wss://"+wssServerURL+"/wss", nil) })