X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fworker%2Fabstract-worker.test.js;h=e9da7a874de95d1739be54216ed1a540a3aff9d3;hb=440dd7d77da6e027b3b9543504addcf805f76c1d;hp=9c8dd0973e9fe43ab0a9fe50e13b89b2fbe669be;hpb=e1ffb94fcac39a7459c47edaa086a7729a1dec44;p=poolifier.git diff --git a/tests/worker/abstract-worker.test.js b/tests/worker/abstract-worker.test.js index 9c8dd097..e9da7a87 100644 --- a/tests/worker/abstract-worker.test.js +++ b/tests/worker/abstract-worker.test.js @@ -2,24 +2,18 @@ const { expect } = require('expect') const { ClusterWorker, KillBehaviors, ThreadWorker } = require('../../lib') describe('Abstract worker test suite', () => { - class StubPoolWithIsMainWorker extends ThreadWorker { + class StubWorkerWithMainWorker extends ThreadWorker { constructor (fn, opts) { super(fn, opts) - this.mainWorker = false + this.mainWorker = undefined } } - it('Verify that fn function is mandatory', () => { - expect(() => new ClusterWorker()).toThrowError( - new Error('fn parameter is mandatory') - ) - }) - it('Verify worker options default values', () => { const worker = new ThreadWorker(() => {}) - expect(worker.opts.maxInactiveTime).toBe(1000 * 60) + expect(worker.opts.maxInactiveTime).toStrictEqual(60000) expect(worker.opts.killBehavior).toBe(KillBehaviors.SOFT) - expect(worker.opts.async).toBe(false) + expect(worker.opts.async).toBe(undefined) }) it('Verify that worker options are set at worker creation', () => { @@ -28,20 +22,240 @@ describe('Abstract worker test suite', () => { async: true, killBehavior: KillBehaviors.HARD }) - expect(worker.opts.maxInactiveTime).toBe(6000) + expect(worker.opts.maxInactiveTime).toStrictEqual(6000) expect(worker.opts.killBehavior).toBe(KillBehaviors.HARD) - expect(worker.opts.async).toBe(true) + expect(worker.opts.async).toBe(undefined) }) - it('Verify that handleError function is working properly', () => { - const error = new Error('My error') + it('Verify that taskFunctions parameter is mandatory', () => { + expect(() => new ClusterWorker()).toThrowError( + 'taskFunctions parameter is mandatory' + ) + }) + + it('Verify that taskFunctions parameter is a function or a plain object', () => { + expect(() => new ClusterWorker(0)).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + expect(() => new ClusterWorker('')).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + expect(() => new ClusterWorker(true)).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + expect(() => new ClusterWorker([])).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + expect(() => new ClusterWorker(new Map())).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + expect(() => new ClusterWorker(new Set())).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + expect(() => new ClusterWorker(new WeakMap())).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + expect(() => new ClusterWorker(new WeakSet())).toThrowError( + new TypeError( + 'taskFunctions parameter is not a function or a plain object' + ) + ) + }) + + it('Verify that taskFunctions parameter is not an empty object', () => { + expect(() => new ClusterWorker({})).toThrowError( + new Error('taskFunctions parameter object is empty') + ) + }) + + it('Verify that taskFunctions parameter with unique function is taken', () => { const worker = new ThreadWorker(() => {}) - expect(worker.handleError(error)).toBe(error) + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.size).toBe(2) + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + }) + + it('Verify that taskFunctions parameter with multiple task functions contains function', () => { + const fn1 = () => { + return 1 + } + const fn2 = '' + expect(() => new ThreadWorker({ fn1, fn2 })).toThrowError( + new TypeError('A taskFunctions parameter object value is not a function') + ) + }) + + it('Verify that taskFunctions parameter with multiple task functions is taken', () => { + const fn1 = () => { + return 1 + } + const fn2 = () => { + return 2 + } + const worker = new ClusterWorker({ fn1, fn2 }) + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn2')).toBeInstanceOf(Function) + expect(worker.taskFunctions.size).toBe(3) + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + }) + + it('Verify that handleError() method works properly', () => { + const error = new Error('Error as an error') + const worker = new ClusterWorker(() => {}) + expect(worker.handleError(error)).not.toBeInstanceOf(Error) + expect(worker.handleError(error)).toStrictEqual(error.message) + const errorMessage = 'Error as a string' + expect(worker.handleError(errorMessage)).toStrictEqual(errorMessage) }) - it('Verify that get main worker throw error if main worker is not set', () => { + it('Verify that getMainWorker() throw error if main worker is not set', () => { expect(() => - new StubPoolWithIsMainWorker(() => {}).getMainWorker() - ).toThrowError(new Error('Main worker was not set')) + new StubWorkerWithMainWorker(() => {}).getMainWorker() + ).toThrowError('Main worker not set') + }) + + it('Verify that hasTaskFunction() works', () => { + const fn1 = () => { + return 1 + } + const fn2 = () => { + return 2 + } + const worker = new ClusterWorker({ fn1, fn2 }) + expect(worker.hasTaskFunction('default')).toBe(true) + expect(worker.hasTaskFunction('fn1')).toBe(true) + expect(worker.hasTaskFunction('fn2')).toBe(true) + expect(worker.hasTaskFunction('fn3')).toBe(false) + }) + + it('Verify that addTaskFunction() works', () => { + const fn1 = () => { + return 1 + } + const fn2 = () => { + return 2 + } + const fn1Replacement = () => { + return 3 + } + const worker = new ThreadWorker(fn1) + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.size).toBe(2) + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + expect(() => worker.addTaskFunction('default', fn2)).toThrowError( + new Error('Cannot add a task function with the default reserved name') + ) + worker.addTaskFunction('fn2', fn2) + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn2')).toBeInstanceOf(Function) + expect(worker.taskFunctions.size).toBe(3) + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + worker.addTaskFunction('fn1', fn1Replacement) + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn2')).toBeInstanceOf(Function) + expect(worker.taskFunctions.size).toBe(3) + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + }) + + it('Verify that removeTaskFunction() works', () => { + const fn1 = () => { + return 1 + } + const fn2 = () => { + return 2 + } + const worker = new ClusterWorker({ fn1, fn2 }) + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn2')).toBeInstanceOf(Function) + expect(worker.taskFunctions.size).toBe(3) + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + expect(() => worker.removeTaskFunction('default')).toThrowError( + new Error( + 'Cannot remove the task function with the default reserved name' + ) + ) + expect(() => worker.removeTaskFunction('fn1')).toThrowError( + new Error( + 'Cannot remove the task function used as the default task function' + ) + ) + worker.removeTaskFunction('fn2') + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn2')).toBeUndefined() + expect(worker.taskFunctions.size).toBe(2) + }) + + it('Verify that listTaskFunctions() works', () => { + const fn1 = () => { + return 1 + } + const fn2 = () => { + return 2 + } + const worker = new ClusterWorker({ fn1, fn2 }) + expect(worker.listTaskFunctions()).toStrictEqual(['default', 'fn1', 'fn2']) + }) + + it('Verify that setDefaultTaskFunction() works', () => { + const fn1 = () => { + return 1 + } + const fn2 = () => { + return 2 + } + const worker = new ThreadWorker({ fn1, fn2 }) + expect(worker.taskFunctions.get('default')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get('fn2')).toBeInstanceOf(Function) + expect(worker.taskFunctions.size).toBe(3) + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + expect(() => worker.setDefaultTaskFunction('default')).toThrowError( + new Error( + 'Cannot set the default task function reserved name as the default task function' + ) + ) + worker.setDefaultTaskFunction('fn1') + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn1') + ) + worker.setDefaultTaskFunction('fn2') + expect(worker.taskFunctions.get('default')).toStrictEqual( + worker.taskFunctions.get('fn2') + ) }) })