Function TestRateLimiterMemoryStore_TimeOrdering

middleware/rate_limiter_test.go:615–646

TestRateLimiterMemoryStore_TimeOrdering verifies time ordering consistency in rate limiting decisions

(t *testing.T)

// TestRateLimiterMemoryStore_TimeOrdering verifies time ordering consistency in rate limiting decisions
func TestRateLimiterMemoryStore_TimeOrdering(t *testing.T) {
	t.Parallel()

	store := NewRateLimiterMemoryStoreWithConfig(RateLimiterMemoryStoreConfig{
		Rate:      1,
		Burst:     2,
		ExpiresIn: 5 * time.Second,
	})

	currentTime := time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC)
	store.timeNow = func() time.Time {
		return currentTime
	}

	// First two requests should succeed (burst=2)
	allowed1, _ := store.Allow("user1")
	assert.True(t, allowed1, "Request 1 should be allowed (burst)")

	allowed2, _ := store.Allow("user1")
	assert.True(t, allowed2, "Request 2 should be allowed (burst)")

	// Third request should be denied
	allowed3, _ := store.Allow("user1")
	assert.False(t, allowed3, "Request 3 should be denied (burst exhausted)")

	// Advance time by 1 second
	currentTime = currentTime.Add(1 * time.Second)

	// Fourth request should succeed
	allowed4, _ := store.Allow("user1")
	assert.True(t, allowed4, "Request 4 should be allowed (1 token available)")
}