Function TestRateLimiter

middleware/rate_limiter_test.go:21–60

(t *testing.T)

)

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

	handler := func(c *echo.Context) error {
		return c.String(http.StatusOK, "test")
	}

	var inMemoryStore = NewRateLimiterMemoryStoreWithConfig(RateLimiterMemoryStoreConfig{Rate: 1, Burst: 3})

	mw := RateLimiterWithConfig(RateLimiterConfig{Store: inMemoryStore})

	testCases := []struct {
		id        string
		expectErr string
	}{
		{id: "127.0.0.1"},
		{id: "127.0.0.1"},
		{id: "127.0.0.1"},
		{id: "127.0.0.1", expectErr: "code=429, message=rate limit exceeded"},
		{id: "127.0.0.1", expectErr: "code=429, message=rate limit exceeded"},
		{id: "127.0.0.1", expectErr: "code=429, message=rate limit exceeded"},
		{id: "127.0.0.1", expectErr: "code=429, message=rate limit exceeded"},
	}

	for _, tc := range testCases {
		req := httptest.NewRequest(http.MethodGet, "/", nil)
		req.Header.Add(echo.HeaderXRealIP, tc.id)

		rec := httptest.NewRecorder()
		c := e.NewContext(req, rec)

		err := mw(handler)(c)
		if tc.expectErr != "" {
			assert.EqualError(t, err, tc.expectErr)
		} else {
			assert.NoError(t, err)
		}
		assert.Equal(t, http.StatusOK, rec.Code)
	}
}

func TestMustRateLimiterWithConfig_panicBehaviour(t *testing.T) {
	var inMemoryStore = NewRateLimiterMemoryStoreWithConfig(RateLimiterMemoryStoreConfig{Rate: 1, Burst: 3})