test: added test for allowPublishZeroPeers param in publish function

ChainSafe · Jan 9, 2023 · ce30c04 · ce30c04
1 parent c4b1bea
commit ce30c04
Showing 1 changed file with 18 additions and 1 deletion.
diff --git a/test/gossip.spec.ts b/test/gossip.spec.ts
@@ -26,7 +26,8 @@ describe('gossip', () => {
         scoreParams: {
           IPColocationFactorThreshold: GossipsubDhi + 3
         },
-        maxInboundDataLength: 4000000
+        maxInboundDataLength: 4000000,
+        allowPublishToZeroPeers: false
       }
     })
   })
@@ -80,6 +81,22 @@ describe('gossip', () => {
     nodeASpy.pushGossip.restore()
   })
 
+  it('Should allow publishing to zero peers if flag is passed', async function () {
+    this.timeout(10e4)
+    const nodeA = nodes[0]
+    const topic = 'Z'
+
+    const publishResult = await nodeA.pubsub.publish(topic, uint8ArrayFromString('hey'), {
+      allowPublishToZeroPeers: true
+    })
+
+    // gossip happens during the heartbeat
+    await pEvent(nodeA.pubsub, 'gossipsub:heartbeat')
+
+    // should have sent message to peerB
+    expect(publishResult.recipients).to.deep.equal([])
+  })
+
   it('should reject incoming messages bigger than maxInboundDataLength limit', async function () {
     this.timeout(10e4)
     const nodeA = nodes[0]