Mocha测试共享行为实现咨询：如何复用钩子避免重复代码

一、自定义可复用钩子函数（无需切换测试库） #

这是最贴合需求的低成本方案，直接把重复的before/after/beforeEach逻辑封装成独立函数，在测试套件中调用即可实现你想要的简洁写法。

示例实现： #

// test-helpers.js
// 封装数据库初始化/销毁逻辑
function runWithDatabase(context, dbConfig) {
  let db;
  // 挂载before钩子
  context.before(async function() {
    db = await setupDatabase(dbConfig);
    // 将数据库实例挂载到上下文，方便测试用例直接调用
    context.db = db;
  });

  // 挂载after钩子
  context.after(async function() {
    await deleteDb(db);
  });

  // 挂载beforeEach钩子
  context.beforeEach(async function() {
    await resetDb(db);
  });
}

// 封装环境变量备份/恢复逻辑
function runWithFixedEnvironment(context, fixtureEnv) {
  let envBackup;
  context.before(async function() {
    envBackup = saveAllEnvironmentVariables();
    Object.assign(process.env, fixtureEnv);
  });

  context.after(async function() {
    restoreAllEnvironmentVariables(envBackup);
  });
}

在测试中使用： #

const { runWithDatabase, runWithFixedEnvironment } = require('./test-helpers');

describe('测试特定环境下的功能', function(){
  runWithDatabase(this, {name: 'test-db-123'});
  runWithFixedEnvironment(this, {foo: 'bar', cow: 'moo'});

  it('验证数据库读取逻辑', async function() {
    const data = await this.db.query('SELECT * FROM test_table');
    // 编写断言逻辑
  });
});

所有复用逻辑集中在工具文件中，维护和修改都只需要改一处，完全避免重复代码。

二、进阶：Mixin模式封装组合逻辑 #

如果需要同时复用多组钩子，可以把它们封装成mixin对象，合并到测试上下文：

// test-helpers.js
const databaseMixin = {
  async setupDb(config) {
    this.db = await setupDatabase(config);
  },
  async teardownDb() {
    await deleteDb(this.db);
  },
  async resetDb() {
    await resetDb(this.db);
  }
};

function withDatabase(dbConfig) {
  return function(context) {
    Object.assign(context, databaseMixin);
    context.before(async function() { await this.setupDb(dbConfig); });
    context.after(async function() { await this.teardownDb(); });
    context.beforeEach(async function() { await this.resetDb(); });
  };
}

使用方式： #

describe('测试功能模块', function(){
  withDatabase({name: 'test-db-123'})(this);

  it('验证数据写入', async function() {
    await this.db.insert({id: 1, value: 'test'});
    // 断言逻辑
  });
});

三、切换支持扩展机制的测试库（可选） #

如果愿意替换测试库，以下工具原生支持类似JUnit Rule/Extension的复用能力：

• Jest：可通过setupFilesAfterEnv配置全局钩子，或自定义testEnvironment封装全局初始化逻辑；也能编写testWithDb这类自定义测试函数复用逻辑。
• AVA：支持测试宏（Test Macros），把重复逻辑封装成宏后直接在测试中调用。
• Cypress（端到端测试场景）：通过自定义命令和插件轻松复用前置逻辑。

不过遗留项目切换测试库需要评估迁移成本，优先推荐第一种自定义钩子方案，成本最低且适配现有代码。

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