单元测试中 Lodash once() 函数状态重置策略

// 假设这是你的 Jest 配置中指定的 setupFiles 文件，例如 setupTests.js
// 或者直接在你的测试文件顶部

// 导入真实的 lodash 模块，以便我们可以复制其非 once 的部分
const lodash = jest.requireActual("lodash");

// 模拟 lodash 模块
jest.mock("lodash", () => ({
  ...lodash, // 保留 lodash 模块中除 once 之外的所有导出
  once: jest.fn((fn) => fn), // 模拟 once 函数，使其直接返回传入的原始函数 fn
}));

// --- 以下是测试文件的示例 ---
import { once } from 'lodash'; // 此时导入的 once 是被模拟过的

describe('My Function with Mocked Once (Disabled Cache)', () => {
  let expensiveCalculation;
  let wrappedFunction;

  beforeEach(() => {
    // 每次测试前，创建一个新的 Jest mock 函数作为昂贵的计算
    expensiveCalculation = jest.fn(() => "calculated result");
    // 使用模拟的 once 包装 expensiveCalculation
    wrappedFunction = once(expensiveCalculation);
  });

  afterEach(() => {
    // 清除所有 Jest mock 函数的调用历史，包括 expensiveCalculation
    jest.clearAllMocks();
  });

  test('should call the original function every time when cache is disabled', () => {
    wrappedFunction(); // 第一次调用
    wrappedFunction(); // 第二次调用
    expect(expensiveCalculation).toHaveBeenCalledTimes(2); // 预期原始函数被调用两次
  });

  test('another test should also call the original function every time (due to beforeEach)', () => {
    wrappedFunction(); // 再次调用
    expect(expensiveCalculation).toHaveBeenCalledTimes(1); // 由于 beforeEach 重新设置了 expensiveCalculation，所以这次是第一次调用
  });
});