import { expect } from 'expect'
-import { DynamicClusterPool, PoolEvents } from '../../../lib/index.js'
-import { TaskFunctions } from '../../test-types.js'
-import { sleep, waitWorkerEvents } from '../../test-utils.js'
+
+import {
+ DynamicClusterPool,
+ PoolEvents,
+ WorkerChoiceStrategies,
+} from '../../../lib/index.cjs'
+import { TaskFunctions } from '../../test-types.cjs'
+import { sleep, waitPoolEvents, waitWorkerEvents } from '../../test-utils.cjs'
describe('Dynamic cluster pool test suite', () => {
const min = 1
const pool = new DynamicClusterPool(
min,
max,
- './tests/worker-files/cluster/testWorker.js',
+ './tests/worker-files/cluster/testWorker.cjs',
{
- errorHandler: e => console.error(e)
+ errorHandler: e => console.error(e),
}
)
it('Verify that the function is executed in a worker cluster', async () => {
let result = await pool.execute({
- function: TaskFunctions.fibonacci
+ function: TaskFunctions.fibonacci,
})
- expect(result).toBe(75025)
+ expect(result).toBe(354224848179262000000)
result = await pool.execute({
- function: TaskFunctions.factorial
+ function: TaskFunctions.factorial,
})
expect(result).toBe(9.33262154439441e157)
})
pool.emitter.on(PoolEvents.destroy, () => ++poolDestroy)
expect(pool.emitter.eventNames()).toStrictEqual([
PoolEvents.busy,
- PoolEvents.destroy
+ PoolEvents.destroy,
])
await pool.destroy()
const numberOfExitEvents = await exitPromise
expect(pool.started).toBe(false)
- expect(pool.emitter.eventNames()).toStrictEqual([])
+ expect(pool.emitter.eventNames()).toStrictEqual([
+ PoolEvents.busy,
+ PoolEvents.destroy,
+ ])
expect(pool.readyEventEmitted).toBe(false)
expect(pool.workerNodes.length).toBe(0)
expect(numberOfExitEvents).toBe(min)
it('Validation of inputs test', () => {
expect(() => new DynamicClusterPool(min)).toThrow(
- "Cannot find the worker file 'undefined'"
+ 'The worker file path must be specified'
)
})
const pool = new DynamicClusterPool(
min,
max,
- './tests/worker-files/cluster/testWorker.js'
+ './tests/worker-files/cluster/testWorker.cjs'
)
const result = await pool.execute()
expect(result).toStrictEqual({ ok: 1 })
const longRunningPool = new DynamicClusterPool(
min,
max,
- './tests/worker-files/cluster/longRunningWorkerHardBehavior.js',
+ './tests/worker-files/cluster/longRunningWorkerHardBehavior.cjs',
{
errorHandler: e => console.error(e),
onlineHandler: () => console.info('long executing worker is online'),
- exitHandler: () => console.info('long executing worker exited')
+ exitHandler: () => console.info('long executing worker exited'),
}
)
expect(longRunningPool.workerNodes.length).toBe(min)
await waitWorkerEvents(longRunningPool, 'exit', max - min)
expect(longRunningPool.workerNodes.length).toBe(min)
expect(
- longRunningPool.workerChoiceStrategyContext.workerChoiceStrategies.get(
- longRunningPool.workerChoiceStrategyContext.workerChoiceStrategy
+ longRunningPool.workerChoiceStrategiesContext.workerChoiceStrategies.get(
+ longRunningPool.workerChoiceStrategiesContext
+ .defaultWorkerChoiceStrategy
).nextWorkerNodeKey
).toBeLessThan(longRunningPool.workerNodes.length)
// We need to clean up the resources after our test
const longRunningPool = new DynamicClusterPool(
min,
max,
- './tests/worker-files/cluster/longRunningWorkerSoftBehavior.js',
+ './tests/worker-files/cluster/longRunningWorkerSoftBehavior.cjs',
{
errorHandler: e => console.error(e),
onlineHandler: () => console.info('long executing worker is online'),
- exitHandler: () => console.info('long executing worker exited')
+ exitHandler: () => console.info('long executing worker exited'),
}
)
expect(longRunningPool.workerNodes.length).toBe(min)
const pool = new DynamicClusterPool(
0,
max,
- './tests/worker-files/cluster/testWorker.js'
+ './tests/worker-files/cluster/testWorker.cjs'
)
expect(pool).toBeInstanceOf(DynamicClusterPool)
// We need to clean up the resources after our test
await pool.destroy()
})
+
+ 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++) {
+ // eslint-disable-next-line @typescript-eslint/no-unused-expressions
+ 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)
+ }
+ // We need to clean up the resources after our test
+ await pool.destroy()
+ }
+ })
})