1 const expect
= require('expect')
2 const { DynamicClusterPool
} = require('../../../lib/index')
5 const pool
= new DynamicClusterPool(
8 './tests/worker/cluster/testWorker.js',
10 errorHandler
: e
=> console
.error(e
)
14 describe('Dynamic cluster pool test suite ', () => {
15 it('Verify that the function is executed in a worker cluster', async () => {
16 const result
= await pool
.execute({ test
: 'test' })
17 expect(result
).toBeDefined()
18 expect(result
).toBeFalsy()
21 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
).toBeLessThanOrEqual(max
)
30 expect(pool
.workers
.length
).toBeGreaterThan(min
)
31 pool
.workers
.forEach(w
=> {
36 expect(fullPool
> 1).toBeTruthy()
37 await
new Promise(resolve
=> setTimeout(resolve
, 5000))
38 expect(closedWorkers
).toBe(max
- min
)
41 it('Verify scale worker up and down is working', async () => {
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
).toBeGreaterThan(min
)
47 await
new Promise(resolve
=> setTimeout(resolve
, 3000))
48 expect(pool
.workers
.length
).toBe(min
)
49 for (let i
= 0; i
< max
* 10; i
++) {
50 pool
.execute({ test
: 'test' })
52 expect(pool
.workers
.length
).toBeGreaterThan(min
)
53 await
new Promise(resolve
=> setTimeout(resolve
, 3000))
54 expect(pool
.workers
.length
).toBe(min
)
56 it('Shutdown test', async () => {
58 pool
.workers
.forEach(w
=> {
64 await
new Promise(resolve
=> setTimeout(resolve
, 2000))
65 expect(closedWorkers
).toBe(min
)
68 it('Validations test', () => {
71 const pool1
= new DynamicClusterPool()
76 expect(error
).toBeTruthy()
77 expect(error
.message
).toBeTruthy()
80 it('Should work even without opts in input', async () => {
81 const pool1
= new DynamicClusterPool(
84 './tests/worker/cluster/testWorker.js'
86 const res
= await pool1
.execute({ test
: 'test' })
87 expect(res
).toBeFalsy()
90 it('Verify scale processes up and down is working when long running task is used', async () => {
91 const longRunningPool
= new DynamicClusterPool(
94 './tests/worker/cluster/longRunningWorker.js'
96 expect(longRunningPool
.workers
.length
).toBe(min
)
97 for (let i
= 0; i
< max
* 10; i
++) {
98 longRunningPool
.execute({ test
: 'test' })
100 expect(longRunningPool
.workers
.length
).toBe(max
)
101 await
new Promise(resolve
=> setTimeout(resolve
, 3000))
102 // Here we expect the workers to be at the max size since that the task is still running
103 expect(longRunningPool
.workers
.length
).toBe(max
)