From e933dac898891a88f507a39d499997c54b4c62f4 Mon Sep 17 00:00:00 2001 From: Braydon Fuller Date: Fri, 27 Mar 2015 11:40:19 -0400 Subject: [PATCH] Added default relay value for Version message. --- lib/messages/commands/version.js | 1 + test/messages/commands/index.js | 17 +++++++++++++++++ 2 files changed, 18 insertions(+) diff --git a/lib/messages/commands/version.js b/lib/messages/commands/version.js index 46083f0..6d18c62 100644 --- a/lib/messages/commands/version.js +++ b/lib/messages/commands/version.js @@ -39,6 +39,7 @@ function VersionMessage(obj) { this.timestamp = this.timestamp || new Date(); this.subversion = this.subversion || '/bitcore:' + packageInfo.version + '/'; this.startHeight = this.startHeight || 0; + this.relay = this.relay === false ? false : true; } inherits(VersionMessage, Message); diff --git a/test/messages/commands/index.js b/test/messages/commands/index.js index d2f72a3..73b812d 100644 --- a/test/messages/commands/index.js +++ b/test/messages/commands/index.js @@ -129,7 +129,24 @@ describe('Command Messages', function() { var message = messages.MerkleBlock({merkleBlock: 'not a merkle block'}); }).should.throw('An instance of MerkleBlock'); }); + }); + describe('Version', function() { + it('should set the default relay property as true', function() { + var message = messages.Version(); + should.exist(message.relay); + message.relay.should.equal(true); + }); + it('should set the relay as false', function() { + var message = messages.Version({relay: false}); + should.exist(message.relay); + message.relay.should.equal(false); + }); + it('should set the relay as true', function() { + var message = messages.Version({relay: true}); + should.exist(message.relay); + message.relay.should.equal(true); + }); }); });