流程.js
const { getConnection, closeConnection } = require('./utils/db-connection.js');
const getQueryCount = query => new Promise((resolve, rejects) => {
getConnection().then((conn) => {
conn.query(query, (err, count) => {
closeConnection(conn);
if (err) {
log.info(`Get Count Error: ${err}`);
return rejects(err);
}
return resolve(count[0][1]);
});
});
});
Run Code Online (Sandbox Code Playgroud)
流程.test.js
const dbConn = require('../src/utils/db-connection.js');
describe('all count', () => {
beforeEach(() => {
sinon.stub(dbConn, 'getConnection').resolves({ query: sinon.stub().yields(null, [[0, 5]]) });
sinon.stub(dbConn, 'closeConnection');
const process = require('./src/process'); //module is loaded after dependencies are stubbed
});
it('getQueryCount', () => { …
Run Code Online (Sandbox Code Playgroud)