CyTube/test/testutil/db.js
2017-05-31 22:46:15 -07:00

25 lines
497 B
JavaScript

const Promise = require('bluebird');
function MockDB(mockTx) {
this.mockTx = mockTx;
}
MockDB.prototype = {
runTransaction: function runTransaction(fn) {
return fn(this.mockTx);
}
};
function MockTx() {
}
['insert', 'update', 'select', 'del', 'where', 'table'].forEach(method => {
MockTx.prototype[method] = function () {
return Promise.reject(new Error(`No stub defined for method "${method}"`));
};
});
exports.MockDB = MockDB;
exports.MockTx = MockTx;