python - unittest.mock.补丁 : Context manager vs setUp/tearDown in unittest

Question

似乎有两种使用方式 unittest.mock.patch : 有更好的方法吗？

使用上下文管理器和 with 语句:

class MyTest(TestCase):
    def test_something(self):
        with patch('package.module.Class') as MockClass:
            assert package.module.Class is MockClass

或者从设置和拆卸/清理中调用启动和停止:

class MyTest(TestCase):
    def setUp(self):
        patcher = patch('package.module.Class')
        self.MockClass = patcher.start()
        self.addCleanup(patcher.stop)

    def test_something(self):
        assert package.module.Class is self.MockClass

context manager 版本的代码更少，因此更易于阅读。我有什么理由更愿意使用 TestCase setUp/tearDown 基础架构？

Answer 1

喜欢在 setUp 中打补丁的主要原因是，如果您有多个测试需要打补丁。在这种情况下，您需要在每个测试中复制 with 语句。

如果您只有一个测试需要补丁，我更喜欢 with 语句以提高可读性。