@types/node update to 18.8.5
[poolifier.git] / tests / pools / selection-strategies / weighted-round-robin-worker-choice-strategy.test.js
CommitLineData
0220f124
JB
1const { expect } = require('expect')
2const sinon = require('sinon')
3const { FixedThreadPool } = require('../../../lib/index')
4const {
5 WeightedRoundRobinWorkerChoiceStrategy
6} = require('../../../lib/pools/selection-strategies/weighted-round-robin-worker-choice-strategy')
f931db5c 7const TestUtils = require('../../test-utils')
0220f124
JB
8
9describe('Weighted round robin strategy worker choice strategy test suite', () => {
10 // const min = 1
11 const max = 3
12 let pool
13
14 before(() => {
15 pool = new FixedThreadPool(max, './tests/worker-files/thread/testWorker.js')
16 })
17
18 afterEach(() => {
19 sinon.restore()
20 })
21
22 after(async () => {
23 await pool.destroy()
24 })
25
6cdd998c 26 it('Verify that reset() resets internals', () => {
0220f124 27 const strategy = new WeightedRoundRobinWorkerChoiceStrategy(pool)
f931db5c
JB
28 strategy.previousWorkerIndex = TestUtils.generateRandomInteger(
29 Number.MAX_SAFE_INTEGER
30 )
31 strategy.currentWorkerIndex = TestUtils.generateRandomInteger(
32 Number.MAX_SAFE_INTEGER
33 )
0220f124
JB
34 const workersTaskRunTimeClearStub = sinon
35 .stub(strategy.workersTaskRunTime, 'clear')
36 .returns()
37 const initWorkersTaskRunTimeStub = sinon
38 .stub(strategy, 'initWorkersTaskRunTime')
39 .returns()
40 const resetResult = strategy.reset()
41 expect(resetResult).toBe(true)
42 expect(strategy.previousWorkerIndex).toBe(0)
43 expect(strategy.currentWorkerIndex).toBe(0)
0220f124
JB
44 expect(workersTaskRunTimeClearStub.calledOnce).toBe(true)
45 expect(initWorkersTaskRunTimeStub.calledOnce).toBe(true)
46 })
47})