Function TestReverseProxyRoute

internal/route/reverse_proxy_test.go:170–271

(t *testing.T)

}

func TestReverseProxyRoute(t *testing.T) {
	t.Run("LinkToLoadBalancer", func(t *testing.T) {
		testTask := task.GetTestTask(t)
		entrypoint.SetCtx(testTask, newTestEntrypoint())

		cfg := Route{
			Alias:  "test",
			Scheme: route.SchemeHTTP,
			Host:   "example.com",
			Port:   Port{Proxy: 80},
			LoadBalance: &types.LoadBalancerConfig{
				Link: "test",
			},
		}
		cfg1 := Route{
			Alias:  "test1",
			Scheme: route.SchemeHTTP,
			Host:   "example.com",
			Port:   Port{Proxy: 80},
			LoadBalance: &types.LoadBalancerConfig{
				Link: "test",
			},
		}
		r, err := NewStartedTestRoute(t, &cfg)
		require.NoError(t, err)
		assert.NotNil(t, r)
		r2, err := NewStartedTestRoute(t, &cfg1)
		require.NoError(t, err)
		assert.NotNil(t, r2)
	})
	t.Run("LoadBalancerRoute", func(t *testing.T) {
		testTask := task.GetTestTask(t)
		entrypoint.SetCtx(testTask, newTestEntrypoint())

		newServer := func() *httptest.Server {
			return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
				w.WriteHeader(http.StatusOK)
			}))
		}

		srv1 := newServer()
		t.Cleanup(srv1.Close)
		srv2 := newServer()
		t.Cleanup(srv2.Close)
		srv3 := newServer()
		t.Cleanup(srv3.Close)

		makeRoute := func(alias string, target *httptest.Server) *Route {
			t.Helper()

			targetURL, err := url.Parse(target.URL)
			require.NoError(t, err)

			host, portStr, err := net.SplitHostPort(targetURL.Host)
			require.NoError(t, err)
			port, err := strconv.Atoi(portStr)
			require.NoError(t, err)