test: use sinon stub to simplify code

[poolifier.git] / tests / worker / cluster-worker.test.mjs

import { expect } from 'expect'
import { restore, stub } from 'sinon'
import { ClusterWorker } from '../../lib/index.js'

describe('Cluster worker test suite', () => {
  const sendStub = stub().returns()
  class SpyWorker extends ClusterWorker {
    getMainWorker () {
      return { send: sendStub }
    }
  }

  afterEach(() => {
    restore()
  })

  it('Verify that handleError() method is working properly', () => {
    const error = new Error('Error as an error')
    const worker = new ClusterWorker(() => {})
    expect(worker.handleError(error)).not.toBeInstanceOf(Error)
    expect(worker.handleError(error)).toStrictEqual(error.message)
    const errorMessage = 'Error as a string'
    expect(worker.handleError(errorMessage)).toStrictEqual(errorMessage)
  })

  it('Verify worker invokes the getMainWorker() and send() methods', () => {
    const worker = new SpyWorker(() => {})
    worker.sendToMainWorker({ ok: 1 })
    expect(worker.getMainWorker().send.calledOnce).toBe(true)
  })
})