运行在 Python 中使用缓存装饰器进行单元测试

Question

所以我正在开发一个应用程序，在导入某些记录时，需要重新计算某些字段。为了防止每次检查都读取数据库，有一个缓存装饰器，因此在导入过程中每 n 秒只执行一次数据库读取。麻烦来自构建测试用例。以下确实有效，但它有一个丑陋的睡眠。

# The decorator I need to patch

@cache_function_call(2.0)
def _latest_term_modified():
    return PrimaryTerm.objects.latest('object_modified').object_modified


# The 2.0 sets the TTL of the decorator. So I need to switch out 
# self.ttl for this decorated function before 
# this test. Right now I'm just using a sleep, which works

    @mock.patch.object(models.Student, 'update_anniversary')
    def test_import_on_term_update(self, mock_update):
        self._import_student()
        latest_term = self._latest_term_mod()
        latest_term.save()
        time.sleep(3)
        self._import_student()
        self.assertEqual(mock_update.call_count, 2)

装饰器本身如下所示：

class cache_function_call(object):
    """Cache an argument-less function call for 'ttl' seconds."""
    def __init__(self, ttl):
        self.cached_result = None
        self.timestamp = 0
        self.ttl = ttl

    def __call__(self, func):
        @wraps(func)
        def inner():
            now = time.time()
            if now > self.timestamp + self.ttl:
                self.cached_result = func()
                self.timestamp = now
            return self.cached_result
        return inner

我尝试在导入模型之前设置装饰器：

decorators.cache_function_call = lambda x : x
import models

但即使在文件的顶部，django 仍然会在运行我的 tests.py 之前初始化模型并且该函数仍然使用缓存装饰器而不是我的 lambda/noop 装饰器进行装饰.

编写此测试的最佳方法是什么，这样我就睡不着了。我可以以某种方式在运行我的导入之前设置装饰器的 ttl 吗？

Answer 1

您可以稍微更改装饰器 class。

在 decorators.py 中的模块级别设置全局

BAILOUT = False

并在您的装饰器 class 中，更改：

def __call__(self, func):
    @wraps(func)
    def inner():
        now = time.time()
        if BAILOUT or now > self.timestamp + self.ttl:
            self.cached_result = func()
            self.timestamp = now
        return self.cached_result
    return inner

然后在您的测试集 decorators.BAILOUT = True 中，嘿，太棒了！-)

运行在 Python 中使用缓存装饰器进行单元测试

Running unit tests in Python with a caching decorator

python

django

caching

unit-testing

django-models

运行 在 Python 中使用缓存装饰器进行单元测试

Running unit tests in Python with a caching decorator

python

django

caching

unit-testing

django-models

运行在 Python 中使用缓存装饰器进行单元测试