X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fpools%2Fselection-strategies%2Fweighted-round-robin-worker-choice-strategy.test.js;h=9a27479bb58e33140429157adbd62b48c5a8d30c;hb=f6186241747440b21caff5102d5b1f3bdf27ad23;hp=664a447a1e829fc902842f0cc03966006d513540;hpb=0220f124adb20bbaebb927962b9346daa95a1d96;p=poolifier.git

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 664a447a..9a27479b 100644
--- 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
@@ -1,9 +1,10 @@
 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')
+const { generateRandomInteger } = require('../../test-utils')
 
 describe('Weighted round robin strategy worker choice strategy test suite', () => {
   // const min = 1
@@ -22,20 +23,16 @@ describe('Weighted round robin strategy worker choice strategy test suite', () =
     await pool.destroy()
   })
 
-  it.only('Verify that reset() resets internals', () => {
+  it('Verify that reset() resets internals', () => {
     const strategy = new WeightedRoundRobinWorkerChoiceStrategy(pool)
-    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(strategy.defaultWorkerWeight).toBeGreaterThan(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.workerVirtualTaskRunTime).toBe(0)
   })
 })