X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fpools%2Fcluster%2Fdynamic.test.mjs;h=540223bf2840ab384e11d4b077a5f84cd27a8144;hb=c1d4808dda55cd9ca2c7fdd5ccc330c3a7642a35;hp=fa318f4bc62d451f2f08661ff8bd291e65f48b2f;hpb=28881126dd2ce52b9879cd30b3e3ef414ac1b074;p=poolifier.git diff --git a/tests/pools/cluster/dynamic.test.mjs b/tests/pools/cluster/dynamic.test.mjs index fa318f4b..540223bf 100644 --- a/tests/pools/cluster/dynamic.test.mjs +++ b/tests/pools/cluster/dynamic.test.mjs @@ -1,7 +1,11 @@ import { expect } from 'expect' -import { DynamicClusterPool, PoolEvents } from '../../../lib/index.cjs' +import { + DynamicClusterPool, + PoolEvents, + WorkerChoiceStrategies +} from '../../../lib/index.cjs' import { TaskFunctions } from '../../test-types.cjs' -import { sleep, waitWorkerEvents } from '../../test-utils.cjs' +import { sleep, waitPoolEvents, waitWorkerEvents } from '../../test-utils.cjs' describe('Dynamic cluster pool test suite', () => { const min = 1 @@ -153,27 +157,36 @@ describe('Dynamic cluster pool test suite', () => { await pool.destroy() }) - it.skip('Verify that a pool with zero worker works', async () => { - const pool = new DynamicClusterPool( - 0, - max, - './tests/worker-files/thread/testWorker.mjs' - ) - expect(pool.starting).toBe(false) - expect(pool.workerNodes.length).toBe(pool.info.minSize) - for (let run = 0; run < 4; run++) { - // pool.enableTasksQueue(true, { concurrency: 2 }) - const maxMultiplier = 10000 - const promises = new Set() - for (let i = 0; i < max * maxMultiplier; i++) { - promises.add(pool.execute()) + it('Verify that a pool with zero worker works', async () => { + for (const workerChoiceStrategy of Object.values(WorkerChoiceStrategies)) { + const pool = new DynamicClusterPool( + 0, + max, + './tests/worker-files/cluster/testWorker.cjs', + { + workerChoiceStrategy + } + ) + expect(pool.starting).toBe(false) + expect(pool.readyEventEmitted).toBe(false) + for (let run = 0; run < 2; run++) { + run % 2 !== 0 && pool.enableTasksQueue(true) + const maxMultiplier = 4 + const promises = new Set() + expect(pool.workerNodes.length).toBe(pool.info.minSize) + for (let i = 0; i < max * maxMultiplier; i++) { + promises.add(pool.execute()) + } + await Promise.all(promises) + expect(pool.readyEventEmitted).toBe(true) + expect(pool.workerNodes.length).toBeGreaterThan(pool.info.minSize) + expect(pool.workerNodes.length).toBeLessThanOrEqual(pool.info.maxSize) + await waitPoolEvents(pool, PoolEvents.empty, 1) + expect(pool.readyEventEmitted).toBe(false) + expect(pool.workerNodes.length).toBe(pool.info.minSize) } - await Promise.all(promises) - expect(pool.workerNodes.length).toBe(max) - await waitWorkerEvents(pool, 'exit', max) - expect(pool.workerNodes.length).toBe(pool.info.minSize) + // We need to clean up the resources after our test + await pool.destroy() } - // We need to clean up the resources after our test - await pool.destroy() }) })