test(gossipsub): Part 2 Test cases covering JOIN and LEAVE Events

tests/pubsub/testgossipmembership.nim

@@ -66,7 +66,7 @@ suite "GossipSub Topic Membership Tests":
   # Helper function to subscribe to topics
   proc subscribeToTopics(gossipSub: TestGossipSub, topics: seq[string]) =
     for topic in topics:
-      gossipSub.PubSub.subscribe(
+      gossipSub.subscribe(
         topic,
         proc(topic: string, data: seq[byte]): Future[void] {.async.} =
           discard
@@ -114,7 +114,7 @@ suite "GossipSub Topic Membership Tests":
     check topic notin gossipSub.mesh
     check topic in gossipSub.gossipsub
 
-    # The topic should remain in gossipsub (for fanout)
+    # The topic should remain in gossipsub
 
     await allFuturesThrowing(conns.mapIt(it.close()))
     await gossipSub.switch.stop()
@@ -174,3 +174,74 @@ suite "GossipSub Topic Membership Tests":
 
     await allFuturesThrowing(conns.mapIt(it.close()))
     await gossipSub.switch.stop()
+
+  # Test for verifying peers joining a topic using `JOIN(topic)`
+  asyncTest "handle JOIN topic and mesh is updated":
+    let topic = "test-join-topic"
+
+    # Initialize the GossipSub system and simulate peer connections
+    let (gossipSub, conns) = setupGossipSub(topic, 5)
+
+    # Simulate peer joining the topic
+    subscribeToTopics(gossipSub, @[topic])
+
+    # Check that peers are added to the mesh and the topic is tracked
+    check gossipSub.mesh[topic].len > 0
+    check gossipSub.topics.contains(topic)
+
+    # Clean up by closing connections and stopping the gossipSub switch
+    await allFuturesThrowing(conns.mapIt(it.close()))
+    await gossipSub.switch.stop()
+
+  # Test for verifying peers leaving a topic using `LEAVE(topic)`
+  asyncTest "handle LEAVE topic and mesh is updated":
+    let topic = "test-leave-topic"
+
+    # Initialize the GossipSub system and simulate peer connections
+    let (gossipSub, conns) = setupGossipSub(topic, 5)
+
+    # Simulate peer joining the topic first
+    subscribeToTopics(gossipSub, @[topic])
+
+    # Now simulate peer leaving the topic
+    unsubscribeFromTopics(gossipSub, @[topic])
+
+    # Check that peers are removed from the mesh but the topic remains in gossipsub
+    check topic notin gossipSub.mesh
+    check topic in gossipSub.gossipsub
+
+    # Clean up by closing connections and stopping the gossipSub switch
+    await allFuturesThrowing(conns.mapIt(it.close()))
+    await gossipSub.switch.stop()
+
+  # Test the behavior when multiple peers join and leave a topic simultaneously.
+  asyncTest "multiple peers join and leave topic simultaneously":
+    let topic = "test-multi-join-leave"
+
+    # Initialize the GossipSub system and simulate peer connections for 6 peers
+    let (gossipSub, conns) = setupGossipSub(@[topic], 6)
+
+    # Ensure the topic is correctly initialized in mesh and gossipsub
+    doAssert gossipSub.mesh.contains(topic), "Topic not found in mesh"
+    doAssert gossipSub.gossipsub.contains(topic), "Topic not found in gossipsub"
+
+    # Simulate 6 peers joining the topic
+    subscribeToTopics(gossipSub, @[topic])
+
+    check gossipSub.mesh[topic].len == 6
+
+    # Simulate 3 peers leaving the topic by unsubscribing them individually
+    var peersToUnsubscribe = gossipSub.mesh[topic].toSeq()[0 .. 2]
+    for peer in peersToUnsubscribe:
+      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
+
+    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)
+
+    await allFuturesThrowing(conns.mapIt(it.close()))
+    await gossipSub.switch.stop()