X-Git-Url: https://git.piment-noir.org/?a=blobdiff_plain;f=tests%2Fworker%2Fcluster-worker.test.js;h=2d9104bb3c185b52bf05f8ec114d8d7008745583;hb=refs%2Ftags%2Fv2.6.9;hp=ff749c8f721f964762c144ece84a7933711367fa;hpb=78cea37e264d5ca527bc42eb056f3b9579a2b2c4;p=poolifier.git

diff --git a/tests/worker/cluster-worker.test.js b/tests/worker/cluster-worker.test.js
index ff749c8f..2d9104bb 100644
--- a/tests/worker/cluster-worker.test.js
+++ b/tests/worker/cluster-worker.test.js
@@ -2,14 +2,30 @@ const { expect } = require('expect')
 const { ClusterWorker } = require('../../lib')
 
 describe('Cluster worker test suite', () => {
+  let numberOfMessagesSent = 0
+  const send = function () {
+    ++numberOfMessagesSent
+  }
+  class SpyWorker extends ClusterWorker {
+    getMainWorker () {
+      return { send }
+    }
+  }
+
   it('Verify worker has default maxInactiveTime', () => {
     const worker = new ClusterWorker(() => {})
     expect(worker.opts.maxInactiveTime).toStrictEqual(60000)
   })
 
-  it('Verify that handleError function works properly', () => {
+  it('Verify that handleError() method works properly', () => {
     const errorMessage = 'Error as a string'
     const worker = new ClusterWorker(() => {})
     expect(worker.handleError(errorMessage)).toStrictEqual(errorMessage)
   })
+
+  it('Verify worker invoke the getMainWorker() and send() methods', () => {
+    const worker = new SpyWorker(() => {})
+    worker.sendToMainWorker({ ok: 1 })
+    expect(numberOfMessagesSent).toBe(1)
+  })
 })