limiter_test.go

// Copyright 2025 FishGoddess. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.

package goes

import (
	"sync"
	"testing"
	"time"
)

// go test -v -cover -run=^TestLimiter$
func TestLimiter(t *testing.T) {
	limit := 16
	limiter := NewLimiter(uint(limit))

	var countMap = make(map[int64]int, 16)
	var lock sync.Mutex

	totalCount := 10 * limit
	for i := 0; i < totalCount; i++ {
		limiter.Go(func() {
			now := time.Now().UnixMilli() / 10

			lock.Lock()
			countMap[now] = countMap[now] + 1
			lock.Unlock()

			time.Sleep(10 * time.Millisecond)
		})
	}

	limiter.Wait()

	gotTotalCount := 0
	for now, count := range countMap {
		gotTotalCount = gotTotalCount + count

		if count != limit {
			t.Fatalf("now %d: count %d != limit %d", now, count, limit)
		}
	}

	if gotTotalCount != totalCount {
		t.Fatalf("gotTotalCount %d != totalCount %d", gotTotalCount, totalCount)
	}
}

// go test -v -cover -run=^TestLimiterMinMax$
func TestLimiterMinMax(t *testing.T) {
	limiter := NewLimiter(minLimit - 1)

	got := cap(limiter.tokens)
	if got != minLimit {
		t.Fatalf("got %d != want %d", got, minLimit)
	}

	limiter = NewLimiter(maxLimit + 1)

	got = cap(limiter.tokens)
	if got != maxLimit {
		t.Fatalf("got %d != want %d", got, maxLimit)
	}
}