| 1 | import { expect } from 'expect' |
| 2 | import { restore, stub } from 'sinon' |
| 3 | |
| 4 | import { ThreadWorker } from '../../lib/index.cjs' |
| 5 | import { DEFAULT_TASK_NAME } from '../../lib/utils.cjs' |
| 6 | |
| 7 | describe('Thread worker test suite', () => { |
| 8 | afterEach(() => { |
| 9 | restore() |
| 10 | }) |
| 11 | |
| 12 | it('Verify worker properties value after initialization', () => { |
| 13 | const worker = new ThreadWorker(() => {}) |
| 14 | expect(worker.isMain).toBe(true) |
| 15 | expect(worker.mainWorker).toBe(null) |
| 16 | expect(worker.taskFunctions).toBeInstanceOf(Map) |
| 17 | expect(worker.taskFunctions.size).toBe(2) |
| 18 | }) |
| 19 | |
| 20 | it('Verify that sync kill handler is called when worker is killed', () => { |
| 21 | const worker = new ThreadWorker(() => {}, { |
| 22 | killHandler: stub().returns() |
| 23 | }) |
| 24 | worker.isMain = false |
| 25 | worker.port = { |
| 26 | postMessage: stub().returns(), |
| 27 | unref: stub().returns(), |
| 28 | close: stub().returns() |
| 29 | } |
| 30 | worker.handleKillMessage() |
| 31 | expect(worker.port.postMessage.calledOnce).toBe(true) |
| 32 | expect(worker.port.unref.calledOnce).toBe(true) |
| 33 | expect(worker.port.close.calledOnce).toBe(true) |
| 34 | expect(worker.opts.killHandler.calledOnce).toBe(true) |
| 35 | }) |
| 36 | |
| 37 | it('Verify that removeTaskFunction() is working', () => { |
| 38 | const fn1 = () => { |
| 39 | return 1 |
| 40 | } |
| 41 | const fn2 = () => { |
| 42 | return 2 |
| 43 | } |
| 44 | const worker = new ThreadWorker({ fn1, fn2 }) |
| 45 | worker.port = { |
| 46 | postMessage: stub().returns() |
| 47 | } |
| 48 | expect(worker.removeTaskFunction(0, fn1)).toStrictEqual({ |
| 49 | status: false, |
| 50 | error: new TypeError('name parameter is not a string') |
| 51 | }) |
| 52 | expect(worker.removeTaskFunction('', fn1)).toStrictEqual({ |
| 53 | status: false, |
| 54 | error: new TypeError('name parameter is an empty string') |
| 55 | }) |
| 56 | expect(worker.taskFunctions.get(DEFAULT_TASK_NAME)).toStrictEqual({ |
| 57 | taskFunction: expect.any(Function) |
| 58 | }) |
| 59 | expect(worker.taskFunctions.get('fn1')).toStrictEqual({ |
| 60 | taskFunction: expect.any(Function) |
| 61 | }) |
| 62 | expect(worker.taskFunctions.get('fn2')).toStrictEqual({ |
| 63 | taskFunction: expect.any(Function) |
| 64 | }) |
| 65 | expect(worker.taskFunctions.size).toBe(3) |
| 66 | expect(worker.taskFunctions.get(DEFAULT_TASK_NAME)).toStrictEqual( |
| 67 | worker.taskFunctions.get('fn1') |
| 68 | ) |
| 69 | expect(worker.removeTaskFunction(DEFAULT_TASK_NAME)).toStrictEqual({ |
| 70 | status: false, |
| 71 | error: new Error( |
| 72 | 'Cannot remove the task function with the default reserved name' |
| 73 | ) |
| 74 | }) |
| 75 | expect(worker.removeTaskFunction('fn1')).toStrictEqual({ |
| 76 | status: false, |
| 77 | error: new Error( |
| 78 | 'Cannot remove the task function used as the default task function' |
| 79 | ) |
| 80 | }) |
| 81 | worker.removeTaskFunction('fn2') |
| 82 | expect(worker.taskFunctions.get(DEFAULT_TASK_NAME)).toStrictEqual({ |
| 83 | taskFunction: expect.any(Function) |
| 84 | }) |
| 85 | expect(worker.taskFunctions.get('fn1')).toStrictEqual({ |
| 86 | taskFunction: expect.any(Function) |
| 87 | }) |
| 88 | expect(worker.taskFunctions.get('fn2')).toBeUndefined() |
| 89 | expect(worker.taskFunctions.size).toBe(2) |
| 90 | expect(worker.port.postMessage.calledOnce).toBe(true) |
| 91 | }) |
| 92 | |
| 93 | it('Verify that handleError() method is working properly', () => { |
| 94 | const error = new Error('Error as an error') |
| 95 | const worker = new ThreadWorker(() => {}) |
| 96 | expect(worker.handleError(error)).toBeInstanceOf(Error) |
| 97 | expect(worker.handleError(error)).toStrictEqual(error) |
| 98 | const errorMessage = 'Error as a string' |
| 99 | expect(worker.handleError(errorMessage)).toStrictEqual(errorMessage) |
| 100 | }) |
| 101 | |
| 102 | it('Verify that sendToMainWorker() method invokes the port property postMessage() method', () => { |
| 103 | const worker = new ThreadWorker(() => {}) |
| 104 | worker.port = { postMessage: stub().returns() } |
| 105 | worker.sendToMainWorker({ ok: 1 }) |
| 106 | expect(worker.port.postMessage.calledOnce).toBe(true) |
| 107 | }) |
| 108 | }) |