1 import cluster, { Worker as ClusterWorker } from 'node:cluster'
2 import { Worker as ThreadWorker } from 'node:worker_threads'
4 import { expect } from 'expect'
6 import { CircularBuffer } from '../../lib/circular-buffer.cjs'
7 import { WorkerTypes } from '../../lib/index.cjs'
10 DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS,
11 getDefaultTasksQueueOptions,
14 updateMeasurementStatistics
15 } from '../../lib/pools/utils.cjs'
16 import { MeasurementHistorySize } from '../../lib/pools/worker.cjs'
18 describe('Pool utils test suite', () => {
19 it('Verify DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS values', () => {
20 expect(DEFAULT_MEASUREMENT_STATISTICS_REQUIREMENTS).toStrictEqual({
27 it('Verify getDefaultTasksQueueOptions() behavior', () => {
29 expect(getDefaultTasksQueueOptions(poolMaxSize)).toStrictEqual({
31 size: Math.pow(poolMaxSize, 2),
33 tasksStealingOnBackPressure: false,
34 tasksFinishedTimeout: 2000
38 it('Verify updateMeasurementStatistics() behavior', () => {
39 // const circularBuffer = new CircularBuffer(MeasurementHistorySize)
40 const measurementStatistics = {
41 history: new CircularBuffer(MeasurementHistorySize)
43 updateMeasurementStatistics(
44 measurementStatistics,
45 { aggregate: true, average: false, median: false },
48 expect(measurementStatistics).toMatchObject({
53 updateMeasurementStatistics(
54 measurementStatistics,
55 { aggregate: true, average: false, median: false },
58 expect(measurementStatistics).toMatchObject({
63 updateMeasurementStatistics(
64 measurementStatistics,
65 { aggregate: true, average: true, median: false },
68 expect(measurementStatistics).toMatchObject({
72 average: 0.0010000000474974513
74 updateMeasurementStatistics(
75 measurementStatistics,
76 { aggregate: true, average: true, median: false },
79 expect(measurementStatistics).toMatchObject({
83 average: 0.0020000000367872417
85 updateMeasurementStatistics(
86 measurementStatistics,
87 { aggregate: true, average: false, median: true },
90 expect(measurementStatistics).toMatchObject({
94 median: 0.003000000026077032
96 updateMeasurementStatistics(
97 measurementStatistics,
98 { aggregate: true, average: true, median: false },
101 expect(measurementStatistics).toMatchObject({
105 average: 0.004999999975552782
109 it('Verify createWorker() behavior', () => {
113 './tests/worker-files/thread/testWorker.mjs',
116 ).toBeInstanceOf(ThreadWorker)
120 './tests/worker-files/cluster/testWorker.mjs',
123 ).toBeInstanceOf(ClusterWorker)
126 it('Verify getWorkerType() behavior', () => {
129 new ThreadWorker('./tests/worker-files/thread/testWorker.mjs')
131 ).toBe(WorkerTypes.thread)
132 expect(getWorkerType(cluster.fork())).toBe(WorkerTypes.cluster)
135 it('Verify getWorkerId() behavior', () => {
136 const threadWorker = new ThreadWorker(
137 './tests/worker-files/thread/testWorker.mjs'
139 const clusterWorker = cluster.fork()
140 expect(getWorkerId(threadWorker)).toBe(threadWorker.threadId)
141 expect(getWorkerId(clusterWorker)).toBe(clusterWorker.id)