status-im
/
nim-libp2p
mirror of https://github.com/status-im/nim-libp2p.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

added alex changes

This commit is contained in:
shashankshampi 2024-10-26 11:46:17 +05:30
parent 1bfdded523
commit 49ce782eed
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/pubsub/testgossipmembership.nim
View File

@ -112,7 +112,7 @@ suite "GossipSub Topic Membership Tests":
await allFuturesThrowing(nodes.mapIt(it.switch.start())) await allFuturesThrowing(nodes.mapIt(it.switch.start()))
# When nodes subscribe and then unsubscribe to multiple topics # When nodes subscribe to multiple topics
await subscribeNodes(nodes) await subscribeNodes(nodes)
for node in nodes: for node in nodes:
for topic in topics: for topic in topics:
@ -168,11 +168,8 @@ suite "GossipSub Topic Membership Tests":
discard discard
, ,
) )
else:
# Then the subscription count should not exceed the limit
check gossipSub.topics.len == gossipSub.topicsHigh
check gossipSub.topics.len == gossipSub.topicsHigh check gossipSub.topics.len == gossipSubParams
await allFuturesThrowing(nodes.mapIt(allFutures(it.switch.stop()))) await allFuturesThrowing(nodes.mapIt(allFutures(it.switch.stop())))
@ -186,7 +183,7 @@ suite "GossipSub Topic Membership Tests":
await allFuturesThrowing(nodes.mapIt(it.switch.start())) await allFuturesThrowing(nodes.mapIt(it.switch.start()))
# When nodes join and subscribe to the topic # When nodes subscribe to the topic
await subscribeNodes(nodes) await subscribeNodes(nodes)
for node in nodes: for node in nodes:
node.subscribe(topic, voidTopicHandler) node.subscribe(topic, voidTopicHandler)
@ -197,6 +194,7 @@ suite "GossipSub Topic Membership Tests":
for node in nodes: for node in nodes:
let currentGossip = GossipSub(node) let currentGossip = GossipSub(node)
check currentGossip.mesh[topic].len == numberOfNodes - 1 check currentGossip.mesh[topic].len == numberOfNodes - 1
check currentGossip.gossipsub.hasKey(topic)
check currentGossip.topics.contains(topic) check currentGossip.topics.contains(topic)
await allFuturesThrowing(nodes.mapIt(allFutures(it.switch.stop()))) await allFuturesThrowing(nodes.mapIt(allFutures(it.switch.stop())))
@ -210,20 +208,21 @@ suite "GossipSub Topic Membership Tests":
nodes = generateNodes(numberOfNodes, gossip = true) nodes = generateNodes(numberOfNodes, gossip = true)
nodesFut = await allFinished(nodes.mapIt(it.switch.start())) nodesFut = await allFinished(nodes.mapIt(it.switch.start()))
# When they all join the topic
await subscribeNodes(nodes) await subscribeNodes(nodes)
for node in nodes: for node in nodes:
node.subscribe(topic, voidTopicHandler) node.subscribe(topic, voidTopicHandler)
await sleepAsync(2 * DURATION_TIMEOUT) await sleepAsync(2 * DURATION_TIMEOUT)
# When they all join the topic, their attributes should reflect this # Their attributes should reflect in this loop
for i in 0 ..< numberOfNodes: for i in 0 ..< numberOfNodes:
let currentGossip = GossipSub(nodes[i]) let currentGossip = GossipSub(nodes[i])
check currentGossip.gossipsub.hasKey(topic) check currentGossip.gossipsub.hasKey(topic)
check currentGossip.mesh.hasKey(topic) check currentGossip.mesh.hasKey(topic)
check currentGossip.topics.contains(topic) check currentGossip.topics.contains(topic)
# When peers subscribe to each other's topics # Make sure all nodes are connected between themselves.
for x in 0 ..< numberOfNodes: for x in 0 ..< numberOfNodes:
for y in 0 ..< numberOfNodes: for y in 0 ..< numberOfNodes:
if x != y: if x != y: