From 4bcb567d47d80cfbd2b5ff5b5642945913a45083 Mon Sep 17 00:00:00 2001
From: Giovanni Petrantoni <giovanni@fragcolor.xyz>
Date: Thu, 9 Jul 2020 12:34:36 +0900
Subject: [PATCH] fix gossip tests

---
 libp2p/protocols/pubsub/gossipsub.nim | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/libp2p/protocols/pubsub/gossipsub.nim b/libp2p/protocols/pubsub/gossipsub.nim
index d2efad5..9494b18 100644
--- a/libp2p/protocols/pubsub/gossipsub.nim
+++ b/libp2p/protocols/pubsub/gossipsub.nim
@@ -168,7 +168,8 @@ proc rebalanceMesh(g: GossipSub, topic: string) {.async.} =
         # send a graft message to the peer
         await p.sendPrune(@[topic])
         g.mesh[topic].excl(id)
-        g.gossipsub[topic].incl(id)
+        if topic in g.gossipsub:
+          g.gossipsub[topic].incl(id)
 
     libp2p_gossipsub_peers_per_topic_gossipsub
       .set(g.gossipsub.getOrDefault(topic).len.int64,