327 lines
9.9 KiB
JavaScript
327 lines
9.9 KiB
JavaScript
'use strict';
|
|
|
|
var chai = require('chai');
|
|
var should = chai.should();
|
|
var assert = chai.assert;
|
|
var expect = chai.expect;
|
|
var DBService = require('../../../lib/services/db');
|
|
var sinon = require('sinon');
|
|
var Levelup = require('levelup');
|
|
var Tx = require('bcoin').tx;
|
|
|
|
describe('DB', function() {
|
|
|
|
var dbService;
|
|
var tx = Tx.fromRaw( '0100000004de9b4bb17f627096a9ee0b4528e4eae17df5b5c69edc29704c2e84a7371db29f010000006b483045022100f5b1a0d33b7be291c3953c25f8ae39d98601aa7099a8674daf638a08b86c7173022006ce372da5ad088a1cc6e5c49c2760a1b6f085eb1b51b502211b6bc9508661f9012102ec5e3731e54475dd2902326f43602a03ae3d62753324139163f81f20e787514cffffffff7a1d4e5fc2b8177ec738cd723a16cf2bf493791e55573445fc0df630fe5e2d64010000006b483045022100cf97f6cb8f126703e9768545dfb20ffb10ba78ae3d101aa46775f5a239b075fc02203150c4a89a11eaf5e404f4f96b62efa4455e9525765a025525c7105a7e47b6db012102c01e11b1d331f999bbdb83e8831de503cd52a01e3834a95ccafd615c67703d77ffffffff9e52447116415ca0d0567418a1a4ef8f27be3ff5a96bf87c922f3723d7db5d7c000000006b483045022100f6c117e536701be41a6b0b544d7c3b1091301e4e64a6265b6eb167b15d16959d022076916de4b115e700964194ce36a24cb9105f86482f4abbc63110c3f537cd5770012102ddf84cc7bee2d6a82ac09628a8ad4a26cd449fc528b81e7e6cc615707b8169dfffffffff5815d9750eb3572e30d6fd9df7afb4dbd76e042f3aa4988ac763b3fdf8397f80010000006a473044022028f4402b736066d93d2a32b28ccd3b7a21d84bb58fcd07fe392a611db94cdec5022018902ee0bf2c3c840c1b81ead4e6c87c88c48b2005bf5eea796464e561a620a8012102b6cdd1a6cd129ef796faeedb0b840fcd0ca00c57e16e38e46ee7028d59812ae7ffffffff0220a10700000000001976a914c342bcd1a7784d9842f7386b8b3b8a3d4171a06e88ac59611100000000001976a91449f8c749a9960dc29b5cbe7d2397cea7d26611bb88ac00000000', 'hex');
|
|
|
|
var sandbox;
|
|
beforeEach(function() {
|
|
sandbox = sinon.sandbox.create();
|
|
dbService = new DBService({
|
|
node: {
|
|
services: [],
|
|
datadir: '/tmp',
|
|
network: 'regtest',
|
|
on: sinon.stub()
|
|
}
|
|
});
|
|
});
|
|
|
|
afterEach(function() {
|
|
sandbox.restore();
|
|
});
|
|
|
|
describe('#start', function() {
|
|
it('should start the db service by creating a db dir, ' +
|
|
' if necessary, and setting the store', function(done) {
|
|
dbService.start(function() {
|
|
dbService._store.should.be.instanceOf(Levelup);
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('#stop', function() {
|
|
it('should stop if store not open', function(done) {
|
|
dbService.stop(function() {
|
|
var close = sandbox.stub().callsArg(0);
|
|
dbService._store = { close: close };
|
|
dbService._stopping.should.be.true;
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should stop if store open', function(done) {
|
|
dbService.stop(function() {
|
|
var close = sandbox.stub().callsArg(0);
|
|
dbService._store = { close: close, isOpen: sinon.stub().returns(true) };
|
|
dbService._stopping.should.be.true;
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('#_onError', function() {
|
|
it('should stop the db', function() {
|
|
var stop = sandbox.stub();
|
|
dbService.node = { stop: stop };
|
|
dbService._onError(new Error('some error'));
|
|
stop.should.be.calledOnce;
|
|
});
|
|
});
|
|
|
|
describe('#_setDataPath', function() {
|
|
|
|
it('should set the data path', function() {
|
|
dbService._setDataPath();
|
|
dbService.dataPath.should.equal('/tmp/regtest/bitcorenode.db');
|
|
});
|
|
|
|
});
|
|
|
|
describe('#_setVersion', function() {
|
|
it('should set the version', function(done) {
|
|
var put = sandbox.stub(dbService, 'put').callsArgWith(2, null);
|
|
dbService._setVersion(function(err) {
|
|
put.should.be.calledOnce;
|
|
put.args[0][0].toString('hex').should.deep.equal('ffff76657273696f6e');
|
|
put.args[0][1].toString('hex').should.deep.equal('00000001');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('#get', function() {
|
|
it('should get a value from the db', function(done) {
|
|
var get = sandbox.stub().callsArgWith(2, null, 'data');
|
|
dbService._store = { get: get };
|
|
dbService.get('key', function(err, value) {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
value.should.equal('data');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should not get a value while the node is shutting down', function(done) {
|
|
dbService._stopping = true;
|
|
dbService.get('key', function(err, value) {
|
|
err.message.should.equal('Shutdown sequence underway, not able to complete the query');
|
|
done();
|
|
});
|
|
});
|
|
|
|
});
|
|
|
|
describe('#put', function() {
|
|
it('should put a value in the db', function(done) {
|
|
var put = sandbox.stub().callsArgWith(2, null);
|
|
dbService._store = { put: put };
|
|
dbService.put(new Buffer('key'), new Buffer('value'), function(err) {
|
|
if (err) {
|
|
return done(err);
|
|
}
|
|
put.should.be.calledOnce;
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should not allow an operation while the node is shutting down', function(done) {
|
|
dbService._stopping = true;
|
|
dbService.put(new Buffer('key'), new Buffer('value'), function(err) {
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('#batch', function() {
|
|
|
|
it('should save a batch of operations', function(done) {
|
|
|
|
var batch = sandbox.stub().callsArgWith(1, null);
|
|
dbService._store = { batch: batch };
|
|
|
|
dbService.batch([], function(err) {
|
|
|
|
if(err) {
|
|
return done(err);
|
|
}
|
|
|
|
batch.callCount.should.equal(1);
|
|
done();
|
|
|
|
|
|
});
|
|
});
|
|
|
|
it('should not call batch whilst shutting down', function(done) {
|
|
|
|
dbService._stopping = true;
|
|
|
|
var batch = sandbox.stub().callsArgWith(1, null);
|
|
dbService._store = { batch: batch };
|
|
|
|
dbService.batch(batch, function(err) {
|
|
|
|
if(err) {
|
|
return done(err);
|
|
}
|
|
|
|
batch.callCount.should.equal(0);
|
|
done();
|
|
|
|
|
|
});
|
|
|
|
});
|
|
});
|
|
|
|
describe('#createReadStream', function() {
|
|
|
|
it('should get a read stream', function() {
|
|
|
|
var on = sandbox.stub();
|
|
var stream = { on: on };
|
|
var createReadStream = sandbox.stub().returns(stream);
|
|
dbService._store = { createReadStream: createReadStream };
|
|
dbService.createReadStream([]).should.deep.equal(stream);
|
|
createReadStream.callCount.should.equal(1);
|
|
|
|
});
|
|
|
|
it('should not get a read stream if the node is stopping', function() {
|
|
|
|
dbService._stopping = true;
|
|
|
|
var on = sandbox.stub();
|
|
var stream = { on: on };
|
|
var createReadStream = sandbox.stub().returns(stream);
|
|
dbService._store = { createReadStream: createReadStream };
|
|
var stream = dbService.createReadStream([]);
|
|
expect(stream).to.be.undefined;
|
|
|
|
});
|
|
|
|
});
|
|
|
|
describe('#createKeyStream', function() {
|
|
|
|
it('should get a key stream', function() {
|
|
|
|
var on = sandbox.stub();
|
|
var stream = { on: on };
|
|
var createKeyStream = sandbox.stub().returns(stream);
|
|
dbService._store = { createKeyStream: createKeyStream };
|
|
dbService.createKeyStream([]).should.deep.equal(stream);
|
|
createKeyStream.callCount.should.equal(1);
|
|
|
|
});
|
|
|
|
it('should not get a key stream if the node is stopping', function() {
|
|
|
|
dbService._stopping = true;
|
|
|
|
var on = sandbox.stub();
|
|
var stream = { on: on };
|
|
var createKeyStream = sandbox.stub().returns(stream);
|
|
dbService._store = { createKeyStream: createKeyStream };
|
|
var stream = dbService.createKeyStream([]);
|
|
expect(stream).to.be.undefined;
|
|
|
|
});
|
|
});
|
|
|
|
describe('#close', function() {
|
|
this.timeout(3000);
|
|
it('should close the store if there is a store and it is open', function(done) {
|
|
|
|
var close = sandbox.stub().callsArgWith(0, null);
|
|
dbService._store = { isClosed: sinon.stub().returns(false), close: close };
|
|
|
|
dbService.close(function(err) {
|
|
if(err) {
|
|
return done(err);
|
|
}
|
|
close.callCount.should.equal(1);
|
|
done();
|
|
});
|
|
});
|
|
this.timeout(2000);
|
|
});
|
|
|
|
describe('#getServiceTip', function() {
|
|
it('should get service tip for previously saved', function(done) {
|
|
|
|
var tipBuf = Buffer.concat([ new Buffer('deadbeef', 'hex'), new Buffer(tx.txid(), 'hex') ]);
|
|
var get = sandbox.stub(dbService, 'get').callsArgWith(1, null, tipBuf);
|
|
dbService.getServiceTip('test', function(err, tip) {
|
|
|
|
if(err) {
|
|
return done(err);
|
|
}
|
|
|
|
get.callCount.should.equal(1);
|
|
tip.height.should.equal(0xdeadbeef);
|
|
tip.hash.should.equal(tx.txid());
|
|
done();
|
|
|
|
});
|
|
});
|
|
|
|
it('should get service tip for not previously saved', function(done) {
|
|
|
|
var get = sandbox.stub(dbService, 'get').callsArgWith(1, null, null);
|
|
dbService.getServiceTip('test', function(err, tip) {
|
|
|
|
if(err) {
|
|
return done(err);
|
|
}
|
|
|
|
get.callCount.should.equal(1);
|
|
tip.height.should.equal(0);
|
|
tip.hash.should.equal('0f9188f13cb7b2c71f2a335e3a4fc328bf5beb436012afca590b1a11466e2206');
|
|
done();
|
|
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('#getPrefix', function() {
|
|
|
|
it('should get the db prefix for a service when one already exists', function(done) {
|
|
var get = sandbox.stub(dbService, 'get').callsArgWith(1, null, new Buffer('0000', 'hex'));
|
|
dbService.getPrefix('test', function(err, prefix) {
|
|
|
|
if(err) {
|
|
return done(err);
|
|
}
|
|
|
|
get.callCount.should.equal(1);
|
|
prefix.toString('hex').should.equal('0000');
|
|
done();
|
|
});
|
|
});
|
|
|
|
it('should get the db prefix for a service when one does not already exist', function(done) {
|
|
var put = sandbox.stub(dbService, 'put').callsArgWith(2, null);
|
|
var get = sandbox.stub(dbService, 'get');
|
|
get.onCall(0).callsArgWith(1, null, null);
|
|
get.onCall(1).callsArgWith(1, null, new Buffer('eeee', 'hex'));
|
|
dbService.getPrefix('test', function(err, prefix) {
|
|
|
|
if(err) {
|
|
return done(err);
|
|
}
|
|
|
|
get.callCount.should.equal(2);
|
|
put.callCount.should.equal(2);
|
|
put.args[1][1].toString('hex').should.equal('eeef');
|
|
prefix.toString('hex').should.equal('eeee');
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|