mirror of https://github.com/nuxt/nuxt.git
118 lines
3.3 KiB
JavaScript
118 lines
3.3 KiB
JavaScript
import consola from 'consola'
|
|
import { sequence, parallel, chainFn } from '../src/task'
|
|
|
|
describe('util: task', () => {
|
|
beforeEach(() => {
|
|
jest.clearAllMocks()
|
|
})
|
|
|
|
test('should call fn in sequence', async () => {
|
|
const fn = jest.fn(consola.log)
|
|
await sequence([1, 2, 3, 4], fn)
|
|
|
|
expect(fn).toBeCalledTimes(4)
|
|
expect(consola.log).toBeCalledTimes(4)
|
|
expect(consola.log).nthCalledWith(1, 1)
|
|
expect(consola.log).nthCalledWith(2, 2)
|
|
expect(consola.log).nthCalledWith(3, 3)
|
|
expect(consola.log).nthCalledWith(4, 4)
|
|
})
|
|
|
|
test('should call fn in parallel', async () => {
|
|
jest.spyOn(Promise, 'all')
|
|
jest.spyOn(Promise, 'resolve')
|
|
|
|
await parallel([1, 2, 3, 4], (num, index) => [num, index])
|
|
|
|
expect(Promise.all).toBeCalledTimes(1)
|
|
expect(Promise.resolve).toBeCalledTimes(4)
|
|
expect(Promise.resolve).nthCalledWith(1, [1, 0])
|
|
expect(Promise.resolve).nthCalledWith(2, [2, 1])
|
|
expect(Promise.resolve).nthCalledWith(3, [3, 2])
|
|
expect(Promise.resolve).nthCalledWith(4, [4, 3])
|
|
|
|
Promise.all.mockRestore()
|
|
Promise.resolve.mockRestore()
|
|
})
|
|
|
|
test('chainFn (mutate, mutate)', () => {
|
|
// Pass more than one argument to test that they're actually taken into account
|
|
const firstFn = function (obj, count) {
|
|
obj.foo = count + 1
|
|
}
|
|
const secondFn = function (obj, count) {
|
|
obj.bar = count + 2
|
|
}
|
|
|
|
const chainedFn = chainFn(firstFn, secondFn)
|
|
expect(chainedFn({}, 10)).toEqual({ foo: 11, bar: 12 })
|
|
})
|
|
|
|
test('chainFn (mutate, return)', () => {
|
|
const firstFn = function (obj, count) {
|
|
obj.foo = count + 1
|
|
}
|
|
const secondFn = function (obj, count) {
|
|
return Object.assign({}, obj, { bar: count + 2 })
|
|
}
|
|
|
|
const chainedFn = chainFn(firstFn, secondFn)
|
|
expect(chainedFn({}, 10)).toEqual({ foo: 11, bar: 12 })
|
|
})
|
|
|
|
test('chainFn (return, mutate)', () => {
|
|
const firstFn = function (obj, count) {
|
|
return Object.assign({}, obj, { foo: count + 1 })
|
|
}
|
|
const secondFn = function (obj, count) {
|
|
obj.bar = count + 2
|
|
}
|
|
|
|
const chainedFn = chainFn(firstFn, secondFn)
|
|
expect(chainedFn({}, 10)).toEqual({ foo: 11, bar: 12 })
|
|
})
|
|
|
|
test('chainFn (return, return)', () => {
|
|
const firstFn = function (obj, count) {
|
|
return Object.assign({}, obj, { foo: count + 1 })
|
|
}
|
|
const secondFn = function (obj, count) {
|
|
return Object.assign({}, obj, { bar: count + 2 })
|
|
}
|
|
|
|
const chainedFn = chainFn(firstFn, secondFn)
|
|
expect(chainedFn({}, 10)).toEqual({ foo: 11, bar: 12 })
|
|
})
|
|
|
|
test('chainFn (return, non-function)', () => {
|
|
const firstFn = function (obj, count) {
|
|
return Object.assign({}, obj, { foo: count + 1 })
|
|
}
|
|
const secondFn = ''
|
|
|
|
const chainedFn = chainFn(firstFn, secondFn)
|
|
expect(chainedFn).toBe(firstFn)
|
|
})
|
|
|
|
test('chainFn (non-function, return)', () => {
|
|
const firstFn = ''
|
|
const secondFn = function (obj, count) {
|
|
return Object.assign({}, obj, { bar: count + 2 })
|
|
}
|
|
|
|
const chainedFn = chainFn(firstFn, secondFn)
|
|
expect(chainedFn({}, 10)).toEqual({ bar: 12 })
|
|
})
|
|
|
|
test('chainFn (promise)', async () => {
|
|
const firstFn = () => Promise.resolve({ foo: 1 })
|
|
const secondFn = function (obj) {
|
|
obj.foo++
|
|
return Promise.resolve()
|
|
}
|
|
|
|
const chainedFn = chainFn(firstFn, secondFn)
|
|
expect(await chainedFn()).toEqual({ foo: 2 })
|
|
})
|
|
})
|