单元测试中 Lodash once() 函数状态重置策略-js教程-PHP中文网

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

聖光之護

发布： 2025-10-24 12:29:10

原创

277人浏览过

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

本文探讨了在单元测试中如何解决 lodash `once()` 函数因其内部状态持久化而导致的测试污染问题。通过利用 jest 等测试框架的模块模拟（mocking）能力，可以有效地重置或自定义 `once()` 的行为，确保每个测试用例都在一个干净、可控的环境中运行，从而提高测试的可靠性和隔离性。

引言：lodash.once() 的测试挑战

lodash.once() 是一个实用的工具函数，它确保一个函数无论被调用多少次，都只执行一次，并缓存其首次执行的结果。这种行为在许多场景下非常有用，例如初始化昂贵的计算或确保只执行一次副作用操作。然而，在进行单元测试时，once() 的这种状态持久化特性却可能引入问题。

当一个被 once() 包装的函数在某个测试用例中执行后，其内部状态（已执行、已缓存结果）会持续存在。如果在后续的测试用例中再次调用同一个被 once() 包装的函数，它将不会重新执行，而是直接返回之前缓存的结果。这导致了测试间的状态污染，使得测试用例不再独立，结果可能变得不可预测或难以调试。为了编写健壮且独立的单元测试，我们需要一种机制来“重置” once() 的行为，确保每个测试用例都能从一个干净、初始化的状态开始。

解决方案：利用 Jest 模块模拟重置 once() 行为

解决 lodash.once() 在单元测试中状态污染问题的核心策略是使用模块模拟（Module Mocking）。Jest 提供了强大的 jest.mock() 功能，允许我们拦截对特定模块的导入，并替换其内部的函数实现。通过模拟 lodash 模块中的 once 函数，我们可以完全控制其行为，使其在测试环境中按照我们期望的方式工作，从而实现状态的重置或自定义。

Quicktools Background Remover

Picsart推出的图片背景移除工具

查看详情

方法一：禁用 once() 的缓存行为

最直接的解决方案是完全禁用 once() 的缓存行为，使其在每次被调用时都执行原始函数。这种方法适用于那些测试场景中，我们只关心原始函数是否被调用，而不关心 once() 的缓存逻辑。

我们可以通过在 Jest 的 setupFiles 配置中或直接在测试文件中使用 jest.mock() 来实现：

// 假设这是你的 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，所以这次是第一次调用
  });
});

登录后复制

解释： 通过 jest.fn((fn) => fn)，我们让 lodash.once 的模拟版本不再执行任何缓存

以上就是单元测试中 Lodash once() 函数状态重置策略的详细内容，更多请关注php中文网其它相关文章！