libp2p/go-libp2p-resource-manager
1
0
Fork 0
mirror of https://github.com/libp2p/go-libp2p-resource-manager.git synced 2025-03-24 08:10:55 +08:00
Code Issues Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
vyzo 2022-01-05 17:06:44 +02:00
parent 7d5bd7a861
commit 4082746269
2 changed files with 159 additions and 117 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

122
rcmgr_test.go
View File

@ -17,79 +17,99 @@ func TestResourceManager(t *testing.T) {
mgr := NewResourceManager(
&BasicLimiter{
SystemLimits: &StaticLimit{
Memory: 16384,
StreamsInbound: 3,
StreamsOutbound: 3,
ConnsInbound: 3,
ConnsOutbound: 3,
FD: 2,
Memory: 16384,
BaseLimit: BaseLimit{
StreamsInbound: 3,
StreamsOutbound: 3,
ConnsInbound: 3,
ConnsOutbound: 3,
FD: 2,
},
},
TransientLimits: &StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
DefaultServiceLimits: &StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
ServiceLimits: map[string]Limit{
svcA: &StaticLimit{
Memory: 8192,
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 1,
Memory: 8192,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 1,
},
},
},
DefaultProtocolLimits: &StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
},
},
ProtocolLimits: map[protocol.ID]Limit{
protoA: &StaticLimit{
Memory: 8192,
StreamsInbound: 2,
StreamsOutbound: 2,
Memory: 8192,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
},
},
},
DefaultPeerLimits: &StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
PeerLimits: map[peer.ID]Limit{
peerA: &StaticLimit{
Memory: 8192,
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
PeerLimits: map[peer.ID]Limit{
peerA: &StaticLimit{
Memory: 8192,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 1,
},
},
},
ConnLimits: &StaticLimit{
Memory: 4096,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
StreamLimits: &StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
},
},
}).(*resourceManager)

154
scope_test.go
View File

@ -38,12 +38,14 @@ func checkStatus(t *testing.T, expected, status network.MemoryStatus) {
func TestResources(t *testing.T) {
rc := resources{limit: &StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
}}
checkResources(t, &rc, network.ScopeStat{})
@ -185,12 +187,14 @@ func TestResources(t *testing.T) {
func TestResourceScopeSimple(t *testing.T) {
s := newResourceScope(
&StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
nil,
)
@ -312,12 +316,14 @@ func testResourceScopeBasic(t *testing.T, s *resourceScope) {
func TestResourceScopeTxnBasic(t *testing.T) {
s := newResourceScope(
&StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
nil,
)
@ -345,12 +351,14 @@ func TestResourceScopeTxnBasic(t *testing.T) {
func TestResourceScopeTxnZombie(t *testing.T) {
s := newResourceScope(
&StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
nil,
)
@ -385,12 +393,14 @@ func TestResourceScopeTxnZombie(t *testing.T) {
func TestResourceScopeTxnTree(t *testing.T) {
s := newResourceScope(
&StaticLimit{
Memory: 4096,
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 1,
StreamsOutbound: 1,
ConnsInbound: 1,
ConnsOutbound: 1,
FD: 1,
},
},
nil,
)
@ -492,67 +502,79 @@ func TestResourceScopeDAG(t *testing.T) {
// ------> s6
s1 := newResourceScope(
&StaticLimit{
Memory: 4096,
StreamsInbound: 4,
StreamsOutbound: 4,
ConnsInbound: 4,
ConnsOutbound: 4,
FD: 4,
Memory: 4096,
BaseLimit: BaseLimit{
StreamsInbound: 4,
StreamsOutbound: 4,
ConnsInbound: 4,
ConnsOutbound: 4,
FD: 4,
},
},
nil,
)
s2 := newResourceScope(
&StaticLimit{
Memory: 2048,
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
Memory: 2048,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
},
},
[]*resourceScope{s1},
)
s3 := newResourceScope(
&StaticLimit{
Memory: 2048,
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
Memory: 2048,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
},
},
[]*resourceScope{s1},
)
s4 := newResourceScope(
&StaticLimit{
Memory: 2048,
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
Memory: 2048,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
},
},
[]*resourceScope{s2, s3, s1},
)
s5 := newResourceScope(
&StaticLimit{
Memory: 2048,
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
Memory: 2048,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
},
},
[]*resourceScope{s2, s1},
)
s6 := newResourceScope(
&StaticLimit{
Memory: 2048,
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
Memory: 2048,
BaseLimit: BaseLimit{
StreamsInbound: 2,
StreamsOutbound: 2,
ConnsInbound: 2,
ConnsOutbound: 2,
FD: 2,
},
},
[]*resourceScope{s3, s1},
)