35 lines
568 B
JavaScript
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;
|