fix: listener should emit connection (#30)

libp2p · Aug 15, 2018 · 1ac430d · 1ac430d
1 parent a35ce50
commit 1ac430d
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/src/listener.js b/src/listener.js
@@ -23,6 +23,7 @@ module.exports = (swarm, options, connHandler) => {
   const utils = utilsFactory(swarm)
 
   listener.stopHandler = new Stop(swarm)
+  listener.stopHandler.on('connection', (conn) => listener.emit('connection', conn))
   listener.hopHandler = new Hop(swarm, options.hop)
 
   /**

diff --git a/test/listener.spec.js b/test/listener.spec.js
@@ -135,6 +135,41 @@ describe('listener', function () {
       )
     })
 
+    it(`should emit 'connection'`, function (done) {
+      handlerSpy(multicodec.relay, conn)
+
+      let relayMsg = {
+        type: proto.CircuitRelay.Type.STOP,
+        srcPeer: {
+          id: `QmSswe1dCFRepmhjAMR5VfHeokGLcvVggkuDJm7RMfJSrE`,
+          addrs: [`/ipfs/QmSswe1dCFRepmhjAMR5VfHeokGLcvVggkuDJm7RMfJSrE`]
+        },
+        dstPeer: {
+          id: `QmQvM2mpqkjyXWbTHSUidUAWN26GgdMphTh9iGDdjgVXCy`,
+          addrs: [`/ipfs/QmQvM2mpqkjyXWbTHSUidUAWN26GgdMphTh9iGDdjgVXCy`]
+        }
+      }
+
+      listener.stopHandler.handle = (message, sh) => {
+        const newConn = new Connection(sh.rest())
+        listener.stopHandler.emit('connection', newConn)
+      }
+
+      listener.on('connection', (conn) => {
+        expect(conn).to.be.instanceof(Connection)
+        done()
+      })
+
+      pull(
+        pull.values([proto.CircuitRelay.encode(relayMsg)]),
+        lp.encode(),
+        pull.collect((err, encoded) => {
+          expect(err).to.not.exist()
+          encoded.forEach((e) => shake.write(e))
+        })
+      )
+    })
+
     it(`should handle CAN_HOP`, function (done) {
       handlerSpy(multicodec.relay, conn)