From 572582938f0488333f914bb4a5798cff5c5de6c5 Mon Sep 17 00:00:00 2001 From: "Ryan X. Charles" Date: Mon, 1 Sep 2014 15:53:26 -0700 Subject: [PATCH] more convenient bufferreader --- lib/bufferreader.js | 11 ++++++++--- test/bufferreader.js | 9 +++++++++ 2 files changed, 17 insertions(+), 3 deletions(-) diff --git a/lib/bufferreader.js b/lib/bufferreader.js index cf6b891..b8e2e72 100644 --- a/lib/bufferreader.js +++ b/lib/bufferreader.js @@ -1,10 +1,15 @@ var BN = require('./bn'); -var BufferReader = function BufferReader(obj) { +var BufferReader = function BufferReader(buf) { if (!(this instanceof BufferReader)) - return new BufferReader(obj); - if (obj) + return new BufferReader(buf); + if (Buffer.isBuffer(buf)) { + this.set({buf: buf}); + } + else if (buf) { + var obj = buf; this.set(obj); + } }; BufferReader.prototype.set = function(obj) { diff --git a/test/bufferreader.js b/test/bufferreader.js index d4ffdb3..735d960 100644 --- a/test/bufferreader.js +++ b/test/bufferreader.js @@ -7,6 +7,15 @@ describe('BufferReader', function() { it('should make a new BufferReader', function() { var br = new BufferReader(); should.exist(br); + br = BufferReader(); + should.exist(br); + }); + + it('should create a new bufferreader with a buffer', function() { + var buf = new Buffer(0); + var br = new BufferReader(buf); + should.exist(br); + Buffer.isBuffer(br.buf).should.equal(true); }); describe('#set', function() {