feat: support to provide extra cache while using real_cache in tests

flamingo-run · Feb 28, 2024 · 4cb2e4e · 4cb2e4e
1 parent ba64802
commit 4cb2e4e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/drf_kit/tests.py b/drf_kit/tests.py
@@ -24,8 +24,10 @@ def setUp(self):
         super().setUp()
         cache.clear()
 
-    def real_cache(self):
-        return self.settings(CACHES={"default": {"BACKEND": "django.core.cache.backends.locmem.LocMemCache"}})
+    def real_cache(self, caches: dict | None = None):
+        if not caches:
+            caches = {}
+        return self.settings(CACHES={"default": {"BACKEND": "django.core.cache.backends.locmem.LocMemCache"}} | caches)
 
     def assertNoPendingMigration(self, app_name):  # pylint: disable=invalid-name
         out = StringIO()