mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-03 23:20:25 +08:00
28 lines
721 B
Go
28 lines
721 B
Go
// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
|
|
|
package ratelimit_test
|
|
|
|
import (
|
|
"context"
|
|
"github.com/TeaOSLab/EdgeNode/internal/utils/ratelimit"
|
|
"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
|
|
"testing"
|
|
)
|
|
|
|
func TestBandwidth(t *testing.T) {
|
|
if !testutils.IsSingleTesting() {
|
|
return
|
|
}
|
|
|
|
var bandwidth = ratelimit.NewBandwidth(32 << 10)
|
|
bandwidth.Ack(context.Background(), 123)
|
|
bandwidth.Ack(context.Background(), 16 << 10)
|
|
bandwidth.Ack(context.Background(), 32 << 10)
|
|
}
|
|
|
|
func TestBandwidth_0(t *testing.T) {
|
|
var bandwidth = ratelimit.NewBandwidth(0)
|
|
bandwidth.Ack(context.Background(), 123)
|
|
bandwidth.Ack(context.Background(), 123456)
|
|
}
|