const {
WeightedRoundRobinWorkerChoiceStrategy
} = require('../../../lib/pools/selection-strategies/weighted-round-robin-worker-choice-strategy')
-const TestUtils = require('../../test-utils')
+const { generateRandomInteger } = require('../../test-utils')
describe('Weighted round robin strategy worker choice strategy test suite', () => {
// const min = 1
it('Verify that reset() resets internals', () => {
const strategy = new WeightedRoundRobinWorkerChoiceStrategy(pool)
- strategy.currentWorkerId = TestUtils.generateRandomInteger()
- const workersTaskRunTimeClearStub = sinon
- .stub(strategy.workersTaskRunTime, 'clear')
- .returns()
- const initWorkersTaskRunTimeStub = sinon
- .stub(strategy, 'initWorkersTaskRunTime')
- .returns()
- const resetResult = strategy.reset()
- expect(resetResult).toBe(true)
- expect(strategy.currentWorkerNodeId).toBe(0)
- expect(workersTaskRunTimeClearStub.calledOnce).toBe(true)
- expect(initWorkersTaskRunTimeStub.calledOnce).toBe(true)
+ strategy.currentWorkerId = generateRandomInteger(Number.MAX_SAFE_INTEGER, 1)
+ strategy.workerVirtualTaskRunTime = generateRandomInteger(
+ Number.MAX_SAFE_INTEGER,
+ 1
+ )
+ expect(strategy.reset()).toBe(true)
+ expect(strategy.nextWorkerNodeKey).toBe(0)
+ expect(strategy.previousWorkerNodeKey).toBe(0)
+ expect(strategy.workerNodeVirtualTaskRunTime).toBe(0)
})
})