如何在已配置类级mock的测试类中覆盖单个测试的mock

类级@mock.patch("foo", bar)会将mock逻辑应用到类内所有测试方法上，此时直接给单个方法加@mock.patch("foo", baz)无法生效——因为类级patch的作用域更外层，会覆盖方法级的patch效果。以下是几种更优的实现方式，无需为每个方法重复添加装饰器：

方法1：在测试方法内使用上下文管理器（推荐） #

无需改动现有类级装饰器，仅在需要特殊mock的方法内临时替换目标对象，作用域仅限于当前方法的代码块：

from unittest import mock

@mock.patch("foo", bar)
class TestMyThing:
    def test_A(self):
        assert foo is bar  # 类级mock生效

    def test_D(self):
        # 临时替换为baz，仅在with块内有效
        with mock.patch("foo", baz):
            assert foo is baz  # 此处使用baz
        # 退出块后自动恢复为类级的bar
        assert foo is bar

优点：代码改动极小，不破坏原有测试结构，完全遵循DRY原则，仅针对特殊场景做局部修改。

方法2：拆分测试类+继承复用（适合多场景mock） #

如果存在多个测试方法需要不同的mock值，可以将共享测试逻辑抽离到基础类，再创建不同子类分别应用类级mock：

from unittest import mock

# 基础类：存放所有共享测试逻辑，不带mock装饰器
class BaseTestMyThing:
    def test_A(self):
        assert foo is self.expected_mock

    def test_B(self):
        assert foo is self.expected_mock

# 应用bar作为mock的测试类
@mock.patch("foo", bar)
class TestMyThingWithBar(BaseTestMyThing):
    def setUp(self):
        self.expected_mock = bar

# 应用baz作为mock的测试类，新增test_D
@mock.patch("foo", baz)
class TestMyThingWithBaz(BaseTestMyThing):
    def setUp(self):
        self.expected_mock = baz

    def test_D(self):
        assert foo is baz

优点：逻辑清晰，符合开闭原则——新增mock场景只需新增子类，无需修改原有测试代码；复用共享测试逻辑，避免重复。

方法3：手动管理mock生命周期（不推荐） #

通过手动启停mock来切换值，但这种方式容易因遗漏恢复步骤影响其他测试，仅作为备选：

from unittest import mock

@mock.patch("foo", bar)
class TestMyThing:
    def test_A(self):
        assert foo is bar

    def setUp(self):
        # 保存原始对象引用
        self.original_foo = __import__("__main__").foo

    def test_D(self):
        # 停止所有现有mock
        mock.patch.stopall()
        # 启动新的baz mock
        baz_patch = mock.patch("foo", baz)
        baz_patch.start()
        try:
            assert foo is baz
        finally:
            # 恢复类级的bar mock
            baz_patch.stop()
            mock.patch("foo", bar).start()

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