1 import { expect } from 'expect'
2 import { DynamicThreadPool, PoolEvents } from '../../../lib/index.js'
3 import { TaskFunctions } from '../../test-types.js'
4 import { sleep, waitWorkerEvents } from '../../test-utils.js'
6 describe('Dynamic thread pool test suite', () => {
9 const pool = new DynamicThreadPool(
12 './tests/worker-files/thread/testWorker.mjs',
14 errorHandler: e => console.error(e)
18 it('Verify that the function is executed in a worker thread', async () => {
19 let result = await pool.execute({
20 function: TaskFunctions.fibonacci
22 expect(result).toBe(75025)
23 result = await pool.execute({
24 function: TaskFunctions.factorial
26 expect(result).toBe(9.33262154439441e157)
29 it('Verify that new workers are created when required, max size is not exceeded and that after a while new workers will die', async () => {
31 pool.emitter.on(PoolEvents.busy, () => ++poolBusy)
32 for (let i = 0; i < max * 2; i++) {
35 expect(pool.workerNodes.length).toBeLessThanOrEqual(max)
36 expect(pool.workerNodes.length).toBeGreaterThan(min)
37 expect(poolBusy).toBe(1)
38 const numberOfExitEvents = await waitWorkerEvents(pool, 'exit', max - min)
39 expect(numberOfExitEvents).toBe(max - min)
42 it('Verify scale thread up and down is working', async () => {
43 expect(pool.workerNodes.length).toBe(min)
44 for (let i = 0; i < max * 2; i++) {
47 expect(pool.workerNodes.length).toBe(max)
48 await waitWorkerEvents(pool, 'exit', max - min)
49 expect(pool.workerNodes.length).toBe(min)
50 for (let i = 0; i < max * 2; i++) {
53 expect(pool.workerNodes.length).toBe(max)
54 await waitWorkerEvents(pool, 'exit', max - min)
55 expect(pool.workerNodes.length).toBe(min)
58 it('Shutdown test', async () => {
59 const exitPromise = waitWorkerEvents(pool, 'exit', min)
60 expect(pool.emitter.eventNames()).toStrictEqual([PoolEvents.busy])
62 pool.emitter.on(PoolEvents.destroy, () => ++poolDestroy)
63 expect(pool.emitter.eventNames()).toStrictEqual([
68 const numberOfExitEvents = await exitPromise
69 expect(pool.started).toBe(false)
70 expect(pool.readyEventEmitted).toBe(false)
71 expect(pool.workerNodes.length).toBe(0)
72 expect(numberOfExitEvents).toBe(min)
73 expect(poolDestroy).toBe(1)
76 it('Validation of inputs test', () => {
77 expect(() => new DynamicThreadPool(min)).toThrow(
78 "Cannot find the worker file 'undefined'"
82 it('Should work even without opts in input', async () => {
83 const pool = new DynamicThreadPool(
86 './tests/worker-files/thread/testWorker.mjs'
88 const res = await pool.execute()
89 expect(res).toStrictEqual({ ok: 1 })
90 // We need to clean up the resources after our test
94 it('Verify scale thread up and down is working when long executing task is used:hard', async () => {
95 const longRunningPool = new DynamicThreadPool(
98 './tests/worker-files/thread/longRunningWorkerHardBehavior.mjs',
100 errorHandler: e => console.error(e),
101 onlineHandler: () => console.info('long executing worker is online'),
102 exitHandler: () => console.info('long executing worker exited')
105 expect(longRunningPool.workerNodes.length).toBe(min)
106 for (let i = 0; i < max * 2; i++) {
107 longRunningPool.execute()
109 expect(longRunningPool.workerNodes.length).toBe(max)
110 await waitWorkerEvents(longRunningPool, 'exit', max - min)
111 expect(longRunningPool.workerNodes.length).toBe(min)
113 longRunningPool.workerChoiceStrategyContext.workerChoiceStrategies.get(
114 longRunningPool.workerChoiceStrategyContext.workerChoiceStrategy
116 ).toBeLessThan(longRunningPool.workerNodes.length)
117 // We need to clean up the resources after our test
118 await longRunningPool.destroy()
121 it('Verify scale thread up and down is working when long executing task is used:soft', async () => {
122 const longRunningPool = new DynamicThreadPool(
125 './tests/worker-files/thread/longRunningWorkerSoftBehavior.mjs',
127 errorHandler: e => console.error(e),
128 onlineHandler: () => console.info('long executing worker is online'),
129 exitHandler: () => console.info('long executing worker exited')
132 expect(longRunningPool.workerNodes.length).toBe(min)
133 for (let i = 0; i < max * 2; i++) {
134 longRunningPool.execute()
136 expect(longRunningPool.workerNodes.length).toBe(max)
138 // Here we expect the workerNodes to be at the max size since the task is still executing
139 expect(longRunningPool.workerNodes.length).toBe(max)
140 // We need to clean up the resources after our test
141 await longRunningPool.destroy()
144 it('Verify that a pool with zero worker can be instantiated', async () => {
145 const pool = new DynamicThreadPool(
148 './tests/worker-files/thread/testWorker.mjs'
150 expect(pool).toBeInstanceOf(DynamicThreadPool)
151 // We need to clean up the resources after our test