CyTube/test/testutil/db.js
2017-06-28 22:58:40 -07:00

35 lines
568 B
JavaScript

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