Function TestRateLimiterWithConfig_skipperNoSkip

middleware/rate_limiter_test.go:264–297

(t *testing.T)

}

func TestRateLimiterWithConfig_skipperNoSkip(t *testing.T) {
	e := echo.New()

	var beforeFuncRan bool
	handler := func(c *echo.Context) error {
		return c.String(http.StatusOK, "test")
	}
	var inMemoryStore = NewRateLimiterMemoryStore(5)

	req := httptest.NewRequest(http.MethodGet, "/", nil)
	req.Header.Add(echo.HeaderXRealIP, "127.0.0.1")

	rec := httptest.NewRecorder()

	c := e.NewContext(req, rec)

	mw, err := RateLimiterConfig{
		Skipper: func(c *echo.Context) bool {
			return false
		},
		BeforeFunc: func(c *echo.Context) {
			beforeFuncRan = true
		},
		Store: inMemoryStore,
		IdentifierExtractor: func(ctx *echo.Context) (string, error) {
			return "127.0.0.1", nil
		},
	}.ToMiddleware()
	assert.NoError(t, err)

	_ = mw(handler)(c)

	assert.Equal(t, true, beforeFuncRan)
}

func TestRateLimiterWithConfig_beforeFunc(t *testing.T) {
	e := echo.New()