X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fpools%2Fcluster%2Ffixed.test.js;h=42e13a9a83348f8b4d8c425f4f71ae32c77d6a7e;hb=6cd5248f7b289828220cac63f8fe77d021c572e9;hp=6daec1400dbc9dc0fdb921d6b25b53dd9d751eab;hpb=216541b6ef3426586d4c47bd52845e83bd072d15;p=poolifier.git diff --git a/tests/pools/cluster/fixed.test.js b/tests/pools/cluster/fixed.test.js index 6daec140..42e13a9a 100644 --- a/tests/pools/cluster/fixed.test.js +++ b/tests/pools/cluster/fixed.test.js @@ -1,35 +1,34 @@ const { expect } = require('expect') const { FixedClusterPool, PoolEvents } = require('../../../lib') -const { WorkerFunctions } = require('../../test-types') +const { TaskFunctions } = require('../../test-types') const { waitPoolEvents, waitWorkerEvents } = require('../../test-utils') +const { DEFAULT_TASK_NAME } = require('../../../lib/utils') -describe('Fixed cluster pool test suite', async () => { - const numberOfWorkers = 6 +describe('Fixed cluster pool test suite', () => { + const numberOfWorkers = 8 + const tasksConcurrency = 2 const pool = new FixedClusterPool( numberOfWorkers, './tests/worker-files/cluster/testWorker.js', { - errorHandler: e => console.error(e) + errorHandler: (e) => console.error(e) } ) - let poolReady = 0 - pool.emitter.on(PoolEvents.ready, () => ++poolReady) - await waitPoolEvents(pool, PoolEvents.ready, 1) const queuePool = new FixedClusterPool( numberOfWorkers, './tests/worker-files/cluster/testWorker.js', { enableTasksQueue: true, tasksQueueOptions: { - concurrency: 2 + concurrency: tasksConcurrency }, - errorHandler: e => console.error(e) + errorHandler: (e) => console.error(e) } ) const emptyPool = new FixedClusterPool( numberOfWorkers, './tests/worker-files/cluster/emptyWorker.js', - { exitHandler: () => console.log('empty pool worker exited') } + { exitHandler: () => console.info('empty pool worker exited') } ) const echoPool = new FixedClusterPool( numberOfWorkers, @@ -39,14 +38,14 @@ describe('Fixed cluster pool test suite', async () => { numberOfWorkers, './tests/worker-files/cluster/errorWorker.js', { - errorHandler: e => console.error(e) + errorHandler: (e) => console.error(e) } ) const asyncErrorPool = new FixedClusterPool( numberOfWorkers, './tests/worker-files/cluster/asyncErrorWorker.js', { - errorHandler: e => console.error(e) + errorHandler: (e) => console.error(e) } ) const asyncPool = new FixedClusterPool( @@ -66,11 +65,11 @@ describe('Fixed cluster pool test suite', async () => { it('Verify that the function is executed in a worker cluster', async () => { let result = await pool.execute({ - function: WorkerFunctions.fibonacci + function: TaskFunctions.fibonacci }) expect(result).toBe(75025) result = await pool.execute({ - function: WorkerFunctions.factorial + function: TaskFunctions.factorial }) expect(result).toBe(9.33262154439441e157) }) @@ -81,15 +80,27 @@ describe('Fixed cluster pool test suite', async () => { }) it("Verify that 'ready' event is emitted", async () => { + const pool1 = new FixedClusterPool( + numberOfWorkers, + './tests/worker-files/cluster/testWorker.js', + { + errorHandler: (e) => console.error(e) + } + ) + let poolReady = 0 + pool1.emitter.on(PoolEvents.ready, () => ++poolReady) + await waitPoolEvents(pool1, PoolEvents.ready, 1) expect(poolReady).toBe(1) }) it("Verify that 'busy' event is emitted", async () => { + const promises = new Set() let poolBusy = 0 pool.emitter.on(PoolEvents.busy, () => ++poolBusy) for (let i = 0; i < numberOfWorkers * 2; i++) { - pool.execute() + promises.add(pool.execute()) } + await Promise.all(promises) // The `busy` event is triggered when the number of submitted tasks at once reach the number of fixed pool workers. // So in total numberOfWorkers + 1 times for a loop submitting up to numberOfWorkers * 2 tasks to the fixed pool. expect(poolBusy).toBe(numberOfWorkers + 1) @@ -97,34 +108,61 @@ describe('Fixed cluster pool test suite', async () => { it('Verify that tasks queuing is working', async () => { const promises = new Set() - const maxMultiplier = 2 + const maxMultiplier = 3 // Must be greater than tasksConcurrency for (let i = 0; i < numberOfWorkers * maxMultiplier; i++) { promises.add(queuePool.execute()) } expect(promises.size).toBe(numberOfWorkers * maxMultiplier) for (const workerNode of queuePool.workerNodes) { + expect(workerNode.usage.tasks.executing).toBeGreaterThanOrEqual(0) expect(workerNode.usage.tasks.executing).toBeLessThanOrEqual( queuePool.opts.tasksQueueOptions.concurrency ) expect(workerNode.usage.tasks.executed).toBe(0) - expect(workerNode.usage.tasks.queued).toBeGreaterThan(0) - expect(workerNode.usage.tasks.maxQueued).toBeGreaterThan(0) + expect(workerNode.usage.tasks.queued).toBe( + maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency + ) + expect(workerNode.usage.tasks.maxQueued).toBe( + maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency + ) + expect(workerNode.usage.tasks.stolen).toBe(0) } - expect(queuePool.info.executingTasks).toBe(numberOfWorkers) + expect(queuePool.info.executedTasks).toBe(0) + expect(queuePool.info.executingTasks).toBe( + numberOfWorkers * queuePool.opts.tasksQueueOptions.concurrency + ) expect(queuePool.info.queuedTasks).toBe( - numberOfWorkers * maxMultiplier - numberOfWorkers + numberOfWorkers * + (maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency) ) expect(queuePool.info.maxQueuedTasks).toBe( - numberOfWorkers * maxMultiplier - numberOfWorkers + numberOfWorkers * + (maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency) ) + expect(queuePool.info.backPressure).toBe(false) + expect(queuePool.info.stolenTasks).toBe(0) await Promise.all(promises) for (const workerNode of queuePool.workerNodes) { - expect(workerNode.usage.tasks.executing).toBe(0) - expect(workerNode.usage.tasks.executed).toBeGreaterThan(0) - expect(workerNode.usage.tasks.executed).toBeLessThanOrEqual(maxMultiplier) + expect(workerNode.usage.tasks.executing).toBeGreaterThanOrEqual(0) + expect(workerNode.usage.tasks.executing).toBeLessThanOrEqual( + numberOfWorkers * maxMultiplier + ) + expect(workerNode.usage.tasks.executed).toBe(maxMultiplier) expect(workerNode.usage.tasks.queued).toBe(0) - expect(workerNode.usage.tasks.maxQueued).toBe(1) + expect(workerNode.usage.tasks.maxQueued).toBe( + maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency + ) + expect(workerNode.usage.tasks.stolen).toBeGreaterThanOrEqual(0) + expect(workerNode.usage.tasks.stolen).toBeLessThanOrEqual( + numberOfWorkers * maxMultiplier + ) } + expect(queuePool.info.executedTasks).toBe(numberOfWorkers * maxMultiplier) + expect(queuePool.info.backPressure).toBe(false) + expect(queuePool.info.stolenTasks).toBeGreaterThanOrEqual(0) + expect(queuePool.info.stolenTasks).toBeLessThanOrEqual( + numberOfWorkers * maxMultiplier + ) }) it('Verify that is possible to have a worker that return undefined', async () => { @@ -141,7 +179,7 @@ describe('Fixed cluster pool test suite', async () => { it('Verify that error handling is working properly:sync', async () => { const data = { f: 10 } let taskError - errorPool.emitter.on(PoolEvents.taskError, e => { + errorPool.emitter.on(PoolEvents.taskError, (e) => { taskError = e }) let inError @@ -154,12 +192,13 @@ describe('Fixed cluster pool test suite', async () => { expect(typeof inError === 'string').toBe(true) expect(inError).toBe('Error Message from ClusterWorker') expect(taskError).toStrictEqual({ + name: DEFAULT_TASK_NAME, message: 'Error Message from ClusterWorker', data }) expect( errorPool.workerNodes.some( - workerNode => workerNode.usage.tasks.failed === 1 + (workerNode) => workerNode.usage.tasks.failed === 1 ) ).toBe(true) }) @@ -167,7 +206,7 @@ describe('Fixed cluster pool test suite', async () => { it('Verify that error handling is working properly:async', async () => { const data = { f: 10 } let taskError - asyncErrorPool.emitter.on(PoolEvents.taskError, e => { + asyncErrorPool.emitter.on(PoolEvents.taskError, (e) => { taskError = e }) let inError @@ -180,12 +219,13 @@ describe('Fixed cluster pool test suite', async () => { expect(typeof inError === 'string').toBe(true) expect(inError).toBe('Error Message from ClusterWorker:async') expect(taskError).toStrictEqual({ + name: DEFAULT_TASK_NAME, message: 'Error Message from ClusterWorker:async', data }) expect( asyncErrorPool.workerNodes.some( - workerNode => workerNode.usage.tasks.failed === 1 + (workerNode) => workerNode.usage.tasks.failed === 1 ) ).toBe(true) }) @@ -201,9 +241,12 @@ describe('Fixed cluster pool test suite', async () => { it('Shutdown test', async () => { const exitPromise = waitWorkerEvents(pool, 'exit', numberOfWorkers) + let poolDestroy = 0 + pool.emitter.on(PoolEvents.destroy, () => ++poolDestroy) await pool.destroy() const numberOfExitEvents = await exitPromise expect(numberOfExitEvents).toBe(numberOfWorkers) + expect(poolDestroy).toBe(1) }) it('Verify that cluster pool options are checked', async () => {