All files / src AsyncSeriesWaterfallHook.ts

100% Statements 20/20
100% Branches 2/2
100% Functions 6/6
100% Lines 20/20

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 452x 2x       2x 4x   4x 4x   4x 4x 4x 4x     2x         2x         4x 1x     3x 3x       2x 2x   2x       2x  
import CodeFactory from './CodeFactory'
import Hook from './Hook'
 
class AsyncSeriesWaterfallHookCodeFactory extends CodeFactory {
    content({ onError, onResult }: CodeFactoryContent) {
        return this.callTapsSeries({
            onError: (_i, err, _next, doneBreak) => onError(err) + doneBreak(true),
            onResult: (_i, result, next) => {
                let code = ''
                code += `if(${result} !== undefined) {\n`
                // @ts-ignore
                code += `${this._args[0]} = ${result};\n`
                code += `}\n`
                code += next()
                return code
            },
            // @ts-ignore
            onDone: () => onResult(this._args[0])
        })
    }
}
 
const codeFactory = new AsyncSeriesWaterfallHookCodeFactory()
 
class AsyncSeriesWaterfallHook extends Hook {
    private fns: any[] | undefined
    constructor(args: string[], name?: string) {
        if (args.length < 1) {
            throw new Error('Waterfall hooks must have at least one argument')
        }
 
        super(args, name)
        this.forbiddenCall()
    }
 
    compile(options: HookCompileOptions) {
        this.fns = []
        codeFactory.setup(this.fns, options)
 
        return codeFactory.create(options)
    }
}
 
export default AsyncSeriesWaterfallHook