如何用Jest测试NodeJS中的Worker Thread？代码未执行问题求助

1. 主进程未给Worker发送消息 #

你的foo函数仅创建了Worker实例，但从未调用worker.postMessage()给Worker发送消息，Worker里监听的parentPort.on("message")事件自然不会触发，代码也就不会执行。

修改foo函数，添加消息发送逻辑：

// index.ts 
function foo(taskData: any){
  const worker = new Worker(path.resolve(__dirname, "./worker.ts"));
  // 给Worker发送消息，触发其处理逻辑
  worker.postMessage(taskData);
  worker.on("message", (e) => {
   // 处理Worker返回的结果
  });
  // 返回Promise，方便Jest处理异步测试
  return new Promise((resolve, reject) => {
    worker.on("message", resolve);
    worker.on("error", reject);
  });
}

2. 循环依赖导致模块加载异常 #

Worker里require("./index")，而index.ts又创建Worker，形成循环依赖，可能导致doStuff函数无法被正确加载，或是Worker模块初始化失败。

解决办法：把doStuff抽离到独立文件（比如utils.ts），避免循环依赖：

// utils.ts
export async function doStuff(data: any) {
  // 原doStuff的业务逻辑
}

之后在index.ts和worker.ts中分别引入这个独立文件：

// index.ts
import { doStuff } from "./utils";
// ... 其他代码

// worker.ts
const { parentPort } = require("worker_threads");
const { doStuff } = require("./utils"); // 改为引入utils.ts
// ... 其他代码

3. Jest测试未处理异步逻辑 #

Worker操作是异步的，如果测试用例没处理异步逻辑，Jest会在Worker执行完成前就结束测试，看起来像是代码没执行。

修改测试用例，用async/await等待异步结果：

// foo.test.ts
import { foo } from "./index";

test("测试Worker执行逻辑", async () => {
  const result = await foo("test data");
  expect(result).toEqual({ data: "预期处理结果" });
});

4. Mock Worker的正确姿势 #

如果要Mock Worker，需要模拟Worker类的构造、postMessage方法以及事件监听逻辑，示例如下：

// foo.test.ts
import { Worker } from "worker_threads";
import { foo } from "./index";

// Mock Worker类
jest.mock("worker_threads", () => ({
  Worker: jest.fn().mockImplementation(() => ({
    postMessage: jest.fn(),
    on: jest.fn((event, callback) => {
      // 模拟Worker收到消息后触发回调
      if (event === "message") {
        setTimeout(() => callback({ data: "mock处理结果" }), 0);
      }
    }),
  })),
}));

test("测试Worker调用逻辑", async () => {
  const result = await foo("test data");
  expect(Worker).toHaveBeenCalled();
  expect(result).toEqual({ data: "mock处理结果" });
});

内容的提问来源于stack exchange，提问作者HelloWorldEngineer