status-im/nim-libp2p
2
0
Fork 0
mirror of https://github.com/status-im/nim-libp2p.git synced 2025-02-18 07:46:42 +00:00
Code Issues Projects Releases Wiki Activity

Add PRUNE message test

Browse Source
This commit is contained in:
Alejandro Cabeza Romero 2024-09-10 19:26:59 +02:00
parent 0623044fa1
commit 42b1cdb900
No known key found for this signature in database
GPG Key ID: DA3D14AE478030FD
1 changed files with 63 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
tests/pubsub/testgossipinternal.nim
View File

@ -910,15 +910,14 @@ suite "GossipSub internal":
await teardownTest(gossip0, gossip1) await teardownTest(gossip0, gossip1)
asyncTest "GRAFT messages correctly add peers to mesh": asyncTest "GRAFT messages correctly add peers to mesh":
# Potential flaky test # Potentially flaky test
# Given 2 nodes # Given 2 nodes
let let
topic = "foo" topic = "foobar"
graftMessage = ControlMessage(graft: @[ControlGraft(topicID: topic)]) graftMessage = ControlMessage(graft: @[ControlGraft(topicID: topic)])
numberOfNodes = 2 numberOfNodes = 2
# First of the hack # First part of the hack: Weird dValues so peers are not GRAFTed automatically
# Weird dValues so peers are not GRAFTed automatically
dValues = DValues(dLow: some(0), dHigh: some(0), d: some(0), dOut: some(-1)) dValues = DValues(dLow: some(0), dHigh: some(0), d: some(0), dOut: some(-1))
nodes = generateNodes( nodes = generateNodes(
numberOfNodes, gossip = true, verifySignature = false, dValues = some(dValues) numberOfNodes, gossip = true, verifySignature = false, dValues = some(dValues)
@ -959,6 +958,7 @@ suite "GossipSub internal":
g1.parameters.dLow = 1 g1.parameters.dLow = 1
g1.parameters.dHigh = 1 g1.parameters.dHigh = 1
# Potentially flaky due to this relying on sleep. Race condition against heartbeat.
# When a GRAFT message is sent # When a GRAFT message is sent
g0.broadcast(@[p1], RPCMsg(control: some(graftMessage)), isHighPriority = false) g0.broadcast(@[p1], RPCMsg(control: some(graftMessage)), isHighPriority = false)
g1.broadcast(@[p0], RPCMsg(control: some(graftMessage)), isHighPriority = false) g1.broadcast(@[p0], RPCMsg(control: some(graftMessage)), isHighPriority = false)
@ -975,3 +975,62 @@ suite "GossipSub internal":
# Cleanup # Cleanup
await allFuturesThrowing(nodes.mapIt(it.switch.stop())) await allFuturesThrowing(nodes.mapIt(it.switch.stop()))
asyncTest "PRUNE messages correctly removes peers from mesh":
# Given 2 nodes
let
topic = "foo"
backoff = 1
pruneMessage = ControlMessage(
prune: @[ControlPrune(topicID: topic, peers: @[], backoff: uint64(backoff))]
)
numberOfNodes = 2
nodes = generateNodes(numberOfNodes, gossip = true, verifySignature = false)
nodesFut = nodes.mapIt(it.switch.start())
g0 = GossipSub(nodes[0])
g1 = GossipSub(nodes[1])
tg0 = cast[TestGossipSub](g0)
tg1 = cast[TestGossipSub](g1)
p0 = tg1.getPubSubPeer(nodes[0].peerInfo.peerId)
p1 = tg0.getPubSubPeer(nodes[1].peerInfo.peerId)
discard await allFinished(nodesFut)
# And the nodes are connected
await subscribeNodes(nodes)
# And both subscribe to the topic
g0.subscribe(topic, voidTopicHandler)
g1.subscribe(topic, voidTopicHandler)
await sleepAsync(DURATION_TIMEOUT)
check:
g0.gossipsub.hasPeerId(topic, nodes[1].peerInfo.peerId) == true
g1.gossipsub.hasPeerId(topic, nodes[0].peerInfo.peerId) == true
g0.mesh.hasPeerId(topic, nodes[1].peerInfo.peerId) == true
g1.mesh.hasPeerId(topic, nodes[0].peerInfo.peerId) == true
# When a PRUNE message is sent
g0.broadcast(@[p1], RPCMsg(control: some(pruneMessage)), isHighPriority = false)
await sleepAsync(300.milliseconds)
# Then the peer is PRUNEd
check:
g0.gossipsub.hasPeerId(topic, nodes[1].peerInfo.peerId) == true
g1.gossipsub.hasPeerId(topic, nodes[0].peerInfo.peerId) == true
g0.mesh.hasPeerId(topic, nodes[1].peerInfo.peerId) == true
g1.mesh.hasPeerId(topic, nodes[0].peerInfo.peerId) == false
# When another PRUNE message is sent
g1.broadcast(@[p0], RPCMsg(control: some(pruneMessage)), isHighPriority = false)
await sleepAsync(300.milliseconds)
# Then the peer is PRUNEd
check:
g0.gossipsub.hasPeerId(topic, nodes[1].peerInfo.peerId) == true
g1.gossipsub.hasPeerId(topic, nodes[0].peerInfo.peerId) == true
g0.mesh.hasPeerId(topic, nodes[1].peerInfo.peerId) == false
g1.mesh.hasPeerId(topic, nodes[0].peerInfo.peerId) == false
# Cleanup
await allFuturesThrowing(nodes.mapIt(it.switch.stop()))