66aeaa3a20a9ebf870c67a2c8d99d5c46181f338
1 const expect
= require('expect')
2 const { DynamicThreadPool
} = require('../../../lib/index')
3 const TestUtils
= require('../../test-utils')
6 const pool
= new DynamicThreadPool(
9 './tests/worker-files/thread/testWorker.js',
11 errorHandler
: e
=> console
.error(e
)
15 describe('Dynamic thread pool test suite ', () => {
16 it('Verify that the function is executed in a worker thread', async () => {
17 const result
= await pool
.execute({ test
: 'test' })
18 expect(result
).toBeDefined()
19 expect(result
).toBeFalsy()
22 it('Verify that new workers are created when required, max size is not exceeded and that after a while new workers will die', async () => {
25 pool
.emitter
.on('FullPool', () => fullPool
++)
26 for (let i
= 0; i
< max
* 2; i
++) {
27 promises
.push(pool
.execute({ test
: 'test' }))
29 expect(pool
.workers
.length
).toBe(max
)
30 expect(fullPool
> 1).toBeTruthy()
31 const res
= await TestUtils
.waitExits(pool
, max
- min
)
32 expect(res
).toBe(max
- min
)
35 it('Verify scale thread up and down is working', async () => {
36 expect(pool
.workers
.length
).toBe(min
)
37 for (let i
= 0; i
< max
* 10; i
++) {
38 pool
.execute({ test
: 'test' })
40 expect(pool
.workers
.length
).toBe(max
)
41 await TestUtils
.waitExits(pool
, max
- min
)
42 expect(pool
.workers
.length
).toBe(min
)
43 for (let i
= 0; i
< max
* 10; i
++) {
44 pool
.execute({ test
: 'test' })
46 expect(pool
.workers
.length
).toBe(max
)
47 await TestUtils
.waitExits(pool
, max
- min
)
48 expect(pool
.workers
.length
).toBe(min
)
51 it('Shutdown test', async () => {
53 pool
.workers
.forEach(w
=> {
59 expect(closedThreads
).toBe(min
)
62 it('Validations test', () => {
64 const pool1
= new DynamicThreadPool()
66 new Error('Please specify a file with a worker implementation')
70 it('Should work even without opts in input', async () => {
71 const pool1
= new DynamicThreadPool(
74 './tests/worker-files/thread/testWorker.js'
76 const res
= await pool1
.execute({ test
: 'test' })
77 expect(res
).toBeFalsy()
80 it('Verify scale thread up and down is working when long running task is used:hard', async () => {
81 const longRunningPool
= new DynamicThreadPool(
84 './tests/worker-files/thread/longRunningWorkerHardBehavior.js',
86 errorHandler
: e
=> console
.error(e
),
87 onlineHandler
: () => console
.log('worker is online')
90 expect(longRunningPool
.workers
.length
).toBe(min
)
91 for (let i
= 0; i
< max
* 10; i
++) {
92 longRunningPool
.execute({ test
: 'test' })
94 expect(longRunningPool
.workers
.length
).toBe(max
)
95 await TestUtils
.waitExits(longRunningPool
, max
- min
)
96 expect(longRunningPool
.workers
.length
).toBe(min
)
99 it('Verify scale thread up and down is working when long running task is used:soft', async () => {
100 const longRunningPool
= new DynamicThreadPool(
103 './tests/worker-files/thread/longRunningWorkerSoftBehavior.js',
105 errorHandler
: e
=> console
.error(e
),
106 onlineHandler
: () => console
.log('worker is online')
109 expect(longRunningPool
.workers
.length
).toBe(min
)
110 for (let i
= 0; i
< max
* 10; i
++) {
111 longRunningPool
.execute({ test
: 'test' })
113 expect(longRunningPool
.workers
.length
).toBe(max
)
114 await TestUtils
.sleep(1500)
115 // Here we expect the workers to be at the max size since that the task is still running
116 expect(longRunningPool
.workers
.length
).toBe(max
)