基于Jest+Supertest的Node.js API抽象测试函数技术咨询

这思路真的很棒！把重复的权限场景测试逻辑抽成通用函数，完全符合DRY（Don't Repeat Yourself）原则，能帮你省不少重复代码的功夫～

合理性肯定 #

• 核心方向正确：把「未授权/普通用户/管理员」这三类最常见的权限测试场景统一封装，避免每个API都写三遍几乎一样的测试代码，极大提升了代码复用性
• 基础可配置性到位：method、url、statuses、body这些核心参数都支持外部传入，能适配大部分REST API的测试需求

1. 补全并优化请求头逻辑 #

你代码里的Author...应该是Authorization头的截断，这里要注意只有当token存在时才设置这个头，避免传入null导致请求头异常：

// 替换原来的请求构建逻辑
let requestChain = request(app)[method](url)
  .set('Accept', 'application/json');

// 处理请求体（避免空对象无意义发送）
if (Object.keys(body).length > 0) {
  requestChain = requestChain.send(body);
}

// 处理认证头
const currentToken = tokens[i];
if (currentToken) {
  requestChain = requestChain.set('Authorization', `Bearer ${currentToken}`);
}

const response = await requestChain;

2. 让状态码断言更清晰 #

现在的statuses数组最好和testNames的顺序严格对应（未授权→普通用户→管理员），测试时直接断言对应下标状态码，同时可以给statuses加默认值，覆盖最常见的场景：

// 修改函数参数默认值
function testIt(method = 'get', url = '/', statuses = [401, 200, 200], body = {}) {
  // ...
  // 测试里的断言
  expect(response.statusCode).toBe(statuses[i]);
}

3. 用forEach简化循环逻辑 #

把原来的for循环换成场景对象数组+forEach，代码更简洁易读，也方便后续扩展场景：

const testScenarios = [
  { name: 'unauthorized', token: null },
  { name: 'user', token: tokensConfig.user },
  { name: 'admin', token: tokensConfig.admin }
];

testScenarios.forEach((scenario, index) => {
  test(scenario.name, async () => {
    // 这里用scenario.token代替tokens[i]
    // ... 其他请求和断言逻辑
  });
});

4. 增加自定义断言的扩展性 #

如果不同API需要额外的响应体校验，可以加一个customAssertions参数，让每个场景支持自定义断言：

function testIt(method = 'get', url = '/', statuses = [401, 200, 200], body = {}, customAssertions = []) {
  const testScenarios = [
    { name: 'unauthorized', token: null },
    { name: 'user', token: tokensConfig.user },
    { name: 'admin', token: tokensConfig.admin }
  ];

  testScenarios.forEach((scenario, index) => {
    test(scenario.name, async () => {
      // ... 请求逻辑
      expect(response.statusCode).toBe(statuses[index]);
      // 执行自定义断言（如果有）
      if (customAssertions[index]) {
        customAssertions[index](response);
      }
    });
  });
}

// 使用示例：给不同场景加专属断言
testIt('get', '/api/admin/dashboard', [401, 403, 200], {}, [
  null, // 未授权场景不需要额外断言
  (res) => expect(res.body.message).toBe('权限不足'),
  (res) => expect(res.body.dashboardData).toBeDefined()
]);

5. 处理边界场景 #

比如当某个API不需要传body时，避免send空对象；如果你的项目里有其他认证类型（比如Basic认证），可以加一个authType参数来适配，但如果只是Bearer认证，目前的逻辑就足够了。

• 如果API的权限场景不止这三种怎么办？
  可以把testScenarios改成函数参数传入，比如：

  function testIt(method, url, statuses, body, scenarios) {
    scenarios.forEach((scenario, index) => {
      // ... 测试逻辑
    });
  }
  

  这样你可以根据不同API传入自定义的场景数组，比如增加moderator角色。

• 如何处理POST/PUT的复杂请求体？
  直接把完整的请求体对象传入body参数即可，supertest的send方法会自动处理JSON格式的请求体（因为你已经设置了Accept: application/json）。

• 要不要把这个函数放到单独的工具模块里？
  非常建议！把这个通用测试函数抽成test-utils.js之类的工具模块，所有测试文件都可以导入使用，进一步提升代码复用性。

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