Nuxt/test/utils/mock-log.js
2018-04-18 23:11:06 +08:00

25 lines
593 B
JavaScript

/* eslint-disable no-console */
export default function mockLog(levels = 'all', logger = console) {
if (levels === 'all') {
levels = ['trace', 'debug', 'log', 'info', 'warn', 'error']
} else if (typeof levels === 'string') {
levels = [levels]
}
beforeAll(() => {
for (let level of levels) {
jest.spyOn(logger, level).mockImplementation(() => {})
}
})
beforeEach(() => {
for (let level of levels) {
logger[level].mockClear()
}
})
afterAll(() => {
for (let level of levels) {
logger[level].mockRestore()
}
})
return logger
}