napi-rs/examples/napi/__tests__/worker-thread.spec.ts
forehal a781a4f27e feat(cli): brand new cli tool with both cli and programmatical usage (#1492)
BREAKING CHANGE: requires node >= 16 and some cli options have been renamed
2023-04-06 11:04:53 +08:00

70 lines
1.8 KiB
TypeScript

import { join } from 'path'
import { Worker } from 'worker_threads'
import test from 'ava'
import { Animal, Kind, DEFAULT_COST } from '../index'
const t =
process.arch === 'arm64' && process.platform === 'linux' ? test.skip : test
t('should be able to require in worker thread', async (t) => {
await Promise.all(
Array.from({ length: 100 }).map(() => {
const w = new Worker(join(__dirname, 'worker.js'))
return new Promise<void>((resolve, reject) => {
w.postMessage({ type: 'require' })
w.on('message', (msg) => {
t.is(msg, Animal.withKind(Kind.Cat).whoami() + DEFAULT_COST)
resolve()
})
w.on('error', (err) => {
reject(err)
})
})
.then(() => w.terminate())
.then(() => {
t.pass()
})
}),
)
})
t('custom GC works on worker_threads', async (t) => {
await Promise.all(
Array.from({ length: 50 }).map(() =>
Promise.all([
new Promise<Worker>((resolve, reject) => {
const w = new Worker(join(__dirname, 'worker.js'))
w.postMessage({
type: 'async:buffer',
})
w.on('message', (msg) => {
t.is(msg, 'done')
resolve(w)
})
w.on('error', (err) => {
reject(err)
})
}).then((w) => {
return w.terminate()
}),
new Promise<Worker>((resolve, reject) => {
const w = new Worker(join(__dirname, 'worker.js'))
w.postMessage({
type: 'async:arraybuffer',
})
w.on('message', (msg) => {
t.is(msg, 'done')
resolve(w)
})
w.on('error', (err) => {
reject(err)
})
}).then((w) => {
return w.terminate()
}),
]),
),
)
})