2a0a8e2e70f02e15839d89d972d433254ab744e6
1 const { expect
} = require('expect')
2 const { FixedClusterPool
} = require('../../../lib/index')
3 const TestUtils
= require('../../test-utils')
4 const numberOfWorkers
= 10
5 const pool
= new FixedClusterPool(
7 './tests/worker-files/cluster/testWorker.js',
9 errorHandler
: e
=> console
.error(e
)
12 const emptyPool
= new FixedClusterPool(
14 './tests/worker-files/cluster/emptyWorker.js',
15 { exitHandler
: () => console
.log('empty pool worker exited') }
17 const echoPool
= new FixedClusterPool(
19 './tests/worker-files/cluster/echoWorker.js'
21 const errorPool
= new FixedClusterPool(
23 './tests/worker-files/cluster/errorWorker.js',
25 errorHandler
: e
=> console
.error(e
)
28 const asyncErrorPool
= new FixedClusterPool(
30 './tests/worker-files/cluster/asyncErrorWorker.js',
32 errorHandler
: e
=> console
.error(e
)
35 const asyncPool
= new FixedClusterPool(
37 './tests/worker-files/cluster/asyncWorker.js'
40 describe('Fixed cluster pool test suite', () => {
41 after('Destroy all pools', async () => {
42 // We need to clean up the resources after our test
43 await echoPool
.destroy()
44 await asyncPool
.destroy()
45 await errorPool
.destroy()
46 await asyncErrorPool
.destroy()
47 await emptyPool
.destroy()
50 it('Choose worker round robin test', async () => {
51 const results
= new Set()
52 for (let i
= 0; i
< numberOfWorkers
; i
++) {
53 results
.add(pool
.chooseWorker().id
)
55 expect(results
.size
).toBe(numberOfWorkers
)
58 it('Verify that the function is executed in a worker cluster', async () => {
59 const result
= await pool
.execute({ test
: 'test' })
60 expect(result
).toBeDefined()
61 expect(result
).toBeFalsy()
64 it('Verify that is possible to invoke the execute method without input', async () => {
65 const result
= await pool
.execute()
66 expect(result
).toBeDefined()
67 expect(result
).toBeFalsy()
70 it('Verify that busy event is emitted', async () => {
73 pool
.emitter
.on('busy', () => poolBusy
++)
74 for (let i
= 0; i
< numberOfWorkers
* 2; i
++) {
75 promises
.push(pool
.execute({ test
: 'test' }))
77 // The `busy` event is triggered when the number of submitted tasks at once reach the number of fixed pool workers.
78 // So in total numberOfWorkers + 1 times for a loop submitting up to numberOfWorkers * 2 tasks to the fixed pool.
79 expect(poolBusy
).toBe(numberOfWorkers
+ 1)
82 it('Verify that is possible to have a worker that return undefined', async () => {
83 const result
= await emptyPool
.execute()
84 expect(result
).toBeFalsy()
87 it('Verify that data are sent to the worker correctly', async () => {
88 const data
= { f
: 10 }
89 const result
= await echoPool
.execute(data
)
90 expect(result
).toBeTruthy()
91 expect(result
.f
).toBe(data
.f
)
94 it('Verify that error handling is working properly:sync', async () => {
95 const data
= { f
: 10 }
98 await errorPool
.execute(data
)
102 expect(inError
).toBeDefined()
103 expect(typeof inError
=== 'string').toBe(true)
104 expect(inError
).toBe('Error Message from ClusterWorker')
107 it('Verify that error handling is working properly:async', async () => {
108 const data
= { f
: 10 }
111 await asyncErrorPool
.execute(data
)
115 expect(inError
).toBeDefined()
116 expect(typeof inError
=== 'string').toBe(true)
117 expect(inError
).toBe('Error Message from ClusterWorker:async')
120 it('Verify that async function is working properly', async () => {
121 const data
= { f
: 10 }
122 const startTime
= new Date().getTime()
123 const result
= await asyncPool
.execute(data
)
124 const usedTime
= new Date().getTime() - startTime
125 expect(result
).toBeTruthy()
126 expect(result
.f
).toBe(data
.f
)
127 expect(usedTime
).toBeGreaterThanOrEqual(2000)
130 it('Shutdown test', async () => {
131 const exitPromise
= TestUtils
.waitExits(pool
, numberOfWorkers
)
133 const res
= await exitPromise
134 expect(res
).toBe(numberOfWorkers
)
137 it('Should work even without opts in input', async () => {
138 const pool1
= new FixedClusterPool(
140 './tests/worker-files/cluster/testWorker.js'
142 const res
= await pool1
.execute({ test
: 'test' })
143 expect(res
).toBeFalsy()
144 // We need to clean up the resources after our test
145 await pool1
.destroy()
148 it('Verify that a pool with zero worker fails', async () => {
151 new FixedClusterPool(0, './tests/worker-files/cluster/testWorker.js')
152 ).toThrowError(new Error('Cannot instantiate a fixed pool with no worker'))