add bn support to varint class
This commit is contained in:
parent
2f9bc222e5
commit
4594cbb9e7
|
@ -31,6 +31,11 @@ Varint.prototype.fromBufferReader = function(br) {
|
||||||
return this;
|
return this;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
Varint.prototype.fromBN = function(bn) {
|
||||||
|
this.buf = BufferWriter().writeVarintBN(bn).concat();
|
||||||
|
return this;
|
||||||
|
};
|
||||||
|
|
||||||
Varint.prototype.fromNumber = function(num) {
|
Varint.prototype.fromNumber = function(num) {
|
||||||
this.buf = BufferWriter().writeVarintNum(num).concat();
|
this.buf = BufferWriter().writeVarintNum(num).concat();
|
||||||
return this;
|
return this;
|
||||||
|
@ -40,6 +45,10 @@ Varint.prototype.toBuffer = function() {
|
||||||
return this.buf;
|
return this.buf;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
Varint.prototype.toBN = function() {
|
||||||
|
return BufferReader(this.buf).readVarintBN();
|
||||||
|
};
|
||||||
|
|
||||||
Varint.prototype.toNumber = function() {
|
Varint.prototype.toNumber = function() {
|
||||||
return BufferReader(this.buf).readVarintNum();
|
return BufferReader(this.buf).readVarintNum();
|
||||||
};
|
};
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
var BN = require('../lib/bn');
|
||||||
var should = require('chai').should();
|
var should = require('chai').should();
|
||||||
var BufferReader = require('../lib/bufferreader');
|
var BufferReader = require('../lib/bufferreader');
|
||||||
var BufferWriter = require('../lib/bufferwriter');
|
var BufferWriter = require('../lib/bufferwriter');
|
||||||
|
@ -48,6 +49,15 @@ describe('Varint', function() {
|
||||||
|
|
||||||
});
|
});
|
||||||
|
|
||||||
|
describe('#fromBN', function() {
|
||||||
|
|
||||||
|
it('should set a number', function() {
|
||||||
|
var varint = Varint().fromNumber(BN(5));
|
||||||
|
varint.toNumber().should.equal(5);
|
||||||
|
});
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
describe('#fromNumber', function() {
|
describe('#fromNumber', function() {
|
||||||
|
|
||||||
it('should set a number', function() {
|
it('should set a number', function() {
|
||||||
|
@ -67,6 +77,15 @@ describe('Varint', function() {
|
||||||
|
|
||||||
});
|
});
|
||||||
|
|
||||||
|
describe('#toBN', function() {
|
||||||
|
|
||||||
|
it('should return a buffer', function() {
|
||||||
|
var varint = Varint(5);
|
||||||
|
varint.toBN().toString().should.equal(BN(5).toString());
|
||||||
|
});
|
||||||
|
|
||||||
|
});
|
||||||
|
|
||||||
describe('#toNumber', function() {
|
describe('#toNumber', function() {
|
||||||
|
|
||||||
it('should return a buffer', function() {
|
it('should return a buffer', function() {
|
||||||
|
|
Loading…
Reference in New Issue