perf: use a single map to store pool workers and their related data

[poolifier.git] / tests / pools / selection-strategies / weighted-round-robin-worker-choice-strategy.test.js

diff --git a/tests/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.test.js b/tests/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.test.js
index 24246595e6bdc92bcd40afad6638c31ef3842e83..cc3e66a49e752b349e4ae8153facbddfadca81ac 100644 (file)
--- a/tests/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.test.js
+++ b/tests/pools/selection-strategies/weighted-round-robin-worker-choice-strategy.test.js
@@ -4,6 +4,7 @@ const { FixedThreadPool } = require('../../../lib/index')
 const {
   WeightedRoundRobinWorkerChoiceStrategy
 } = require('../../../lib/pools/selection-strategies/weighted-round-robin-worker-choice-strategy')
+const TestUtils = require('../../test-utils')
 
 describe('Weighted round robin strategy worker choice strategy test suite', () => {
   // const min = 1
@@ -24,6 +25,7 @@ describe('Weighted round robin strategy worker choice strategy test suite', () =
 
   it('Verify that reset() resets internals', () => {
     const strategy = new WeightedRoundRobinWorkerChoiceStrategy(pool)
+    strategy.currentWorkerId = TestUtils.generateRandomInteger()
     const workersTaskRunTimeClearStub = sinon
       .stub(strategy.workersTaskRunTime, 'clear')
       .returns()
@@ -32,9 +34,7 @@ describe('Weighted round robin strategy worker choice strategy test suite', () =
       .returns()
     const resetResult = strategy.reset()
     expect(resetResult).toBe(true)
-    expect(strategy.previousWorkerIndex).toBe(0)
-    expect(strategy.currentWorkerIndex).toBe(0)
-    expect(strategy.defaultWorkerWeight).toBeGreaterThan(0)
+    expect(strategy.currentWorkerId).toBe(0)
     expect(workersTaskRunTimeClearStub.calledOnce).toBe(true)
     expect(initWorkersTaskRunTimeStub.calledOnce).toBe(true)
   })