fix: test for worker file existence

[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 b21c41ff954f72aa3c1dee9d9381c985b2413e01..e9de670a8a67edcad5dd72118ab4c80de9af686d 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,7 +4,7 @@ const { FixedThreadPool } = require('../../../lib')
 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
@@ -25,17 +25,14 @@ 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(
-      Number.MAX_SAFE_INTEGER,
-      1
-    )
-    strategy.workerVirtualTaskRunTime = TestUtils.generateRandomInteger(
+    strategy.currentWorkerId = generateRandomInteger(Number.MAX_SAFE_INTEGER, 1)
+    strategy.workerVirtualTaskRunTime = generateRandomInteger(
       Number.MAX_SAFE_INTEGER,
       1
     )
     const resetResult = strategy.reset()
     expect(resetResult).toBe(true)
-    expect(strategy.nextWorkerNodeId).toBe(0)
+    expect(strategy.nextWorkerNodeKey).toBe(0)
     expect(strategy.workerVirtualTaskRunTime).toBe(0)
   })
 })