+ it('Verify WEIGHTED_ROUND_ROBIN strategy internals are resets after setting it', async () => {
+ let pool = new FixedThreadPool(
+ max,
+ './tests/worker-files/thread/testWorker.js'
+ )
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .currentWorkerIndex
+ ).toBeUndefined()
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .defaultWorkerWeight
+ ).toBeUndefined()
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workersTaskRunTime
+ ).toBeUndefined()
+ pool.setWorkerChoiceStrategy(WorkerChoiceStrategies.WEIGHTED_ROUND_ROBIN)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .currentWorkerIndex
+ ).toBe(0)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .defaultWorkerWeight
+ ).toBeGreaterThan(0)
+ for (const worker of pool.workerChoiceStrategyContext
+ .getWorkerChoiceStrategy()
+ .workersTaskRunTime.keys()) {
+ expect(
+ pool.workerChoiceStrategyContext
+ .getWorkerChoiceStrategy()
+ .workersTaskRunTime.get(worker).runTime
+ ).toBe(0)
+ }
+ await pool.destroy()
+ pool = new DynamicThreadPool(
+ min,
+ max,
+ './tests/worker-files/thread/testWorker.js'
+ )
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.currentWorkerIndex
+ ).toBeUndefined()
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.defaultWorkerWeight
+ ).toBeUndefined()
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.workersTaskRunTime
+ ).toBeUndefined()
+ pool.setWorkerChoiceStrategy(WorkerChoiceStrategies.WEIGHTED_ROUND_ROBIN)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.currentWorkerIndex
+ ).toBe(0)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.defaultWorkerWeight
+ ).toBeGreaterThan(0)
+ for (const worker of pool.workerChoiceStrategyContext
+ .getWorkerChoiceStrategy()
+ .workerChoiceStrategy.workersTaskRunTime.keys()) {
+ expect(
+ pool.workerChoiceStrategyContext
+ .getWorkerChoiceStrategy()
+ .workerChoiceStrategy.workersTaskRunTime.get(worker).runTime
+ ).toBe(0)
+ }
+ // We need to clean up the resources after our test
+ await pool.destroy()
+ })
+