From 125c84e40d6ee90f9b5e91f732a92e7b2503ebd8 Mon Sep 17 00:00:00 2001
From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= <jerome.benoit@piment-noir.org>
Date: Mon, 1 Jan 2024 11:17:18 +0100
Subject: [PATCH] test: improve worker choice strategies options tests
MIME-Version: 1.0
Content-Type: text/plain; charset=utf8
Content-Transfer-Encoding: 8bit

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---
 tests/utils.test.mjs | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/tests/utils.test.mjs b/tests/utils.test.mjs
index 5e089e29..5f54f6fc 100644
--- a/tests/utils.test.mjs
+++ b/tests/utils.test.mjs
@@ -244,6 +244,21 @@ describe('Utils test suite', () => {
       './tests/worker-files/thread/testWorker.mjs'
     )
     expect(getWorkerChoiceStrategyRetries(pool)).toBe(pool.info.maxSize * 2)
+    const workerChoiceStrategyOptions = {
+      runTime: { median: true },
+      waitTime: { median: true },
+      elu: { median: true },
+      weights: {
+        0: 100,
+        1: 100
+      }
+    }
+    expect(
+      getWorkerChoiceStrategyRetries(pool, workerChoiceStrategyOptions)
+    ).toBe(
+      pool.info.maxSize +
+        Object.keys(workerChoiceStrategyOptions.weights).length
+    )
     await pool.destroy()
   })
 
@@ -262,6 +277,18 @@ describe('Utils test suite', () => {
         [pool.info.maxSize - 1]: expect.any(Number)
       })
     })
+    const workerChoiceStrategyOptions = {
+      runTime: { median: true },
+      waitTime: { median: true },
+      elu: { median: true },
+      weights: {
+        0: 100,
+        1: 100
+      }
+    }
+    expect(
+      buildWorkerChoiceStrategyOptions(pool, workerChoiceStrategyOptions)
+    ).toStrictEqual(workerChoiceStrategyOptions)
     await pool.destroy()
   })
 
-- 
2.34.1