vacp2p/nim-libp2p
1
0
Fork 0
mirror of https://github.com/vacp2p/nim-libp2p.git synced 2025-03-03 09:31:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix in test logic for multiple peers join and leave topic simultaneously

Browse Source
This commit is contained in:
shashankshampi 2024-10-08 16:17:42 +05:30
parent 806592ddbb
commit 2d38e8abfc
1 changed files with 8 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
tests/pubsub/testgossipmembership.nim
View File

@ -228,21 +228,20 @@ suite "GossipSub Topic Membership Tests":
# Simulate 6 peers joining the topic
subscribeToTopics(gossipSub, @[topic])
# Check that 6 peers have joined the mesh
check gossipSub.mesh[topic].len == 6 # Mesh should have 6 peers
check gossipSub.mesh[topic].len == 6
# Simulate 3 peers leaving the topic by unsubscribing them
# Simulate 3 peers leaving the topic by unsubscribing them individually
var peersToUnsubscribe = gossipSub.mesh[topic].toSeq()[0 .. 2]
# Select the first 3 peers to unsubscribe
for peer in peersToUnsubscribe:
gossipSub.PubSub.unsubscribeAll(topic) # Unsubscribing from the topic
echo "Unsubscribing peer: ", peer.peerId
gossipSub.mesh[topic].excl(peer)
# Now validate the state of the mesh and gossipsub. Ensure 3 peers are still subscribed
check gossipSub.mesh[topic].len == 3
# Validate that 3 peers are still subscribed and 3 peers have been unsubscribed
check gossipSub.mesh[topic].len == 3 # Ensure 3 peers are still subscribed
for peer in peersToUnsubscribe:
# Ensure the first 3 peers are unsubscribed by checking if they are not in the mesh
check not gossipSub.mesh[topic].contains(peer)
# Ensure the first 3 peers are unsubscribed
# Clean up by closing connections and stopping the gossipSub switch
await allFuturesThrowing(conns.mapIt(it.close()))
await gossipSub.switch.stop()