EdgeNode/internal/utils/ratelimit/counter_test.go

// Copyright 2021 GoEdge goedge.cdn@gmail.com. All rights reserved.

package ratelimit_test

import (
	"testing"
	"time"

	"github.com/TeaOSLab/EdgeNode/internal/utils/ratelimit"
	"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
)

func TestCounter_ACK(t *testing.T) {
	if !testutils.IsSingleTesting() {
		return
	}

	var counter = ratelimit.NewCounter(10)

	go func() {
		for i := 0; i < 10; i++ {
			counter.Ack()
		}
		//counter.Release()
		t.Log("waiting", time.Now().Unix())
		counter.Ack()
		t.Log("done", time.Now().Unix())
	}()

	time.Sleep(1 * time.Second)
	counter.Close()
	time.Sleep(1 * time.Second)
}

func TestCounter_Release(t *testing.T) {
	var counter = ratelimit.NewCounter(10)

	for i := 0; i < 10; i++ {
		counter.Ack()
	}
	for i := 0; i < 10; i++ {
		counter.Release()
	}
	t.Log(counter.Len())
}