Function TestProxyWithNoTargetReturnsBadGateway

middleware/proxy_test.go:461–518

(t *testing.T)

}

func TestProxyWithNoTargetReturnsBadGateway(t *testing.T) {
	targetURL, _ := url.Parse("http://127.0.0.1:8080")
	target := &ProxyTarget{Name: "target", URL: targetURL}
	emptyAfterRemove := NewRoundRobinBalancer([]*ProxyTarget{target})
	assert.True(t, emptyAfterRemove.RemoveTarget("target"))

	testCases := []struct {
		name     string
		balancer ProxyBalancer
	}{
		{
			name:     "random balancer with nil targets",
			balancer: NewRandomBalancer(nil),
		},
		{
			name:     "round-robin balancer with nil targets",
			balancer: NewRoundRobinBalancer(nil),
		},
		{
			name:     "round-robin balancer after removing last target",
			balancer: emptyAfterRemove,
		},
		{
			name:     "custom balancer with nil target",
			balancer: &customBalancer{},
		},
		{
			name:     "custom balancer with nil target URL",
			balancer: &customBalancer{target: &ProxyTarget{Name: "target"}},
		},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			e := echo.New()
			errorHandlerCalled := false
			e.Use(ProxyWithConfig(ProxyConfig{
				Balancer: tc.balancer,
				ErrorHandler: func(c *echo.Context, err error) error {
					errorHandlerCalled = true
					httpErr, ok := err.(*echo.HTTPError)
					assert.True(t, ok, "expected http error to be passed to handler")
					assert.Equal(t, http.StatusBadGateway, httpErr.Code, "expected http bad gateway error to be passed to handler")
					return err
				},
			}))

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

			assert.NotPanics(t, func() {
				e.ServeHTTP(rec, req)
			})
			assert.True(t, errorHandlerCalled)
			assert.Equal(t, http.StatusBadGateway, rec.Code)
		})
	}
}