slider_test.go

package equalizer

import (
	"context"
	"testing"
	"time"

	"github.com/reugn/equalizer/internal/assert"
)

func TestSlider(t *testing.T) {
	slider, err := NewSlider(time.Second, 100*time.Millisecond, 32)
	assert.IsNil(t, err)
	var quota bool
	for i := 0; i < 32; i++ {
		quota = slider.TryAcquire()
	}
	assert.Equal(t, quota, true)

	quota = slider.TryAcquire()
	assert.Equal(t, quota, false)

	time.Sleep(1010 * time.Millisecond)

	quota = slider.TryAcquire()
	assert.Equal(t, quota, true)

	assert.IsNil(t, slider.Acquire(context.Background()))
}

func TestSlider_Context(t *testing.T) {
	slider, err := NewSlider(time.Second, 200*time.Millisecond, 1)
	assert.IsNil(t, err)
	assert.IsNil(t, slider.Acquire(context.Background()))

	ctx, cancel := context.WithDeadline(context.Background(), time.Now().Add(10*time.Millisecond))
	defer cancel()
	err = slider.Acquire(ctx)
	assert.ErrorIs(t, err, context.DeadlineExceeded)
}

func TestSlider_Errors(t *testing.T) {
	slider, err := NewSlider(time.Second, 200*time.Millisecond, -1)
	assert.ErrorContains(t, err, "nonpositive capacity")
	assert.IsNil(t, slider)

	slider, err = NewSlider(100*time.Millisecond, time.Second, 32)
	assert.ErrorContains(t, err, "slidingInterval must not exceed window")
	assert.IsNil(t, slider)
}