diff --git a/lib/pool.js b/lib/pool.js index 8eb6fea..b566f0e 100644 --- a/lib/pool.js +++ b/lib/pool.js @@ -193,6 +193,7 @@ Pool.prototype._connectPeer = function _connectPeer(addr) { host: ip, port: port, messages: self.messages, + network: this.network, relay: self.relay }); @@ -220,6 +221,7 @@ Pool.prototype._addConnectedPeer = function _addConnectedPeer(socket, addr) { if (!this._connectedPeers[addr.hash]) { var peer = new Peer({ socket: socket, + network: this.network, messages: self.messages }); diff --git a/test/pool.js b/test/pool.js index ac29589..8ea1663 100644 --- a/test/pool.js +++ b/test/pool.js @@ -417,6 +417,22 @@ describe('Pool', function() { Peer.prototype.connect.calledOnce.should.equal(true); connectStub.restore(); }); + + it('will pass network to peer', function() { + var connectStub = sinon.stub(Peer.prototype, 'connect'); + var pool = new Pool({network: Networks.testnet, maxSize: 1}); + var ipv6 = '2001:0db8:85a3:0042:1000:8a2e:0370:7334'; + pool._addPeerEventHandlers = sinon.stub(); + pool._connectPeer({ + ip: { + v6: ipv6 + }, + hash: 'hash' + }); + /* jshint sub: true */ + pool._connectedPeers['hash'].network.should.equal(pool.network); + connectStub.restore(); + }); }); describe('#_addConnectedPeer', function() { @@ -444,6 +460,16 @@ describe('Pool', function() { pool._addPeerEventHandlers.calledOnce.should.equal(false); }); + it('will pass network to peer', function() { + /* jshint sub: true */ + var pool = new Pool({network: Networks.testnet, maxSize: 1}); + pool._addConnectedPeer({ + on: sinon.stub() + }, {hash: 'hash'}); + should.exist(pool._connectedPeers['hash']); + pool._connectedPeers['hash'].network.should.equal(pool.network); + }); + }); describe('#listen', function() {