source-worker.compat.spec.ts 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. import { copyFile, mkdtemp, rm } from 'node:fs/promises'
  2. import { tmpdir } from 'node:os'
  3. import { join } from 'node:path'
  4. import { Worker } from 'node:worker_threads'
  5. import { expect, it } from 'vitest'
  6. import { decodeWorkerJson } from '../src/worker-json.ts'
  7. /**
  8. * Prove the unbuilt worker is a self-contained source closure. Copying it out
  9. * of the workspace makes any package runtime import fail even when local
  10. * `lib/` artifacts happen to exist.
  11. */
  12. it('boots the source worker without workspace package outputs', async () => {
  13. const directory = await mkdtemp(join(tmpdir(), 'dsh-code-source-worker-'))
  14. let worker: Worker | undefined
  15. try {
  16. const files = ['worker.ts', 'bootstrap.ts', 'protocol.ts', 'worker-json.ts', 'output-json.ts']
  17. await Promise.all(files.map(async (file) => {
  18. await copyFile(new URL(`../src/${file}`, import.meta.url), join(directory, file))
  19. }))
  20. worker = new Worker(join(directory, 'worker.ts'), {
  21. workerData: { code: 'return { answer: 42 }', namespaces: [], maxOutputBytes: 65_536 },
  22. env: {},
  23. execArgv: [],
  24. })
  25. const message = await new Promise<unknown>((resolve, reject) => {
  26. worker?.once('message', resolve)
  27. worker?.once('error', reject)
  28. })
  29. expect(message).toMatchObject({ type: 'done' })
  30. const value = typeof message === 'object' && message !== null ? (message as { value?: unknown }).value : undefined
  31. expect(decodeWorkerJson(value)).toEqual({ answer: 42 })
  32. } finally {
  33. if (worker) await worker.terminate()
  34. await rm(directory, { recursive: true, force: true })
  35. }
  36. })