如何在Python unittest中编写可复用实例的有状态测试？

你遇到的核心问题是unittest会为每个测试方法创建独立的测试类实例，导致self属性无法跨方法共享。下面是几种在unittest框架内无需额外依赖的解决方式：

1. 使用类级别的fixture（setUpClass） #

这是最常用的方式，通过setUpClass类方法在整个测试类执行前一次性创建所有高成本实例，存储为类属性，所有测试方法共享这些实例。

示例代码：

import unittest

# 模拟你的类层级
class E:
    def __init__(self):
        # 模拟高成本初始化逻辑
        print("初始化E（高成本操作）")

class D:
    def __init__(self):
        print("初始化D（高成本操作）")

class C:
    def __init__(self, d, e):
        self.d = d
        self.e = e
        print("初始化C")

class B:
    def __init__(self):
        print("初始化B（高成本操作）")

class A:
    def __init__(self, b, c):
        self.b = b
        self.c = c
        print("初始化A")

class TestsClass(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        # 一次性创建所有需要复用的实例，存入类属性
        cls.e = E()
        cls.d = D()
        cls.c = C(cls.d, cls.e)
        cls.b = B()
        cls.a = A(cls.b, cls.c)

    def test_e_functionality(self):
        # 通过类名访问共享实例
        self.assertIsNotNone(self.__class__.e)
        # 编写E的测试逻辑
        pass

    def test_d_functionality(self):
        self.assertIsNotNone(self.__class__.d)
        # 编写D的测试逻辑
        pass

    def test_c_functionality(self):
        c_instance = self.__class__.c
        self.assertIs(c_instance.d, self.__class__.d)
        self.assertIs(c_instance.e, self.__class__.e)
        # 编写C的测试逻辑
        pass

    def test_a_functionality(self):
        a_instance = self.__class__.a
        self.assertIs(a_instance.b, self.__class__.b)
        self.assertIs(a_instance.c, self.__class__.c)
        # 编写A的测试逻辑
        pass

if __name__ == '__main__':
    unittest.main()

注意：如果测试方法会修改共享实例的内部状态，可能会导致测试用例之间互相干扰。因此建议确保测试用例是无状态的，或者实例本身是不可变的。

2. 使用模块级别的fixture（setUpModule） #

如果多个测试类需要复用这些高成本实例，可以使用模块级别的初始化方法，在整个模块执行前创建实例，存储为模块变量。

示例代码：

import unittest

# 模块级变量用于存储共享实例
e = None
d = None
c = None
b = None
a = None

def setUpModule():
    """模块初始化时执行，只运行一次"""
    global e, d, c, b, a
    e = E()
    d = D()
    c = C(d, e)
    b = B()
    a = A(b, c)

def tearDownModule():
    """模块结束时执行，用于清理资源"""
    global e, d, c, b, a
    e = None
    d = None
    c = None
    b = None
    a = None

class TestsClass(unittest.TestCase):
    def test_e(self):
        self.assertIsNotNone(e)
        # 测试E的逻辑
        pass

class AnotherTestClass(unittest.TestCase):
    def test_a(self):
        self.assertIsNotNone(a)
        # 测试A的逻辑
        pass

这种方式适合跨测试类复用实例，同样需要注意实例状态的隔离问题。

3. 避免依赖测试执行顺序 #

你原本想通过命名让测试按顺序执行并传递实例，但unittest官方不推荐依赖测试方法的执行顺序（虽然命名为test_00_*这类形式会按字典序执行）。使用上述fixture方式可以完全摆脱对执行顺序的依赖，所有实例在测试开始前就已准备好，测试方法可以独立验证对应类的功能。

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