test: fix test

src/connection.js

@@ -5,6 +5,7 @@ const multistream = require('multistream-select')
 const waterfall = require('async/waterfall')
 const debug = require('debug')
 const log = debug('libp2p:swarm:connection')
+const once = require('once')
 const setImmediate = require('async/setImmediate')
 
 const Circuit = require('libp2p-circuit')
@@ -37,6 +38,7 @@ module.exports = function connection (swarm) {
           conn.getPeerInfo = (cb) => {
             const conn = muxedConn.newStream()
             const ms = new multistream.Dialer()
+            cb = once(cb)
 
             waterfall([
               (cb) => ms.handle(conn, cb),

test/swarm-muxing.node.js

@@ -179,7 +179,7 @@ describe('Switch (everything all together)', () => {
 
     function check (err) {
       expect(err).to.not.exist()
-      if (i++ === 2) { done() }
+      if (++i === 3) { done() }
     }
 
     switchC.handle('/mamao/1.0.0', (protocol, conn) => {
@@ -205,7 +205,7 @@ describe('Switch (everything all together)', () => {
       expect(switchC._peerInfo.isConnected).to.exist()
       expect(switchA._peerInfo.isConnected).to.exist()
 
-      tryEcho(conn, done)
+      tryEcho(conn, check)
     })
   })
 
@@ -233,9 +233,10 @@ describe('Switch (everything all together)', () => {
 
   it('close a muxer emits event', function (done) {
     this.timeout(3 * 1000)
+
     parallel([
-      (cb) => switchC.stop(cb),
-      (cb) => switchA.once('peer-mux-closed', (peerInfo) => cb())
+      (cb) => switchA.once('peer-mux-closed', (peerInfo) => cb()),
+      (cb) => switchC.stop(cb)
     ], done)
   })
 })