Function TestProxyWithNoTargetDoesNotRetry

middleware/proxy_test.go:520–553

(t *testing.T)

}

func TestProxyWithNoTargetDoesNotRetry(t *testing.T) {
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.WriteHeader(http.StatusOK)
	}))
	defer server.Close()
	targetURL, _ := url.Parse(server.URL)

	balancer := &sequenceBalancer{
		targets: []*ProxyTarget{
			nil,
			{Name: "target", URL: targetURL},
		},
	}

	retryFilterCalled := false
	e := echo.New()
	e.Use(ProxyWithConfig(ProxyConfig{
		Balancer:   balancer,
		RetryCount: 1,
		RetryFilter: func(c *echo.Context, err error) bool {
			retryFilterCalled = true
			return true
		},
	}))

	req := httptest.NewRequest(http.MethodGet, "/", nil)
	rec := httptest.NewRecorder()

	e.ServeHTTP(rec, req)

	assert.False(t, retryFilterCalled)
	assert.Equal(t, 1, balancer.calls)
	assert.Equal(t, http.StatusBadGateway, rec.Code)
}

func TestProxyRetries(t *testing.T) {
	newServer := func(res int) (*url.URL, *httptest.Server) {