test: use sinon stub to simplify code

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

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

describe('Thread worker test suite', () => {
  class SpyWorker extends ThreadWorker {
    constructor (fn) {
      super(fn)
      this.port = { postMessage: stub().returns() }
    }
  }

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

  it('Verify that handleError() method is working properly', () => {
    const error = new Error('Error as an error')
    const worker = new ThreadWorker(() => {})
    expect(worker.handleError(error)).toBeInstanceOf(Error)
    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(worker.port.postMessage.calledOnce).toBe(true)
  })
})