X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fworker%2Fthread-worker.test.js;h=9eeaaa0a70d44eeced4d4ecc4fa88f885b426156;hb=52b71763f1a1240170e777049a80df0ad14ce496;hp=c04d3b2f0e980f7fa782a77f56046faae901eafc;hpb=e1ffb94fcac39a7459c47edaa086a7729a1dec44;p=poolifier.git

diff --git a/tests/worker/thread-worker.test.js b/tests/worker/thread-worker.test.js
index c04d3b2f..9eeaaa0a 100644
--- a/tests/worker/thread-worker.test.js
+++ b/tests/worker/thread-worker.test.js
@@ -4,11 +4,12 @@ const { ThreadWorker } = require('../../lib')
 describe('Thread worker test suite', () => {
   let numberOfMessagesPosted = 0
   const postMessage = function () {
-    numberOfMessagesPosted++
+    ++numberOfMessagesPosted
   }
   class SpyWorker extends ThreadWorker {
-    getMainWorker () {
-      return { postMessage }
+    constructor (fn) {
+      super(fn)
+      this.port = { postMessage }
     }
   }
 
@@ -17,7 +18,15 @@ describe('Thread worker test suite', () => {
     expect(worker.opts.maxInactiveTime).toStrictEqual(60000)
   })
 
-  it('Verify worker invoke the getMainWorker and postMessage methods', () => {
+  it('Verify that handleError() method is working properly', () => {
+    const error = new Error('Error as an error')
+    const worker = new ThreadWorker(() => {})
+    expect(worker.handleError(error)).toStrictEqual(error)
+    const errorMessage = 'Error as a string'
+    expect(worker.handleError(errorMessage)).toStrictEqual(errorMessage)
+  })
+
+  it('Verify worker invokes the postMessage() method on port property', () => {
     const worker = new SpyWorker(() => {})
     worker.sendToMainWorker({ ok: 1 })
     expect(numberOfMessagesPosted).toBe(1)