mirror of
https://github.com/libp2p/go-libp2p-resource-manager.git
synced 2025-01-27 12:50:07 +08:00
89 lines
2.5 KiB
Go
89 lines
2.5 KiB
Go
package rcmgr
|
|
|
|
import (
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFileDescriptorCounting(t *testing.T) {
|
|
if runtime.GOOS == "windows" {
|
|
t.Skip("can't read file descriptors on Windows")
|
|
}
|
|
n := getNumFDs()
|
|
require.NotZero(t, n)
|
|
require.Less(t, n, int(1e6))
|
|
}
|
|
|
|
func TestScaling(t *testing.T) {
|
|
base := BaseLimit{
|
|
Streams: 100,
|
|
StreamsInbound: 200,
|
|
StreamsOutbound: 400,
|
|
Conns: 10,
|
|
ConnsInbound: 20,
|
|
ConnsOutbound: 40,
|
|
FD: 1,
|
|
Memory: 1 << 20,
|
|
}
|
|
|
|
t.Run("no scaling if no increase is defined", func(t *testing.T) {
|
|
cfg := ScalingLimitConfig{ServiceBaseLimit: base}
|
|
scaled := cfg.Scale(8<<30, 100)
|
|
require.Equal(t, base, scaled.ServiceDefault)
|
|
})
|
|
|
|
t.Run("scaling", func(t *testing.T) {
|
|
cfg := ScalingLimitConfig{
|
|
TransientBaseLimit: base,
|
|
TransientLimitIncrease: BaseLimitIncrease{
|
|
Streams: 1,
|
|
StreamsInbound: 2,
|
|
StreamsOutbound: 3,
|
|
Conns: 4,
|
|
ConnsInbound: 5,
|
|
ConnsOutbound: 6,
|
|
Memory: 7,
|
|
FDFraction: 0.5,
|
|
},
|
|
}
|
|
scaled := cfg.Scale(128<<20+4<<30, 1000)
|
|
require.Equal(t, 500, scaled.Transient.FD)
|
|
require.Equal(t, base.Streams+4, scaled.Transient.Streams)
|
|
require.Equal(t, base.StreamsInbound+4*2, scaled.Transient.StreamsInbound)
|
|
require.Equal(t, base.StreamsOutbound+4*3, scaled.Transient.StreamsOutbound)
|
|
require.Equal(t, base.Conns+4*4, scaled.Transient.Conns)
|
|
require.Equal(t, base.ConnsInbound+4*5, scaled.Transient.ConnsInbound)
|
|
require.Equal(t, base.ConnsOutbound+4*6, scaled.Transient.ConnsOutbound)
|
|
require.Equal(t, base.Memory+4*7, scaled.Transient.Memory)
|
|
})
|
|
|
|
t.Run("scaling limits in maps", func(t *testing.T) {
|
|
cfg := ScalingLimitConfig{
|
|
ServiceLimits: map[string]baseLimitConfig{
|
|
"A": {
|
|
BaseLimit: BaseLimit{Streams: 10, Memory: 100, FD: 9},
|
|
},
|
|
"B": {
|
|
BaseLimit: BaseLimit{Streams: 20, Memory: 200, FD: 10},
|
|
BaseLimitIncrease: BaseLimitIncrease{Streams: 2, Memory: 3, FDFraction: 0.4},
|
|
},
|
|
},
|
|
}
|
|
scaled := cfg.Scale(128<<20+4<<30, 1000)
|
|
|
|
require.Len(t, scaled.Service, 2)
|
|
require.Contains(t, scaled.Service, "A")
|
|
require.Equal(t, 10, scaled.Service["A"].Streams)
|
|
require.Equal(t, int64(100), scaled.Service["A"].Memory)
|
|
require.Equal(t, 9, scaled.Service["A"].FD)
|
|
|
|
require.Contains(t, scaled.Service, "B")
|
|
require.Equal(t, 20+4*2, scaled.Service["B"].Streams)
|
|
require.Equal(t, int64(200+4*3), scaled.Service["B"].Memory)
|
|
require.Equal(t, 400, scaled.Service["B"].FD)
|
|
|
|
})
|
|
}
|