test: fix sleep() test, take 2

[poolifier.git] / tests / test-utils.js

diff --git a/tests/test-utils.js b/tests/test-utils.js
index ba5e652d0abeb8ccb2f5d39af7336a54ed02c667..3234a6a22cb3af039b81494520178a996a95d9f3 100644 (file)
--- a/tests/test-utils.js
+++ b/tests/test-utils.js
@@ -1,7 +1,7 @@
-const { WorkerFunctions } = require('./test-types')
+const { TaskFunctions } = require('./test-types')
 
 const waitWorkerEvents = async (pool, workerEvent, numberOfEventsToWait) => {
-  return new Promise(resolve => {
+  return new Promise((resolve) => {
     let events = 0
     if (numberOfEventsToWait === 0) {
       resolve(events)
@@ -18,12 +18,12 @@ const waitWorkerEvents = async (pool, workerEvent, numberOfEventsToWait) => {
 }
 
 const waitPoolEvents = async (pool, poolEvent, numberOfEventsToWait) => {
-  return new Promise(resolve => {
+  return new Promise((resolve) => {
     let events = 0
     if (numberOfEventsToWait === 0) {
       resolve(events)
     }
-    pool.emitter.on(poolEvent, () => {
+    pool.emitter?.on(poolEvent, () => {
       ++events
       if (events === numberOfEventsToWait) {
         resolve(events)
@@ -32,11 +32,11 @@ const waitPoolEvents = async (pool, poolEvent, numberOfEventsToWait) => {
   })
 }
 
-const sleep = async ms => {
-  return new Promise(resolve => setTimeout(resolve, ms))
+const sleep = async (ms) => {
+  return new Promise((resolve) => setTimeout(resolve, ms))
 }
 
-const sleepWorkerFunction = async (
+const sleepTaskFunction = async (
   data,
   ms,
   rejection = false,
@@ -65,13 +65,14 @@ const generateRandomInteger = (max = Number.MAX_SAFE_INTEGER, min = 0) => {
   return Math.floor(Math.random() * (max + 1))
 }
 
-const jsonIntegerSerialization = n => {
+const jsonIntegerSerialization = (n) => {
   for (let i = 0; i < n; i++) {
     const o = {
       a: i
     }
     JSON.stringify(o)
   }
+  return { ok: 1 }
 }
 
 /**
@@ -79,7 +80,7 @@ const jsonIntegerSerialization = n => {
  * @param {number} n - The number of fibonacci numbers to generate.
  * @returns {number} - The nth fibonacci number.
  */
-const fibonacci = n => {
+const fibonacci = (n) => {
   if (n <= 1) return n
   return fibonacci(n - 1) + fibonacci(n - 2)
 }
@@ -89,20 +90,20 @@ const fibonacci = n => {
  * @param {number} n - The number to calculate the factorial of.
  * @returns {number} - The factorial of n.
  */
-const factorial = n => {
+const factorial = (n) => {
   if (n === 0) {
     return 1
   }
   return factorial(n - 1) * n
 }
 
-const executeWorkerFunction = data => {
+const executeTaskFunction = (data) => {
   switch (data.function) {
-    case WorkerFunctions.jsonIntegerSerialization:
+    case TaskFunctions.jsonIntegerSerialization:
       return jsonIntegerSerialization(data.n || 100)
-    case WorkerFunctions.fibonacci:
+    case TaskFunctions.fibonacci:
       return fibonacci(data.n || 25)
-    case WorkerFunctions.factorial:
+    case TaskFunctions.factorial:
       return factorial(data.n || 100)
     default:
       throw new Error('Unknown worker function')
@@ -110,13 +111,13 @@ const executeWorkerFunction = data => {
 }
 
 module.exports = {
-  executeWorkerFunction,
+  executeTaskFunction,
   factorial,
   fibonacci,
   generateRandomInteger,
   jsonIntegerSerialization,
   sleep,
-  sleepWorkerFunction,
+  sleepTaskFunction,
   waitPoolEvents,
   waitWorkerEvents
 }