Nuxt/packages/cli/test/utils/mocking.js

114 lines
2.6 KiB
JavaScript
Raw Normal View History

import * as imports from '../../src/imports'
import Command from '../../src/command'
jest.mock('../../src/imports', () => {
return {
core: jest.fn().mockImplementation(() => ({
Nuxt () {}
})),
builder: jest.fn().mockImplementation(() => ({
Builder () {}
})),
generator: jest.fn().mockImplementation(() => ({
Generator () {}
})),
webpack: jest.fn().mockImplementation(() => ({
BundleBuilder () {}
}))
}
})
export const mockGetNuxt = (options = {}, implementation) => {
Command.prototype.getNuxt = jest.fn().mockImplementationOnce(() => {
return Object.assign({
hook: jest.fn(),
options
}, implementation)
})
}
export const mockGetBuilder = (ret) => {
const build = jest.fn().mockImplementationOnce(() => ret)
Command.prototype.getBuilder = jest.fn().mockImplementationOnce(() => ({ build }))
return build
}
export const mockGetGenerator = (ret) => {
const generate = jest.fn()
if (ret) {
generate.mockImplementationOnce(ret)
} else {
generate.mockImplementationOnce(() => ({ errors: [] }))
}
Command.prototype.getGenerator = jest.fn().mockImplementationOnce(() => ({ generate }))
return generate
}
export const mockGetNuxtStart = (ssr) => {
const listen = jest.fn().mockImplementationOnce(() => {
return Promise.resolve()
})
mockGetNuxt({
rootDir: '.',
render: {
ssr
}
}, {
server: {
listen,
2018-11-08 09:15:56 +00:00
listeners: []
}
})
2018-11-08 09:15:56 +00:00
return { listen }
}
export const mockGetNuxtConfig = () => {
const spy = jest.fn()
Command.prototype.getNuxtConfig = spy
return spy
}
export const mockNuxt = (implementation) => {
const Nuxt = function () {}
Object.assign(Nuxt.prototype, {
hook (type, fn) {
if (type === 'watch:restart') {
Nuxt.fileRestartHook = fn
}
},
2018-11-08 09:15:56 +00:00
options: {},
callHook: jest.fn(),
clearHook: jest.fn(),
clearHooks: jest.fn(),
close: jest.fn(),
ready: jest.fn(),
server: {
2018-11-08 09:15:56 +00:00
listeners: [],
listen: jest.fn().mockImplementationOnce(() => Promise.resolve())
}
}, implementation || {})
imports.core.mockImplementation(() => ({ Nuxt }))
return Nuxt
}
export const mockBuilder = (implementation) => {
const Builder = function () {}
Object.assign(Builder.prototype, {
build: jest.fn().mockImplementationOnce(() => Promise.resolve()),
unwatch: jest.fn().mockImplementationOnce(() => Promise.resolve()),
watchRestart: jest.fn().mockImplementationOnce(() => Promise.resolve())
}, implementation || {})
imports.builder.mockImplementation(() => ({ Builder }))
return Builder
}