Function TestKeyAuth_errorHandlerSwallowsError

middleware/key_auth_test.go:346–375

(t *testing.T)

}

func TestKeyAuth_errorHandlerSwallowsError(t *testing.T) {
	handlerCalled := false
	var authValue string
	handler := func(c *echo.Context) error {
		handlerCalled = true
		authValue = c.Get("auth").(string)
		return c.String(http.StatusOK, "test")
	}
	middlewareChain := KeyAuthWithConfig(KeyAuthConfig{
		Validator: testKeyValidator,
		ErrorHandler: func(c *echo.Context, err error) error {
			// could check error to decide if we can swallow the error
			c.Set("auth", "public")
			return nil
		},
		ContinueOnIgnoredError: true,
	})(handler)

	e := echo.New()
	req := httptest.NewRequest(http.MethodGet, "/", nil)
	// no auth header this time
	rec := httptest.NewRecorder()
	c := e.NewContext(req, rec)

	err := middlewareChain(c)

	assert.NoError(t, err)
	assert.True(t, handlerCalled)
	assert.Equal(t, "public", authValue)
}