X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fworker%2Fcluster-worker.test.mjs;h=2ad7911380c9844a12157e0c50f1232f9e991a27;hb=884743b122a66be45c94d83d9230e28a9ab43836;hp=b785010ec5cf7b61999852c53658091abba0e5f5;hpb=f8b3d84c1c95773a1e18d2c025d89d73c4f2ab15;p=poolifier.git diff --git a/tests/worker/cluster-worker.test.mjs b/tests/worker/cluster-worker.test.mjs index b785010e..2ad79113 100644 --- a/tests/worker/cluster-worker.test.mjs +++ b/tests/worker/cluster-worker.test.mjs @@ -1,21 +1,30 @@ import { expect } from 'expect' import { restore, stub } from 'sinon' -import { ClusterWorker } from '../../lib/index.js' -import { DEFAULT_TASK_NAME } from '../../lib/utils.js' + +import { ClusterWorker } from '../../lib/index.cjs' +import { DEFAULT_TASK_NAME } from '../../lib/utils.cjs' describe('Cluster worker test suite', () => { afterEach(() => { restore() }) + it('Verify worker properties value after initialization', () => { + const worker = new ClusterWorker(() => {}) + expect(worker.isMain).toBe(true) + expect(worker.mainWorker).toBe(undefined) + expect(worker.taskFunctions).toBeInstanceOf(Map) + expect(worker.taskFunctions.size).toBe(2) + }) + it('Verify that sync kill handler is called when worker is killed', () => { const worker = new ClusterWorker(() => {}, { - killHandler: stub().returns() + killHandler: stub().returns(), }) worker.isMain = false worker.getMainWorker = stub().returns({ id: 1, - send: stub().returns() + send: stub().returns(), }) worker.handleKillMessage() expect(worker.getMainWorker.calledTwice).toBe(true) @@ -33,19 +42,25 @@ describe('Cluster worker test suite', () => { const worker = new ClusterWorker({ fn1, fn2 }) worker.getMainWorker = stub().returns({ id: 1, - send: stub().returns() + send: stub().returns(), }) expect(worker.removeTaskFunction(0, fn1)).toStrictEqual({ status: false, - error: new TypeError('name parameter is not a string') + error: new TypeError('name parameter is not a string'), }) expect(worker.removeTaskFunction('', fn1)).toStrictEqual({ status: false, - error: new TypeError('name parameter is an empty string') + error: new TypeError('name parameter is an empty string'), + }) + expect(worker.taskFunctions.get(DEFAULT_TASK_NAME)).toStrictEqual({ + taskFunction: expect.any(Function), + }) + expect(worker.taskFunctions.get('fn1')).toStrictEqual({ + taskFunction: expect.any(Function), + }) + expect(worker.taskFunctions.get('fn2')).toStrictEqual({ + taskFunction: expect.any(Function), }) - expect(worker.taskFunctions.get(DEFAULT_TASK_NAME)).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_TASK_NAME)).toStrictEqual( worker.taskFunctions.get('fn1') @@ -54,17 +69,21 @@ describe('Cluster worker test suite', () => { status: false, error: new Error( 'Cannot remove the task function with the default reserved name' - ) + ), }) expect(worker.removeTaskFunction('fn1')).toStrictEqual({ status: false, error: new Error( 'Cannot remove the task function used as the default task function' - ) + ), }) worker.removeTaskFunction('fn2') - expect(worker.taskFunctions.get(DEFAULT_TASK_NAME)).toBeInstanceOf(Function) - expect(worker.taskFunctions.get('fn1')).toBeInstanceOf(Function) + expect(worker.taskFunctions.get(DEFAULT_TASK_NAME)).toStrictEqual({ + taskFunction: expect.any(Function), + }) + expect(worker.taskFunctions.get('fn1')).toStrictEqual({ + taskFunction: expect.any(Function), + }) expect(worker.taskFunctions.get('fn2')).toBeUndefined() expect(worker.taskFunctions.size).toBe(2) expect(worker.getMainWorker.calledTwice).toBe(true) @@ -83,7 +102,7 @@ describe('Cluster worker test suite', () => { it('Verify that sendToMainWorker() method invokes the getMainWorker() and send() methods', () => { const worker = new ClusterWorker(() => {}) worker.getMainWorker = stub().returns({ - send: stub().returns() + send: stub().returns(), }) worker.sendToMainWorker({ ok: 1 }) expect(worker.getMainWorker.calledTwice).toBe(true)