1 const { expect
} = require('expect')
2 const sinon
= require('sinon')
7 } = require('../../../lib')
9 WorkerChoiceStrategyContext
10 } = require('../../../lib/pools/selection-strategies/worker-choice-strategy-context')
12 RoundRobinWorkerChoiceStrategy
13 } = require('../../../lib/pools/selection-strategies/round-robin-worker-choice-strategy')
15 LessUsedWorkerChoiceStrategy
16 } = require('../../../lib/pools/selection-strategies/less-used-worker-choice-strategy')
18 LessBusyWorkerChoiceStrategy
19 } = require('../../../lib/pools/selection-strategies/less-busy-worker-choice-strategy')
21 FairShareWorkerChoiceStrategy
22 } = require('../../../lib/pools/selection-strategies/fair-share-worker-choice-strategy')
24 WeightedRoundRobinWorkerChoiceStrategy
25 } = require('../../../lib/pools/selection-strategies/weighted-round-robin-worker-choice-strategy')
27 describe('Worker choice strategy context test suite', () => {
30 let fixedPool
, dynamicPool
33 fixedPool
= new FixedThreadPool(
35 './tests/worker-files/thread/testWorker.js'
37 dynamicPool
= new DynamicThreadPool(
40 './tests/worker-files/thread/testWorker.js'
49 await fixedPool
.destroy()
50 await dynamicPool
.destroy()
53 it('Verify that constructor() initializes the context with all the available worker choice strategies', () => {
54 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
57 expect(workerChoiceStrategyContext
.workerChoiceStrategies
.size
).toBe(
58 Object
.keys(WorkerChoiceStrategies
).length
62 it('Verify that execute() return the worker chosen by the strategy with fixed pool', () => {
63 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
66 const WorkerChoiceStrategyStub
= sinon
.createStubInstance(
67 RoundRobinWorkerChoiceStrategy
,
69 choose
: sinon
.stub().returns(0)
72 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
73 WorkerChoiceStrategies
.ROUND_ROBIN
75 workerChoiceStrategyContext
.workerChoiceStrategies
.set(
76 workerChoiceStrategyContext
.workerChoiceStrategy
,
77 WorkerChoiceStrategyStub
79 const chosenWorkerKey
= workerChoiceStrategyContext
.execute()
81 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
82 workerChoiceStrategyContext
.workerChoiceStrategy
85 expect(chosenWorkerKey
).toBe(0)
88 it('Verify that execute() return the worker chosen by the strategy with dynamic pool', () => {
89 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
92 const WorkerChoiceStrategyStub
= sinon
.createStubInstance(
93 RoundRobinWorkerChoiceStrategy
,
95 choose
: sinon
.stub().returns(0)
98 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
99 WorkerChoiceStrategies
.ROUND_ROBIN
101 workerChoiceStrategyContext
.workerChoiceStrategies
.set(
102 workerChoiceStrategyContext
.workerChoiceStrategy
,
103 WorkerChoiceStrategyStub
105 const chosenWorkerKey
= workerChoiceStrategyContext
.execute()
107 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
108 workerChoiceStrategyContext
.workerChoiceStrategy
111 expect(chosenWorkerKey
).toBe(0)
114 it('Verify that setWorkerChoiceStrategy() works with ROUND_ROBIN and fixed pool', () => {
115 const workerChoiceStrategy
= WorkerChoiceStrategies
.ROUND_ROBIN
116 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
120 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
123 ).toBeInstanceOf(RoundRobinWorkerChoiceStrategy
)
124 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
127 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
129 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
132 ).toBeInstanceOf(RoundRobinWorkerChoiceStrategy
)
133 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
138 it('Verify that setWorkerChoiceStrategy() works with ROUND_ROBIN and dynamic pool', () => {
139 const workerChoiceStrategy
= WorkerChoiceStrategies
.ROUND_ROBIN
140 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
144 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
147 ).toBeInstanceOf(RoundRobinWorkerChoiceStrategy
)
148 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
151 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
153 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
156 ).toBeInstanceOf(RoundRobinWorkerChoiceStrategy
)
157 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
162 it('Verify that setWorkerChoiceStrategy() works with LESS_USED and fixed pool', () => {
163 const workerChoiceStrategy
= WorkerChoiceStrategies
.LESS_USED
164 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
167 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
169 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
172 ).toBeInstanceOf(LessUsedWorkerChoiceStrategy
)
173 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
178 it('Verify that setWorkerChoiceStrategy() works with LESS_USED and dynamic pool', () => {
179 const workerChoiceStrategy
= WorkerChoiceStrategies
.LESS_USED
180 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
183 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
185 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
188 ).toBeInstanceOf(LessUsedWorkerChoiceStrategy
)
189 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
194 it('Verify that setWorkerChoiceStrategy() works with LESS_BUSY and fixed pool', () => {
195 const workerChoiceStrategy
= WorkerChoiceStrategies
.LESS_BUSY
196 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
199 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
201 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
204 ).toBeInstanceOf(LessBusyWorkerChoiceStrategy
)
205 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
210 it('Verify that setWorkerChoiceStrategy() works with LESS_BUSY and dynamic pool', () => {
211 const workerChoiceStrategy
= WorkerChoiceStrategies
.LESS_BUSY
212 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
215 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
217 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
220 ).toBeInstanceOf(LessBusyWorkerChoiceStrategy
)
221 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
226 it('Verify that setWorkerChoiceStrategy() works with FAIR_SHARE and fixed pool', () => {
227 const workerChoiceStrategy
= WorkerChoiceStrategies
.FAIR_SHARE
228 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
231 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
233 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
236 ).toBeInstanceOf(FairShareWorkerChoiceStrategy
)
237 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
242 it('Verify that setWorkerChoiceStrategy() works with FAIR_SHARE and dynamic pool', () => {
243 const workerChoiceStrategy
= WorkerChoiceStrategies
.FAIR_SHARE
244 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
247 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
249 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
252 ).toBeInstanceOf(FairShareWorkerChoiceStrategy
)
253 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
258 it('Verify that setWorkerChoiceStrategy() works with WEIGHTED_ROUND_ROBIN and fixed pool', () => {
259 const workerChoiceStrategy
= WorkerChoiceStrategies
.WEIGHTED_ROUND_ROBIN
260 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
263 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
265 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
268 ).toBeInstanceOf(WeightedRoundRobinWorkerChoiceStrategy
)
269 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
274 it('Verify that setWorkerChoiceStrategy() works with WEIGHTED_ROUND_ROBIN and dynamic pool', () => {
275 const workerChoiceStrategy
= WorkerChoiceStrategies
.WEIGHTED_ROUND_ROBIN
276 const workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
279 workerChoiceStrategyContext
.setWorkerChoiceStrategy(workerChoiceStrategy
)
281 workerChoiceStrategyContext
.workerChoiceStrategies
.get(
284 ).toBeInstanceOf(WeightedRoundRobinWorkerChoiceStrategy
)
285 expect(workerChoiceStrategyContext
.workerChoiceStrategy
).toBe(
290 it('Verify that worker choice strategy options enable median run time pool statistics', () => {
291 const wwrWorkerChoiceStrategy
= WorkerChoiceStrategies
.WEIGHTED_ROUND_ROBIN
292 let workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
294 wwrWorkerChoiceStrategy
,
299 expect(workerChoiceStrategyContext
.getRequiredStatistics().avgRunTime
).toBe(
302 expect(workerChoiceStrategyContext
.getRequiredStatistics().medRunTime
).toBe(
305 workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
307 wwrWorkerChoiceStrategy
,
312 expect(workerChoiceStrategyContext
.getRequiredStatistics().avgRunTime
).toBe(
315 expect(workerChoiceStrategyContext
.getRequiredStatistics().medRunTime
).toBe(
318 const fsWorkerChoiceStrategy
= WorkerChoiceStrategies
.FAIR_SHARE
319 workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
321 fsWorkerChoiceStrategy
,
326 expect(workerChoiceStrategyContext
.getRequiredStatistics().avgRunTime
).toBe(
329 expect(workerChoiceStrategyContext
.getRequiredStatistics().medRunTime
).toBe(
332 workerChoiceStrategyContext
= new WorkerChoiceStrategyContext(
334 fsWorkerChoiceStrategy
,
339 expect(workerChoiceStrategyContext
.getRequiredStatistics().avgRunTime
).toBe(
342 expect(workerChoiceStrategyContext
.getRequiredStatistics().medRunTime
).toBe(