如何在Jest中测试Worker默认函数内的sortData函数？

你的问题很常见——Web Worker的代码结构经常会把业务逻辑函数放在初始化闭包里，导致测试时无法直接访问。这里有几个实用的解决方案，既不破坏Worker的正常功能，又能方便测试：

方案1：重构代码，单独导出可测试函数 #

这是最简洁的方式，把sortData从闭包里抽出来作为独立导出，同时在Worker逻辑中继续使用它：

// sortWorker.js
// 单独导出sortData供测试使用
export function sortData(data) {
  // 你的排序逻辑
  return data.sort((a, b) => a - b); // 示例逻辑，替换成你的实际代码
}

export default () => {
  self.addEventListener("message", e => {
    postMessage(sortData(e.data));
  });
};

测试时直接导入这个函数即可：

// sortWorker.test.js
import { sortData } from './sortWorker';

test('sortData correctly sorts an array', () => {
  const input = [5, 2, 8, 1];
  const expected = [1, 2, 5, 8];
  expect(sortData(input)).toEqual(expected);
});

这种方式完全不影响Worker的正常使用——当你在主线程中创建Worker时，依然使用默认导出的初始化函数，比如：

// 主线程代码示例
const worker = new Worker(new URL('./sortWorker.js', import.meta.url));

方案2：在测试环境暴露私有函数（无需重构核心逻辑） #

如果因为某些原因不能修改函数的导出结构，可以在Worker代码中添加一个测试环境的判断，把sortData挂载到self对象上：

// sortWorker.js
export default () => {
  function sortData(data) {
    // 你的排序逻辑
    return data.sort((a, b) => a - b);
  }

  // 仅在测试环境暴露函数，避免影响生产环境
  if (process.env.NODE_ENV === 'test') {
    self.sortData = sortData;
  }

  self.addEventListener("message", e => {
    postMessage(sortData(e.data));
  });
};

然后在Jest测试中模拟Worker的全局环境，执行初始化函数后就能访问到sortData：

// sortWorker.test.js
import initWorker from './sortWorker';

test('sortData works as expected', () => {
  // 模拟Worker的self对象
  global.self = {
    addEventListener: jest.fn(),
    postMessage: jest.fn()
  };

  // 执行Worker的初始化逻辑
  initWorker();

  // 现在可以调用暴露出来的sortData函数
  const testData = [3, 1, 4, 1, 5];
  expect(global.self.sortData(testData)).toEqual([1, 1, 3, 4, 5]);

  // 测试完成后清理全局变量，避免影响其他测试用例
  delete global.self;
});

方案3：模拟Worker消息机制测试完整流程 #

如果你想测试Worker的完整消息处理流程（而不仅仅是sortData函数本身），可以直接模拟消息发送的场景：

// sortWorker.test.js
import initWorker from './sortWorker';

test('Worker responds with sorted data when receiving a message', done => {
  global.self = {
    addEventListener: (event, handler) => {
      // 模拟收到message事件
      if (event === 'message') {
        handler({ data: [3, 1, 2] });
      }
    },
    postMessage: (result) => {
      expect(result).toEqual([1, 2, 3]);
      done(); // 告知Jest异步测试已完成
    }
  };

  initWorker();
  delete global.self;
});

这个方案更贴近Worker的实际运行场景，适合测试端到端的消息处理逻辑。

内容的提问来源于stack exchange，提问作者craig-nerd