如何拆分Python Selenium Web应用测试类且避免数据库备份代码重复？

我明白你的痛点——把一个大测试类拆成多个文件后，不想在每个类里重复写备份恢复的逻辑。下面给你几个实用的方案，都是基于unittest生态的，不用额外引入太多第三方工具：

方案1：创建基类测试类（最常用） #

核心思路是把数据库备份/恢复的逻辑抽出来放到一个基类里，其他所有测试类都继承这个基类，自动复用这些方法。

步骤1：编写基类（比如放在base_test.py） #

import unittest
from your_backup_utils import backup_database, restore_database

# 基类名字不要以Test开头，避免被测试运行器直接执行
class BaseDBTest(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        """所有测试类执行前，先备份数据库"""
        try:
            cls.backup_file = backup_database()
            print(f"✅ 数据库已备份到: {cls.backup_file}")
        except Exception as e:
            # 备份失败直接终止测试，避免污染环境
            raise RuntimeError(f"数据库备份失败: {str(e)}") from e

    @classmethod
    def tearDownClass(cls):
        """所有测试类执行后，恢复数据库"""
        try:
            restore_database(cls.backup_file)
            print(f"✅ 已从备份 {cls.backup_file} 恢复数据库")
        except Exception as e:
            print(f"⚠️ 数据库恢复失败，请手动检查: {str(e)}")

步骤2：其他测试类继承基类 #

比如在test_user_functions.py里：

import unittest
from base_test import BaseDBTest

class UserRegressionTest(BaseDBTest):
    def test_user_login(self):
        # 你的登录测试逻辑
        self.assertEqual(..., ...)

    def test_user_profile_update(self):
        # 你的用户资料更新测试逻辑
        pass

再比如test_product_management.py里：

import unittest
from base_test import BaseDBTest

class ProductRegressionTest(BaseDBTest):
    def test_product_add(self):
        # 商品添加测试逻辑
        pass

    def test_product_delete(self):
        # 商品删除测试逻辑
        pass

注意：这种方式会给每个测试类单独执行一次备份和恢复，好处是每个测试类的环境完全独立，不会互相干扰；缺点是如果测试类很多，会增加整体测试时间。

方案2：全局测试套件（一次备份恢复所有测试） #

如果你的所有测试用例之间没有状态污染（比如每个测试都会重置自己修改的数据），可以用这个方案——整个测试流程只做一次备份和恢复，提升测试速度。

编写测试套件文件（比如test_suite.py） #

import unittest
from your_backup_utils import backup_database, restore_database
# 导入所有拆分后的测试类
from test_user_functions import UserRegressionTest
from test_product_management import ProductRegressionTest

def run_all_tests():
    # 先做全局数据库备份
    backup_file = backup_database()
    print(f"🌐 全局数据库备份完成: {backup_file}")

    # 组装所有测试用例
    test_suite = unittest.TestSuite()
    test_suite.addTest(unittest.makeSuite(UserRegressionTest))
    test_suite.addTest(unittest.makeSuite(ProductRegressionTest))

    # 运行测试
    runner = unittest.TextTestRunner(verbosity=2)
    result = runner.run(test_suite)

    # 测试完成后恢复数据库
    restore_database(backup_file)
    print(f"🌐 全局数据库已恢复")

    return result.wasSuccessful()

if __name__ == "__main__":
    success = run_all_tests()
    exit(0 if success else 1)

之后直接运行python test_suite.py就能一次性执行所有测试，且只做一次备份恢复。

方案3：模块级初始化（每个文件一次备份恢复） #

如果同一个测试文件里的多个测试类可以共享同一个数据库状态，可以用unittest的setUpModule和tearDownModule函数，在模块（文件）级别执行备份恢复：

import unittest
from your_backup_utils import backup_database, restore_database

# 模块级变量存储备份路径
_backup_file = None

def setUpModule():
    """当前测试文件所有测试执行前，备份数据库"""
    global _backup_file
    _backup_file = backup_database()
    print(f"📂 当前模块已备份数据库到: {_backup_file}")

def tearDownModule():
    """当前测试文件所有测试执行后，恢复数据库"""
    restore_database(_backup_file)
    print(f"📂 当前模块已恢复数据库")

class UserLoginTest(unittest.TestCase):
    def test_normal_login(self):
        pass

class UserLogoutTest(unittest.TestCase):
    def test_logout(self):
        pass

这个方案介于方案1和方案2之间，适合按业务模块拆分测试文件的场景。

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