如何在Playwright测试结束后自动删除数据库条目？

以下是几种实用的解决思路，根据你的测试框架和数据库类型选择合适的方式：

1. 利用测试框架的钩子函数 #

绝大多数E2E测试框架（如Jest、Cypress、Playwright）都提供了测试前后执行的钩子，在测试结束后调用删除逻辑即可。

比如用Jest的afterEach钩子（每个测试用例结束后执行）：

// 假设使用MongoDB的Mongoose模型
const Document = require('../models/Document');

afterEach(async () => {
  // 通过文档名称精准删除
  await Document.deleteOne({ name: 'Document01' });
  // 或者调用后端API接口删除
  // await fetch('/api/documents/Document01', { method: 'DELETE' });
});

test('创建文档的E2E测试', async () => {
  // 这里是你的测试逻辑，创建Document01
});

如果是Cypress，可以用afterEach钩子：

afterEach(() => {
  cy.request('DELETE', '/api/documents', { name: 'Document01' });
});

2. 数据库事务回滚（高效方案） #

如果你的数据库支持事务（如PostgreSQL、MySQL InnoDB、MongoDB 4.0+），可以在测试开始前开启事务，测试结束后直接回滚，所有测试生成的数据都会被自动撤销，无需单独删除。

以TypeORM为例：

import { getConnection } from 'typeorm';

beforeEach(async () => {
  await getConnection().startTransaction();
});

afterEach(async () => {
  await getConnection().rollbackTransaction();
});

// 测试用例中创建Document01，结束后事务回滚，数据自动消失

3. 测试后执行清理脚本 #

如果是通过命令行批量运行测试，可以在测试命令后追加一个清理脚本，统一删除测试数据。

比如在package.json中配置脚本：

{
  "scripts": {
    "test:e2e": "playwright test && node scripts/clean-test-data.js"
  }
}

scripts/clean-test-data.js的内容：

const mongoose = require('mongoose');
const Document = require('../models/Document');

async function cleanTestDocs() {
  await mongoose.connect('mongodb://localhost:27017/your-db');
  await Document.deleteOne({ name: 'Document01' });
  await mongoose.disconnect();
  console.log('测试文档Document01已删除');
}

cleanTestDocs().catch(err => {
  console.error('清理失败:', err);
  process.exit(1);
});

4. 标记测试数据，批量清理 #

如果测试中会生成多个测试文档，建议创建时给数据加一个测试标记（如isTest: true），测试结束后批量删除所有带标记的数据，避免漏删。

创建文档时：

await Document.create({
  name: 'Document01',
  content: '测试内容',
  isTest: true // 标记为测试数据
});

测试结束后批量删除：

afterAll(async () => {
  await Document.deleteMany({ isTest: true });
});

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