tests: testing multiaddress trim leaves relevant addresses untouched

libp2p · Dec 1, 2017 · 5936229 · daviddias · Dec 1, 2017 · pgte
1 parent b7698d7
commit 5936229
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/test/multiaddr-trim.js b/test/multiaddr-trim.js
@@ -8,15 +8,25 @@ const createNode = require('./utils/node').createNode
 
 describe('multiaddr trim', () => {
   let node
+
+  after((done) => {
+    if (node) {
+      node.stop(done)
+    } else {
+      done()
+    }
+  })
+
   it('can create a test node with an irrelevant multiaddr', (done) => {
     createNode(
       [
-        '/ip4/0.0.0.0/tcp/0/p2p-webrtc-direct'
+        '/ip4/0.0.0.0/tcp/0/wss/p2p-webrtc-direct',
+        '/ip4/0.0.0.0/tcp/0'
       ],
       (err, _node) => {
         expect(err).to.not.exist()
         node = _node
-        expect(node.peerInfo.multiaddrs.toArray()).to.have.length(1)
+        expect(node.peerInfo.multiaddrs.toArray()).to.have.length(2)
         done()
       })
   })
@@ -26,7 +36,8 @@ describe('multiaddr trim', () => {
   })
 
   it('irrelevant multiaddr got trimmed', (done) => {
-    expect(node.peerInfo.multiaddrs.toArray()).to.have.length(0)
+    expect(node.peerInfo.multiaddrs.toArray()).to.have.length(1)
+    expect(node.peerInfo.multiaddrs.toArray()[0].toString()).to.match(/^\/ip4\/0.0.0.0\/tcp\/0\/ipfs\/\w+/)
     done()
   })
 })