const {
WorkerChoiceStrategies,
DynamicThreadPool,
- FixedThreadPool
+ FixedThreadPool,
+ FixedClusterPool
} = require('../../../lib/index')
describe('Selection strategies test suite', () => {
WorkerChoiceStrategies.ROUND_ROBIN
)
expect(
- pool.workerChoiceStrategyContext.getWorkerChoiceStrategy().nextWorkerIndex
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy().nextWorkerId
).toBe(0)
// We need to clean up the resources after our test
await pool.destroy()
await pool.destroy()
})
+ it('Verify ROUND_ROBIN strategy runtime behavior', async () => {
+ let pool = new FixedClusterPool(
+ max,
+ './tests/worker-files/cluster/testWorker.js'
+ )
+ let results = new Set()
+ for (let i = 0; i < max; i++) {
+ results.add(pool.chooseWorker().id)
+ }
+ expect(results.size).toBe(max)
+ await pool.destroy()
+ pool = new FixedThreadPool(max, './tests/worker-files/thread/testWorker.js')
+ results = new Set()
+ for (let i = 0; i < max; i++) {
+ results.add(pool.chooseWorker().threadId)
+ }
+ expect(results.size).toBe(max)
+ await pool.destroy()
+ })
+
+ it('Verify ROUND_ROBIN strategy internals are resets after setting it', async () => {
+ let pool = new FixedThreadPool(
+ max,
+ './tests/worker-files/thread/testWorker.js',
+ { workerChoiceStrategy: WorkerChoiceStrategies.WEIGHTED_ROUND_ROBIN }
+ )
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy().nextWorkerId
+ ).toBeUndefined()
+ pool.setWorkerChoiceStrategy(WorkerChoiceStrategies.ROUND_ROBIN)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy().nextWorkerId
+ ).toBe(0)
+ await pool.destroy()
+ pool = new DynamicThreadPool(
+ min,
+ max,
+ './tests/worker-files/thread/testWorker.js',
+ { workerChoiceStrategy: WorkerChoiceStrategies.WEIGHTED_ROUND_ROBIN }
+ )
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.nextWorkerId
+ ).toBeUndefined()
+ pool.setWorkerChoiceStrategy(WorkerChoiceStrategies.ROUND_ROBIN)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.nextWorkerId
+ ).toBe(0)
+ // We need to clean up the resources after our test
+ await pool.destroy()
+ })
+
it('Verify LESS_RECENTLY_USED strategy is taken at pool creation', async () => {
const pool = new FixedThreadPool(
max,
await pool.destroy()
})
- it('Verify FAIR_SHARE strategy statistics are resets after setting it', async () => {
+ it('Verify FAIR_SHARE strategy internals are resets after setting it', async () => {
let pool = new FixedThreadPool(
max,
'./tests/worker-files/thread/testWorker.js'
WorkerChoiceStrategies.WEIGHTED_ROUND_ROBIN
)
expect(
- pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
- .previousWorkerIndex
- ).toBe(0)
- expect(
- pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
- .currentWorkerIndex
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy().currentWorkerId
).toBe(0)
expect(
pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
await pool.destroy()
})
- it('Verify WEIGHTED_ROUND_ROBIN strategy statistics are resets after setting it', async () => {
+ 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().currentWorkerId
+ ).toBeUndefined()
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .defaultWorkerWeight
+ ).toBeUndefined()
expect(
pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
.workersTaskRunTime
).toBeUndefined()
pool.setWorkerChoiceStrategy(WorkerChoiceStrategies.WEIGHTED_ROUND_ROBIN)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy().currentWorkerId
+ ).toBe(0)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .defaultWorkerWeight
+ ).toBeGreaterThan(0)
for (const worker of pool.workerChoiceStrategyContext
.getWorkerChoiceStrategy()
.workersTaskRunTime.keys()) {
max,
'./tests/worker-files/thread/testWorker.js'
)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.currentWorkerId
+ ).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.currentWorkerId
+ ).toBe(0)
+ expect(
+ pool.workerChoiceStrategyContext.getWorkerChoiceStrategy()
+ .workerChoiceStrategy.defaultWorkerWeight
+ ).toBeGreaterThan(0)
for (const worker of pool.workerChoiceStrategyContext
.getWorkerChoiceStrategy()
.workerChoiceStrategy.workersTaskRunTime.keys()) {