mirror of https://github.com/django/django.git
[5.1.x] Fixed #32831 -– Allowed cache tests to be retried via a new "retry" decorator.
Backport of 957c54d945
from main.
This commit is contained in:
parent
1b7b5e0d17
commit
33cd593420
|
@ -11,6 +11,7 @@ import tempfile
|
|||
import threading
|
||||
import time
|
||||
import unittest
|
||||
from functools import wraps
|
||||
from pathlib import Path
|
||||
from unittest import mock, skipIf
|
||||
|
||||
|
@ -89,6 +90,25 @@ KEY_ERRORS_WITH_MEMCACHED_MSG = (
|
|||
)
|
||||
|
||||
|
||||
def retry(retries=3, delay=1):
|
||||
def decorator(func):
|
||||
@wraps(func)
|
||||
def wrapper(*args, **kwargs):
|
||||
attempts = 0
|
||||
while attempts < retries:
|
||||
try:
|
||||
return func(*args, **kwargs)
|
||||
except AssertionError:
|
||||
attempts += 1
|
||||
if attempts >= retries:
|
||||
raise
|
||||
time.sleep(delay)
|
||||
|
||||
return wrapper
|
||||
|
||||
return decorator
|
||||
|
||||
|
||||
@override_settings(
|
||||
CACHES={
|
||||
"default": {
|
||||
|
@ -489,6 +509,7 @@ class BaseCacheTests:
|
|||
self.assertEqual(cache.get("expire2"), "newvalue")
|
||||
self.assertIs(cache.has_key("expire3"), False)
|
||||
|
||||
@retry()
|
||||
def test_touch(self):
|
||||
# cache.touch() updates the timeout.
|
||||
cache.set("expire1", "very quickly", timeout=1)
|
||||
|
@ -616,6 +637,7 @@ class BaseCacheTests:
|
|||
self.assertEqual(cache.get("key3"), "sausage")
|
||||
self.assertEqual(cache.get("key4"), "lobster bisque")
|
||||
|
||||
@retry()
|
||||
def test_forever_timeout(self):
|
||||
"""
|
||||
Passing in None into timeout results in a value that is cached forever
|
||||
|
@ -1397,6 +1419,7 @@ class LocMemCacheTests(BaseCacheTests, TestCase):
|
|||
self.assertEqual(cache.decr(key), 1)
|
||||
self.assertEqual(expire, cache._expire_info[_key])
|
||||
|
||||
@retry()
|
||||
@limit_locmem_entries
|
||||
def test_lru_get(self):
|
||||
"""get() moves cache keys."""
|
||||
|
@ -1424,6 +1447,7 @@ class LocMemCacheTests(BaseCacheTests, TestCase):
|
|||
for key in range(3):
|
||||
self.assertIsNone(cache.get(key))
|
||||
|
||||
@retry()
|
||||
@limit_locmem_entries
|
||||
def test_lru_incr(self):
|
||||
"""incr() moves cache keys."""
|
||||
|
@ -2674,6 +2698,7 @@ class CacheMiddlewareTest(SimpleTestCase):
|
|||
response = other_with_prefix_view(request, "16")
|
||||
self.assertEqual(response.content, b"Hello World 16")
|
||||
|
||||
@retry()
|
||||
def test_cache_page_timeout(self):
|
||||
# Page timeout takes precedence over the "max-age" section of the
|
||||
# "Cache-Control".
|
||||
|
|
Loading…
Reference in New Issue