X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fpools%2Fcluster%2Ffixed.test.js;h=0415d9c1eaa7a94cdc1f1260a9bac060c5703c04;hb=4e377863ebacc8cead467297d0365e93dec491c3;hp=aa90c8ab4194ca4d4a07d06fbde9233f62a73067;hpb=df593701c4bd494b0e99372fdcc3708412799942;p=poolifier.git diff --git a/tests/pools/cluster/fixed.test.js b/tests/pools/cluster/fixed.test.js index aa90c8ab..0415d9c1 100644 --- a/tests/pools/cluster/fixed.test.js +++ b/tests/pools/cluster/fixed.test.js @@ -1,10 +1,11 @@ const { expect } = require('expect') const { FixedClusterPool, PoolEvents } = require('../../../lib') const { WorkerFunctions } = require('../../test-types') -const TestUtils = require('../../test-utils') +const { waitPoolEvents, waitWorkerEvents } = require('../../test-utils') describe('Fixed cluster pool test suite', () => { const numberOfWorkers = 6 + const tasksConcurrency = 2 const pool = new FixedClusterPool( numberOfWorkers, './tests/worker-files/cluster/testWorker.js', @@ -18,7 +19,7 @@ describe('Fixed cluster pool test suite', () => { { enableTasksQueue: true, tasksQueueOptions: { - concurrency: 2 + concurrency: tasksConcurrency }, errorHandler: e => console.error(e) } @@ -65,7 +66,7 @@ describe('Fixed cluster pool test suite', () => { let result = await pool.execute({ function: WorkerFunctions.fibonacci }) - expect(result).toBe(121393) + expect(result).toBe(75025) result = await pool.execute({ function: WorkerFunctions.factorial }) @@ -74,10 +75,24 @@ describe('Fixed cluster pool test suite', () => { it('Verify that is possible to invoke the execute() method without input', async () => { const result = await pool.execute() - expect(result).toBe(false) + expect(result).toStrictEqual({ ok: 1 }) }) - it("Verify that 'busy' event is emitted", 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", () => { let poolBusy = 0 pool.emitter.on(PoolEvents.busy, () => ++poolBusy) for (let i = 0; i < numberOfWorkers * 2; i++) { @@ -90,35 +105,42 @@ describe('Fixed cluster pool test suite', () => { 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.workerUsage.tasks.executing).toBeLessThanOrEqual( + expect(workerNode.usage.tasks.executing).toBeLessThanOrEqual( queuePool.opts.tasksQueueOptions.concurrency ) - expect(workerNode.workerUsage.tasks.executed).toBe(0) - expect(workerNode.workerUsage.tasks.queued).toBeGreaterThan(0) - expect(workerNode.workerUsage.tasks.maxQueued).toBeGreaterThan(0) + expect(workerNode.usage.tasks.executed).toBe(0) + expect(workerNode.usage.tasks.queued).toBe( + maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency + ) + expect(workerNode.usage.tasks.maxQueued).toBe( + maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency + ) } - expect(queuePool.info.executingTasks).toBe(numberOfWorkers) + 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) ) await Promise.all(promises) for (const workerNode of queuePool.workerNodes) { - expect(workerNode.workerUsage.tasks.executing).toBe(0) - expect(workerNode.workerUsage.tasks.executed).toBeGreaterThan(0) - expect(workerNode.workerUsage.tasks.executed).toBeLessThanOrEqual( - maxMultiplier + expect(workerNode.usage.tasks.executing).toBe(0) + expect(workerNode.usage.tasks.executed).toBe(maxMultiplier) + expect(workerNode.usage.tasks.queued).toBe(0) + expect(workerNode.usage.tasks.maxQueued).toBe( + maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency ) - expect(workerNode.workerUsage.tasks.queued).toBe(0) - expect(workerNode.workerUsage.tasks.maxQueued).toBe(1) } }) @@ -149,12 +171,13 @@ describe('Fixed cluster pool test suite', () => { expect(typeof inError === 'string').toBe(true) expect(inError).toBe('Error Message from ClusterWorker') expect(taskError).toStrictEqual({ + name: 'default', message: 'Error Message from ClusterWorker', data }) expect( errorPool.workerNodes.some( - workerNode => workerNode.workerUsage.tasks.failed === 1 + workerNode => workerNode.usage.tasks.failed === 1 ) ).toBe(true) }) @@ -175,12 +198,13 @@ describe('Fixed cluster pool test suite', () => { expect(typeof inError === 'string').toBe(true) expect(inError).toBe('Error Message from ClusterWorker:async') expect(taskError).toStrictEqual({ + name: 'default', message: 'Error Message from ClusterWorker:async', data }) expect( asyncErrorPool.workerNodes.some( - workerNode => workerNode.workerUsage.tasks.failed === 1 + workerNode => workerNode.usage.tasks.failed === 1 ) ).toBe(true) }) @@ -195,11 +219,7 @@ describe('Fixed cluster pool test suite', () => { }) it('Shutdown test', async () => { - const exitPromise = TestUtils.waitWorkerEvents( - pool, - 'exit', - numberOfWorkers - ) + const exitPromise = waitWorkerEvents(pool, 'exit', numberOfWorkers) await pool.destroy() const numberOfExitEvents = await exitPromise expect(numberOfExitEvents).toBe(numberOfWorkers) @@ -234,7 +254,7 @@ describe('Fixed cluster pool test suite', () => { './tests/worker-files/cluster/testWorker.js' ) const res = await pool1.execute() - expect(res).toBe(false) + expect(res).toStrictEqual({ ok: 1 }) // We need to clean up the resources after our test await pool1.destroy() }) @@ -243,6 +263,6 @@ describe('Fixed cluster pool test suite', () => { expect( () => new FixedClusterPool(0, './tests/worker-files/cluster/testWorker.js') - ).toThrowError('Cannot instantiate a fixed pool with no worker') + ).toThrowError('Cannot instantiate a fixed pool with zero worker') }) })