如何在 JavaScript 中安全地覆盖全局函数以实现单元测试模拟

本文介绍通过函数表达式替代函数声明的方式，在不修改原始逻辑的前提下，灵活覆盖全局函数以支持单元测试中的行为模拟。

在 JavaScript 单元测试中，常需对依赖的底层函数（如 getData）进行模拟（mock），使其返回预设数据而非真实调用（如访问数据库）。但若目标函数使用 函数声明（function getData() { ... }），由于其存在提升（hoisting）且不可重赋值覆盖的特性，在测试中直接赋值 getData = mock_getData 通常无效或行为不可靠。

✅ 正确做法是：将待模拟的函数定义为 函数表达式（function expression），并挂载到全局作用域（如 window 或直接作为全局变量），从而支持运行时动态重写：

// ✅ 推荐：使用函数表达式定义，便于后续覆盖
var getData = function() {
  // 实际逻辑：从数据库获取数据
  throw new Error('Real database call not allowed in tests');
};

var getUsers = function() {
  var data = getData(); // 调用的是可被覆盖的变量引用
  return data.map(u => ({ ...u, fetched: true }));
};

function main() {
  var users = getUsers();
  // ...
}

在测试代码中，只需在 test() 执行前重新赋值 getData，即可让 getUsers() 内部调用自动命中模拟实现：

// ✅ 测试前覆盖全局 getData
var mock_getData = function() {
  return [{ id: 0, name: 'Alice' }, { id: 1, name: 'Bob' }];
};

test('getUsers', function() {
  // 关键：临时覆盖 getData
  var originalGetData = getData;
  getData = mock_getData;

  try {
    var users = getUsers();
    myAssert(users[0].name, 'Alice');
    myAssert(users[1].name, 'Bob');
  } finally {
    // ? 恢复原始函数（避免影响其他测试）
    getData = originalGetData;
  }
});

? 进阶建议：

立即学习“Java免费学习笔记

• 使用 beforeEach/afterEach 机制（即使自研库也可模拟）统一管理 mock 的设置与还原；
• 对于浏览器环境，可将函数挂载至 window.getData；Node.js 环境则挂载到 global.getData；
• 避免污染全局命名空间，可封装为模块或使用 Object.defineProperty(globalThis, 'getData', { writable: true, ... }) 显式控制可写性。

总结：函数声明不可覆盖，函数表达式可重赋值——这是实现轻量级、无框架依赖的函数级 mock 的核心前提。通过结构化地将“可模拟函数”定义为可变引用，你既能保持生产代码简洁，又能为测试提供强大可控的注入能力。