mirror of
https://github.com/cubefs/cubefs.git
synced 2026-08-02 02:00:56 +00:00
5561a1d3a6
with: #1000355286
Signed-off-by: clinx <chenlin1@oppo.com>
(cherry picked from commit b6771c282a)
33 lines
1021 B
Go
33 lines
1021 B
Go
package flashnode
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/cubefs/cubefs/proto"
|
|
"github.com/cubefs/cubefs/remotecache/flashnode/cachengine"
|
|
)
|
|
|
|
func TestCheckRateLimit(t *testing.T) {
|
|
disk := &cachengine.Disk{
|
|
Path: "/tmp/test",
|
|
Status: proto.ReadWrite,
|
|
}
|
|
|
|
keyRateLimitThreshold := int32(1024 * 1024)
|
|
keyLimiterFlow := int64(100)
|
|
allocSize := uint64(2 * 1024 * 1024)
|
|
block := cachengine.NewCacheBlockV2("/cfs_test/tmpfs", "test", "testkey", allocSize, "127.0.0.1", disk, keyRateLimitThreshold, keyLimiterFlow)
|
|
err := block.CheckRateLimit(10, uint64(keyRateLimitThreshold))
|
|
if err != nil {
|
|
t.Fatalf("Expected no error when KeyLimiter is available, got: %v", err)
|
|
}
|
|
|
|
smallAllocSize := uint64(512 * 1024)
|
|
block2 := cachengine.NewCacheBlockV2("/cfs_test/tmpfs", "test", "testkey2", smallAllocSize, "127.0.0.1", disk, keyRateLimitThreshold, keyLimiterFlow)
|
|
|
|
err = block2.CheckRateLimit(10, uint64(keyRateLimitThreshold))
|
|
if err != nil {
|
|
t.Fatalf("Expected no error when KeyLimiter is nil, got: %v", err)
|
|
}
|
|
}
|
