使用 LocMemCache 进行选择性 Django pytest

Question

我有一个基于 Django REST 框架 SimpleRateThrottle 的自定义 Throttling 类，我想用 pytest 测试我的自定义类。由于我的默认测试设置使用 DummyCache，我想为这个特定的测试模块切换到 LocMemCache（SimpleRateThrottle 使用缓存后端来跟踪计数）。有没有办法为选择性测试切换缓存后端？在夹具中设置 settings.CACHE 似乎不起作用。我还尝试在 SimpleRateThrottle 中模拟 default_cache，但我做错了。

naive_throttler.py

from rest_framework.throttling import SimpleRateThrottle

class NaiveThrottler(SimpleRateThrottle):
   ...

rest_framework/throttling.py

from django.core.cache import cache as default_cache  # Or how can I patch this?

class SimpleRateThrottle(BaseThrottle):
...

Answer 1

虽然 Django 提供的函数/装饰器可以工作，但pytest-django 提供了fixture for changing settings for a test。为了更好地遵循pytest 的using fixtures for tests 范例，最好将特定于测试的设置更改如下：

import pytest

def test_cache(settings):
    settings.CACHES = {
        'default': {
            'BACKEND': 'django.core.cache.backends.dummy.DummyCache',
        }
    }
    # test logic here

Answer 2

Django 为此提供了override_settings and modify_settings 装饰器。如果您只想为一项测试更改 CACHES 设置，您可以这样做：

from django.test import TestCase, override_settings

class MyTestCase(TestCase):

    @override_settings(CACHES = {
                           'default': {
                              'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
                           }
                      })
    def test_chache(self):
        # your test code