mirror of
https://github.com/fluencelabs/js-libp2p
synced 2025-03-30 14:21:04 +00:00
96 lines
3.3 KiB
JavaScript
96 lines
3.3 KiB
JavaScript
'use strict'
|
|
/* eslint-env mocha */
|
|
|
|
const { expect } = require('aegir/utils/chai')
|
|
|
|
const AddressManager = require('../../src/address-manager')
|
|
const TransportManager = require('../../src/transport-manager')
|
|
const PeerStore = require('../../src/peer-store')
|
|
const PeerRecord = require('../../src/record/peer-record')
|
|
const Transport = require('libp2p-tcp')
|
|
const PeerId = require('peer-id')
|
|
const multiaddr = require('multiaddr')
|
|
const mockUpgrader = require('../utils/mockUpgrader')
|
|
const sinon = require('sinon')
|
|
const Peers = require('../fixtures/peers')
|
|
const addrs = [
|
|
multiaddr('/ip4/127.0.0.1/tcp/0'),
|
|
multiaddr('/ip4/127.0.0.1/tcp/0')
|
|
]
|
|
|
|
describe('Transport Manager (TCP)', () => {
|
|
let tm
|
|
let localPeer
|
|
|
|
before(async () => {
|
|
localPeer = await PeerId.createFromJSON(Peers[0])
|
|
})
|
|
|
|
beforeEach(() => {
|
|
tm = new TransportManager({
|
|
libp2p: {
|
|
peerId: localPeer,
|
|
multiaddrs: addrs,
|
|
addressManager: new AddressManager({ listen: addrs }),
|
|
peerStore: new PeerStore({ peerId: localPeer })
|
|
},
|
|
upgrader: mockUpgrader,
|
|
onConnection: () => {}
|
|
})
|
|
})
|
|
|
|
afterEach(async () => {
|
|
await tm.removeAll()
|
|
expect(tm._transports.size).to.equal(0)
|
|
})
|
|
|
|
it('should be able to add and remove a transport', async () => {
|
|
tm.add(Transport.prototype[Symbol.toStringTag], Transport)
|
|
expect(tm._transports.size).to.equal(1)
|
|
await tm.remove(Transport.prototype[Symbol.toStringTag])
|
|
})
|
|
|
|
it('should be able to listen', async () => {
|
|
tm.add(Transport.prototype[Symbol.toStringTag], Transport, { listenerOptions: { listen: 'carefully' } })
|
|
const transport = tm._transports.get(Transport.prototype[Symbol.toStringTag])
|
|
const spyListener = sinon.spy(transport, 'createListener')
|
|
await tm.listen(addrs)
|
|
expect(tm._listeners).to.have.key(Transport.prototype[Symbol.toStringTag])
|
|
expect(tm._listeners.get(Transport.prototype[Symbol.toStringTag])).to.have.length(addrs.length)
|
|
|
|
// Ephemeral ip addresses may result in multiple listeners
|
|
expect(tm.getAddrs().length).to.equal(addrs.length)
|
|
await tm.close()
|
|
expect(tm._listeners.get(Transport.prototype[Symbol.toStringTag])).to.have.length(0)
|
|
expect(spyListener.firstCall.firstArg).to.deep.equal({ listen: 'carefully' })
|
|
})
|
|
|
|
it('should create self signed peer record on listen', async () => {
|
|
let signedPeerRecord = await tm.libp2p.peerStore.addressBook.getPeerRecord(localPeer)
|
|
expect(signedPeerRecord).to.not.exist()
|
|
|
|
tm.add(Transport.prototype[Symbol.toStringTag], Transport)
|
|
await tm.listen(addrs)
|
|
|
|
// Should created Self Peer record on new listen address
|
|
signedPeerRecord = await tm.libp2p.peerStore.addressBook.getPeerRecord(localPeer)
|
|
expect(signedPeerRecord).to.exist()
|
|
|
|
const record = PeerRecord.createFromProtobuf(signedPeerRecord.payload)
|
|
expect(record).to.exist()
|
|
expect(record.multiaddrs.length).to.equal(addrs.length)
|
|
addrs.forEach((a, i) => {
|
|
expect(record.multiaddrs[i].equals(a)).to.be.true()
|
|
})
|
|
})
|
|
|
|
it('should be able to dial', async () => {
|
|
tm.add(Transport.prototype[Symbol.toStringTag], Transport)
|
|
await tm.listen(addrs)
|
|
const addr = tm.getAddrs().shift()
|
|
const connection = await tm.dial(addr)
|
|
expect(connection).to.exist()
|
|
await connection.close()
|
|
})
|
|
})
|