Function TestRateLimiterWithConfig_defaultConfig

middleware/rate_limiter_test.go:181–226

(t *testing.T)

}

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

		e := echo.New()

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

		mw, err := RateLimiterConfig{
			Store: inMemoryStore,
		}.ToMiddleware()
		assert.NoError(t, err)

		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 TestRateLimiterWithConfig_skipper(t *testing.T) {
	e := echo.New()