const { expect } = require('expect')
const sinon = require('sinon')
-const { FixedThreadPool } = require('../../../lib/index')
+const { FixedThreadPool } = require('../../../lib')
const {
WeightedRoundRobinWorkerChoiceStrategy
} = require('../../../lib/pools/selection-strategies/weighted-round-robin-worker-choice-strategy')
it('Verify that reset() resets internals', () => {
const strategy = new WeightedRoundRobinWorkerChoiceStrategy(pool)
- strategy.previousWorkerIndex = TestUtils.generateRandomInteger(
- Number.MAX_SAFE_INTEGER
+ strategy.currentWorkerId = TestUtils.generateRandomInteger(
+ Number.MAX_SAFE_INTEGER,
+ 1
)
- strategy.currentWorkerIndex = TestUtils.generateRandomInteger(
- Number.MAX_SAFE_INTEGER
+ strategy.workerVirtualTaskRunTime = TestUtils.generateRandomInteger(
+ Number.MAX_SAFE_INTEGER,
+ 1
)
- 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.previousWorkerIndex).toBe(0)
- expect(strategy.currentWorkerIndex).toBe(0)
- expect(workersTaskRunTimeClearStub.calledOnce).toBe(true)
- expect(initWorkersTaskRunTimeStub.calledOnce).toBe(true)
+ expect(strategy.currentWorkerNodeId).toBe(0)
+ expect(strategy.workerVirtualTaskRunTime).toBe(0)
})
})