如何在Python中单个测试的持续时间内替换类变量？

Question

我正在尝试测试一个缓存类,其中应该自动删除旧项.缓存可以容纳的项目数量由"常量"类变量控制,如下所示:

class Cache(dict):
    MAX_ITEMS = 100

出于测试目的,我想在单个测试用例的持续时间内更改该常量.使用unittest.mock.patch我可以用一个替换类变量,MagicMock但这不是我想要做的.我只需要用不同的号码替换它.

有没有一种优雅的方法在Python(像patch)中执行此操作,或者我应该自己重置变量tearDown？

---

_{我正在使用Python 3.4.3}

Answer 1

补丁可以做到这一点.尝试:

from unittest.mock import patch

class A:
    val = 1

assert A.val == 1
with patch.object(A, "val", "patched_value"):
    assert A.val == "patched_value"

assert A.val == 1

当然你也可以patch.object用作装饰师.

Answer 2

更仔细地阅读文档，结果patch实际上是有一个可选参数来控制变量所要替换的内容（而不是模拟对象），因此我可以这样做：

class CacheTest(TestCase):
    @patch("myproject.Cache.MAX_ITEMS", new=3)
    def testCacheTrim(self):
        # ...

或者，正如Dunes指出的那样：

class CacheTest(TestCase):
    @patch.object(Cache, "MAX_ITEMS", new=3)
    def testCacheTrim(self):
        # ...

可以说看起来更漂亮。