X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;ds=sidebyside;f=tests%2Fpools%2Fthread%2Ffixed.test.js;h=84871e738d3d3184677b1110ad0078671f0e35a5;hb=76a3a76becd6ac41efb3271c708431a666828c30;hp=e4b99a6378d226535f83e47fb5516e3d394212f2;hpb=4e377863ebacc8cead467297d0365e93dec491c3;p=poolifier.git diff --git a/tests/pools/thread/fixed.test.js b/tests/pools/thread/fixed.test.js index e4b99a63..84871e73 100644 --- a/tests/pools/thread/fixed.test.js +++ b/tests/pools/thread/fixed.test.js @@ -1,7 +1,8 @@ const { expect } = require('expect') const { FixedThreadPool, 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 thread pool test suite', () => { const numberOfThreads = 6 @@ -27,7 +28,7 @@ describe('Fixed thread pool test suite', () => { const emptyPool = new FixedThreadPool( numberOfThreads, './tests/worker-files/thread/emptyWorker.js', - { exitHandler: () => console.log('empty pool worker exited') } + { exitHandler: () => console.info('empty pool worker exited') } ) const echoPool = new FixedThreadPool( numberOfThreads, @@ -64,11 +65,11 @@ describe('Fixed thread pool test suite', () => { it('Verify that the function is executed in a worker thread', 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) }) @@ -79,7 +80,7 @@ describe('Fixed thread pool test suite', () => { }) it("Verify that 'ready' event is emitted", async () => { - const pool1 = new FixedThreadPool( + const pool = new FixedThreadPool( numberOfThreads, './tests/worker-files/thread/testWorker.js', { @@ -87,17 +88,20 @@ describe('Fixed thread pool test suite', () => { } ) let poolReady = 0 - pool1.emitter.on(PoolEvents.ready, () => ++poolReady) - await waitPoolEvents(pool1, 'ready', 1) + pool.emitter.on(PoolEvents.ready, () => ++poolReady) + await waitPoolEvents(pool, PoolEvents.ready, 1) expect(poolReady).toBe(1) + await pool.destroy() }) - it("Verify that 'busy' event is emitted", () => { + 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 < numberOfThreads * 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 numberOfThreads + 1 times for a loop submitting up to numberOfThreads * 2 tasks to the fixed pool. expect(poolBusy).toBe(numberOfThreads + 1) @@ -111,6 +115,7 @@ describe('Fixed thread pool test suite', () => { } expect(promises.size).toBe(numberOfThreads * maxMultiplier) for (const workerNode of queuePool.workerNodes) { + expect(workerNode.usage.tasks.executing).toBeGreaterThanOrEqual(0) expect(workerNode.usage.tasks.executing).toBeLessThanOrEqual( queuePool.opts.tasksQueueOptions.concurrency ) @@ -121,7 +126,9 @@ describe('Fixed thread pool test suite', () => { expect(workerNode.usage.tasks.maxQueued).toBe( maxMultiplier - queuePool.opts.tasksQueueOptions.concurrency ) + expect(workerNode.usage.tasks.stolen).toBe(0) } + expect(queuePool.info.executedTasks).toBe(0) expect(queuePool.info.executingTasks).toBe( numberOfThreads * queuePool.opts.tasksQueueOptions.concurrency ) @@ -133,15 +140,30 @@ describe('Fixed thread pool test suite', () => { numberOfThreads * (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.executing).toBeGreaterThanOrEqual(0) + expect(workerNode.usage.tasks.executing).toBeLessThanOrEqual( + numberOfThreads * maxMultiplier + ) 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.usage.tasks.stolen).toBeGreaterThanOrEqual(0) + expect(workerNode.usage.tasks.stolen).toBeLessThanOrEqual( + numberOfThreads * maxMultiplier + ) } + expect(queuePool.info.executedTasks).toBe(numberOfThreads * maxMultiplier) + expect(queuePool.info.backPressure).toBe(false) + expect(queuePool.info.stolenTasks).toBeGreaterThanOrEqual(0) + expect(queuePool.info.stolenTasks).toBeLessThanOrEqual( + numberOfThreads * maxMultiplier + ) }) it('Verify that is possible to have a worker that return undefined', async () => { @@ -155,6 +177,32 @@ describe('Fixed thread pool test suite', () => { expect(result).toStrictEqual(data) }) + it('Verify that transferable objects are sent to the worker correctly', async () => { + let error + let result + try { + result = await pool.execute(undefined, undefined, [ + new ArrayBuffer(16), + new MessageChannel().port1 + ]) + } catch (e) { + error = e + } + expect(result).toStrictEqual({ ok: 1 }) + expect(error).toBeUndefined() + try { + result = await pool.execute(undefined, undefined, [ + new SharedArrayBuffer(16) + ]) + } catch (e) { + error = e + } + expect(result).toStrictEqual({ ok: 1 }) + expect(error).toStrictEqual( + new TypeError('Found invalid object in transferList') + ) + }) + it('Verify that error handling is working properly:sync', async () => { const data = { f: 10 } let taskError @@ -173,7 +221,7 @@ describe('Fixed thread pool test suite', () => { expect(typeof inError.message === 'string').toBe(true) expect(inError.message).toBe('Error Message from ThreadWorker') expect(taskError).toStrictEqual({ - name: 'default', + name: DEFAULT_TASK_NAME, message: new Error('Error Message from ThreadWorker'), data }) @@ -202,7 +250,7 @@ describe('Fixed thread pool test suite', () => { expect(typeof inError.message === 'string').toBe(true) expect(inError.message).toBe('Error Message from ThreadWorker:async') expect(taskError).toStrictEqual({ - name: 'default', + name: DEFAULT_TASK_NAME, message: new Error('Error Message from ThreadWorker:async'), data }) @@ -224,38 +272,55 @@ describe('Fixed thread pool test suite', () => { it('Shutdown test', async () => { const exitPromise = waitWorkerEvents(pool, 'exit', numberOfThreads) + let poolDestroy = 0 + pool.emitter.on(PoolEvents.destroy, () => ++poolDestroy) await pool.destroy() const numberOfExitEvents = await exitPromise + expect(pool.started).toBe(false) + expect(pool.workerNodes.length).toBe(0) expect(numberOfExitEvents).toBe(numberOfThreads) + expect(poolDestroy).toBe(1) }) it('Verify that thread pool options are checked', async () => { const workerFilePath = './tests/worker-files/thread/testWorker.js' - let pool1 = new FixedThreadPool(numberOfThreads, workerFilePath) - expect(pool1.opts.workerOptions).toBeUndefined() - await pool1.destroy() - pool1 = new FixedThreadPool(numberOfThreads, workerFilePath, { + let pool = new FixedThreadPool(numberOfThreads, workerFilePath) + expect(pool.opts.workerOptions).toBeUndefined() + await pool.destroy() + pool = new FixedThreadPool(numberOfThreads, workerFilePath, { workerOptions: { env: { TEST: 'test' }, name: 'test' } }) - expect(pool1.opts.workerOptions).toStrictEqual({ + expect(pool.opts.workerOptions).toStrictEqual({ env: { TEST: 'test' }, name: 'test' }) - await pool1.destroy() + await pool.destroy() }) it('Should work even without opts in input', async () => { - const pool1 = new FixedThreadPool( - numberOfThreads, - './tests/worker-files/thread/testWorker.js' - ) - const res = await pool1.execute() + const workerFilePath = './tests/worker-files/thread/testWorker.js' + const pool = new FixedThreadPool(numberOfThreads, workerFilePath) + const res = await pool.execute() expect(res).toStrictEqual({ ok: 1 }) // We need to clean up the resources after our test - await pool1.destroy() + await pool.destroy() + }) + + it('Verify destroyWorkerNode()', async () => { + const workerFilePath = './tests/worker-files/thread/testWorker.js' + const pool = new FixedThreadPool(numberOfThreads, workerFilePath) + const workerNodeKey = 0 + let exitEvent = 0 + pool.workerNodes[workerNodeKey].worker.on('exit', () => { + ++exitEvent + }) + await expect(pool.destroyWorkerNode(workerNodeKey)).resolves.toBeUndefined() + expect(exitEvent).toBe(1) + expect(pool.workerNodes.length).toBe(numberOfThreads - 1) + await pool.destroy() }) it('Verify that a pool with zero worker fails', async () => {