mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-05 01:20:26 +08:00
55 lines
1.2 KiB
Go
55 lines
1.2 KiB
Go
|
// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
|
||
|
|
|
||
|
|
package fsutils_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
fsutils "github.com/TeaOSLab/EdgeNode/internal/utils/fs"
|
||
|
|
"github.com/iwind/TeaGo/assert"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestLimiter_Ack(t *testing.T) {
|
||
|
|
var a = assert.NewAssertion(t)
|
||
|
|
|
||
|
|
{
|
||
|
|
var limiter = fsutils.NewLimiter(4)
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 4)
|
||
|
|
limiter.Ack()
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 3)
|
||
|
|
limiter.Ack()
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 2)
|
||
|
|
limiter.Release()
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 3)
|
||
|
|
limiter.Release()
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 4)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLimiter_TryAck(t *testing.T) {
|
||
|
|
var a = assert.NewAssertion(t)
|
||
|
|
|
||
|
|
{
|
||
|
|
var limiter = fsutils.NewLimiter(4)
|
||
|
|
var count = limiter.FreeThreads()
|
||
|
|
a.IsTrue(count == 4)
|
||
|
|
for i := 0; i < count; i++ {
|
||
|
|
limiter.Ack()
|
||
|
|
}
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 0)
|
||
|
|
a.IsFalse(limiter.TryAck())
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 0)
|
||
|
|
}
|
||
|
|
|
||
|
|
{
|
||
|
|
var limiter = fsutils.NewLimiter(4)
|
||
|
|
var count = limiter.FreeThreads()
|
||
|
|
a.IsTrue(count == 4)
|
||
|
|
for i := 0; i < count-1; i++ {
|
||
|
|
limiter.Ack()
|
||
|
|
}
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 1)
|
||
|
|
a.IsTrue(limiter.TryAck())
|
||
|
|
a.IsTrue(limiter.FreeThreads() == 0)
|
||
|
|
}
|
||
|
|
}
|