tests.py 62 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408
  1. # -*- coding: utf-8 -*-
  2. # Unit tests for cache framework
  3. # Uses whatever cache backend is set in the test settings file.
  4. import os
  5. import tempfile
  6. import time
  7. import warnings
  8. from django.conf import settings
  9. from django.core import management
  10. from django.core.cache import get_cache, DEFAULT_CACHE_ALIAS
  11. from django.core.cache.backends.base import CacheKeyWarning
  12. from django.http import HttpResponse, HttpRequest, QueryDict
  13. from django.middleware.cache import FetchFromCacheMiddleware, UpdateCacheMiddleware, CacheMiddleware
  14. from django.test import RequestFactory
  15. from django.test.utils import get_warnings_state, restore_warnings_state
  16. from django.utils import translation
  17. from django.utils import unittest
  18. from django.utils.cache import patch_vary_headers, get_cache_key, learn_cache_key
  19. from django.utils.hashcompat import md5_constructor
  20. from django.views.decorators.cache import cache_page
  21. from regressiontests.cache.models import Poll, expensive_calculation
  22. # functions/classes for complex data type tests
  23. def f():
  24. return 42
  25. class C:
  26. def m(n):
  27. return 24
  28. class DummyCacheTests(unittest.TestCase):
  29. # The Dummy cache backend doesn't really behave like a test backend,
  30. # so it has different test requirements.
  31. def setUp(self):
  32. self.cache = get_cache('django.core.cache.backends.dummy.DummyCache')
  33. def test_simple(self):
  34. "Dummy cache backend ignores cache set calls"
  35. self.cache.set("key", "value")
  36. self.assertEqual(self.cache.get("key"), None)
  37. def test_add(self):
  38. "Add doesn't do anything in dummy cache backend"
  39. self.cache.add("addkey1", "value")
  40. result = self.cache.add("addkey1", "newvalue")
  41. self.assertEqual(result, True)
  42. self.assertEqual(self.cache.get("addkey1"), None)
  43. def test_non_existent(self):
  44. "Non-existent keys aren't found in the dummy cache backend"
  45. self.assertEqual(self.cache.get("does_not_exist"), None)
  46. self.assertEqual(self.cache.get("does_not_exist", "bang!"), "bang!")
  47. def test_get_many(self):
  48. "get_many returns nothing for the dummy cache backend"
  49. self.cache.set('a', 'a')
  50. self.cache.set('b', 'b')
  51. self.cache.set('c', 'c')
  52. self.cache.set('d', 'd')
  53. self.assertEqual(self.cache.get_many(['a', 'c', 'd']), {})
  54. self.assertEqual(self.cache.get_many(['a', 'b', 'e']), {})
  55. def test_delete(self):
  56. "Cache deletion is transparently ignored on the dummy cache backend"
  57. self.cache.set("key1", "spam")
  58. self.cache.set("key2", "eggs")
  59. self.assertEqual(self.cache.get("key1"), None)
  60. self.cache.delete("key1")
  61. self.assertEqual(self.cache.get("key1"), None)
  62. self.assertEqual(self.cache.get("key2"), None)
  63. def test_has_key(self):
  64. "The has_key method doesn't ever return True for the dummy cache backend"
  65. self.cache.set("hello1", "goodbye1")
  66. self.assertEqual(self.cache.has_key("hello1"), False)
  67. self.assertEqual(self.cache.has_key("goodbye1"), False)
  68. def test_in(self):
  69. "The in operator doesn't ever return True for the dummy cache backend"
  70. self.cache.set("hello2", "goodbye2")
  71. self.assertEqual("hello2" in self.cache, False)
  72. self.assertEqual("goodbye2" in self.cache, False)
  73. def test_incr(self):
  74. "Dummy cache values can't be incremented"
  75. self.cache.set('answer', 42)
  76. self.assertRaises(ValueError, self.cache.incr, 'answer')
  77. self.assertRaises(ValueError, self.cache.incr, 'does_not_exist')
  78. def test_decr(self):
  79. "Dummy cache values can't be decremented"
  80. self.cache.set('answer', 42)
  81. self.assertRaises(ValueError, self.cache.decr, 'answer')
  82. self.assertRaises(ValueError, self.cache.decr, 'does_not_exist')
  83. def test_data_types(self):
  84. "All data types are ignored equally by the dummy cache"
  85. stuff = {
  86. 'string' : 'this is a string',
  87. 'int' : 42,
  88. 'list' : [1, 2, 3, 4],
  89. 'tuple' : (1, 2, 3, 4),
  90. 'dict' : {'A': 1, 'B' : 2},
  91. 'function' : f,
  92. 'class' : C,
  93. }
  94. self.cache.set("stuff", stuff)
  95. self.assertEqual(self.cache.get("stuff"), None)
  96. def test_expiration(self):
  97. "Expiration has no effect on the dummy cache"
  98. self.cache.set('expire1', 'very quickly', 1)
  99. self.cache.set('expire2', 'very quickly', 1)
  100. self.cache.set('expire3', 'very quickly', 1)
  101. time.sleep(2)
  102. self.assertEqual(self.cache.get("expire1"), None)
  103. self.cache.add("expire2", "newvalue")
  104. self.assertEqual(self.cache.get("expire2"), None)
  105. self.assertEqual(self.cache.has_key("expire3"), False)
  106. def test_unicode(self):
  107. "Unicode values are ignored by the dummy cache"
  108. stuff = {
  109. u'ascii': u'ascii_value',
  110. u'unicode_ascii': u'Iñtërnâtiônàlizætiøn1',
  111. u'Iñtërnâtiônàlizætiøn': u'Iñtërnâtiônàlizætiøn2',
  112. u'ascii': {u'x' : 1 }
  113. }
  114. for (key, value) in stuff.items():
  115. self.cache.set(key, value)
  116. self.assertEqual(self.cache.get(key), None)
  117. def test_set_many(self):
  118. "set_many does nothing for the dummy cache backend"
  119. self.cache.set_many({'a': 1, 'b': 2})
  120. def test_delete_many(self):
  121. "delete_many does nothing for the dummy cache backend"
  122. self.cache.delete_many(['a', 'b'])
  123. def test_clear(self):
  124. "clear does nothing for the dummy cache backend"
  125. self.cache.clear()
  126. def test_incr_version(self):
  127. "Dummy cache versions can't be incremented"
  128. self.cache.set('answer', 42)
  129. self.assertRaises(ValueError, self.cache.incr_version, 'answer')
  130. self.assertRaises(ValueError, self.cache.incr_version, 'does_not_exist')
  131. def test_decr_version(self):
  132. "Dummy cache versions can't be decremented"
  133. self.cache.set('answer', 42)
  134. self.assertRaises(ValueError, self.cache.decr_version, 'answer')
  135. self.assertRaises(ValueError, self.cache.decr_version, 'does_not_exist')
  136. class BaseCacheTests(object):
  137. # A common set of tests to apply to all cache backends
  138. def test_simple(self):
  139. # Simple cache set/get works
  140. self.cache.set("key", "value")
  141. self.assertEqual(self.cache.get("key"), "value")
  142. def test_add(self):
  143. # A key can be added to a cache
  144. self.cache.add("addkey1", "value")
  145. result = self.cache.add("addkey1", "newvalue")
  146. self.assertEqual(result, False)
  147. self.assertEqual(self.cache.get("addkey1"), "value")
  148. def test_prefix(self):
  149. # Test for same cache key conflicts between shared backend
  150. self.cache.set('somekey', 'value')
  151. # should not be set in the prefixed cache
  152. self.assertFalse(self.prefix_cache.has_key('somekey'))
  153. self.prefix_cache.set('somekey', 'value2')
  154. self.assertEqual(self.cache.get('somekey'), 'value')
  155. self.assertEqual(self.prefix_cache.get('somekey'), 'value2')
  156. def test_non_existent(self):
  157. # Non-existent cache keys return as None/default
  158. # get with non-existent keys
  159. self.assertEqual(self.cache.get("does_not_exist"), None)
  160. self.assertEqual(self.cache.get("does_not_exist", "bang!"), "bang!")
  161. def test_get_many(self):
  162. # Multiple cache keys can be returned using get_many
  163. self.cache.set('a', 'a')
  164. self.cache.set('b', 'b')
  165. self.cache.set('c', 'c')
  166. self.cache.set('d', 'd')
  167. self.assertEqual(self.cache.get_many(['a', 'c', 'd']), {'a' : 'a', 'c' : 'c', 'd' : 'd'})
  168. self.assertEqual(self.cache.get_many(['a', 'b', 'e']), {'a' : 'a', 'b' : 'b'})
  169. def test_delete(self):
  170. # Cache keys can be deleted
  171. self.cache.set("key1", "spam")
  172. self.cache.set("key2", "eggs")
  173. self.assertEqual(self.cache.get("key1"), "spam")
  174. self.cache.delete("key1")
  175. self.assertEqual(self.cache.get("key1"), None)
  176. self.assertEqual(self.cache.get("key2"), "eggs")
  177. def test_has_key(self):
  178. # The cache can be inspected for cache keys
  179. self.cache.set("hello1", "goodbye1")
  180. self.assertEqual(self.cache.has_key("hello1"), True)
  181. self.assertEqual(self.cache.has_key("goodbye1"), False)
  182. def test_in(self):
  183. # The in operator can be used to inspet cache contents
  184. self.cache.set("hello2", "goodbye2")
  185. self.assertEqual("hello2" in self.cache, True)
  186. self.assertEqual("goodbye2" in self.cache, False)
  187. def test_incr(self):
  188. # Cache values can be incremented
  189. self.cache.set('answer', 41)
  190. self.assertEqual(self.cache.incr('answer'), 42)
  191. self.assertEqual(self.cache.get('answer'), 42)
  192. self.assertEqual(self.cache.incr('answer', 10), 52)
  193. self.assertEqual(self.cache.get('answer'), 52)
  194. self.assertRaises(ValueError, self.cache.incr, 'does_not_exist')
  195. def test_decr(self):
  196. # Cache values can be decremented
  197. self.cache.set('answer', 43)
  198. self.assertEqual(self.cache.decr('answer'), 42)
  199. self.assertEqual(self.cache.get('answer'), 42)
  200. self.assertEqual(self.cache.decr('answer', 10), 32)
  201. self.assertEqual(self.cache.get('answer'), 32)
  202. self.assertRaises(ValueError, self.cache.decr, 'does_not_exist')
  203. def test_data_types(self):
  204. # Many different data types can be cached
  205. stuff = {
  206. 'string' : 'this is a string',
  207. 'int' : 42,
  208. 'list' : [1, 2, 3, 4],
  209. 'tuple' : (1, 2, 3, 4),
  210. 'dict' : {'A': 1, 'B' : 2},
  211. 'function' : f,
  212. 'class' : C,
  213. }
  214. self.cache.set("stuff", stuff)
  215. self.assertEqual(self.cache.get("stuff"), stuff)
  216. def test_cache_read_for_model_instance(self):
  217. # Don't want fields with callable as default to be called on cache read
  218. expensive_calculation.num_runs = 0
  219. Poll.objects.all().delete()
  220. my_poll = Poll.objects.create(question="Well?")
  221. self.assertEqual(Poll.objects.count(), 1)
  222. pub_date = my_poll.pub_date
  223. self.cache.set('question', my_poll)
  224. cached_poll = self.cache.get('question')
  225. self.assertEqual(cached_poll.pub_date, pub_date)
  226. # We only want the default expensive calculation run once
  227. self.assertEqual(expensive_calculation.num_runs, 1)
  228. def test_cache_write_for_model_instance_with_deferred(self):
  229. # Don't want fields with callable as default to be called on cache write
  230. expensive_calculation.num_runs = 0
  231. Poll.objects.all().delete()
  232. my_poll = Poll.objects.create(question="What?")
  233. self.assertEqual(expensive_calculation.num_runs, 1)
  234. defer_qs = Poll.objects.all().defer('question')
  235. self.assertEqual(defer_qs.count(), 1)
  236. self.assertEqual(expensive_calculation.num_runs, 1)
  237. self.cache.set('deferred_queryset', defer_qs)
  238. # cache set should not re-evaluate default functions
  239. self.assertEqual(expensive_calculation.num_runs, 1)
  240. def test_cache_read_for_model_instance_with_deferred(self):
  241. # Don't want fields with callable as default to be called on cache read
  242. expensive_calculation.num_runs = 0
  243. Poll.objects.all().delete()
  244. my_poll = Poll.objects.create(question="What?")
  245. self.assertEqual(expensive_calculation.num_runs, 1)
  246. defer_qs = Poll.objects.all().defer('question')
  247. self.assertEqual(defer_qs.count(), 1)
  248. self.cache.set('deferred_queryset', defer_qs)
  249. self.assertEqual(expensive_calculation.num_runs, 1)
  250. runs_before_cache_read = expensive_calculation.num_runs
  251. cached_polls = self.cache.get('deferred_queryset')
  252. # We only want the default expensive calculation run on creation and set
  253. self.assertEqual(expensive_calculation.num_runs, runs_before_cache_read)
  254. def test_expiration(self):
  255. # Cache values can be set to expire
  256. self.cache.set('expire1', 'very quickly', 1)
  257. self.cache.set('expire2', 'very quickly', 1)
  258. self.cache.set('expire3', 'very quickly', 1)
  259. time.sleep(2)
  260. self.assertEqual(self.cache.get("expire1"), None)
  261. self.cache.add("expire2", "newvalue")
  262. self.assertEqual(self.cache.get("expire2"), "newvalue")
  263. self.assertEqual(self.cache.has_key("expire3"), False)
  264. def test_unicode(self):
  265. # Unicode values can be cached
  266. stuff = {
  267. u'ascii': u'ascii_value',
  268. u'unicode_ascii': u'Iñtërnâtiônàlizætiøn1',
  269. u'Iñtërnâtiônàlizætiøn': u'Iñtërnâtiônàlizætiøn2',
  270. u'ascii': {u'x' : 1 }
  271. }
  272. for (key, value) in stuff.items():
  273. self.cache.set(key, value)
  274. self.assertEqual(self.cache.get(key), value)
  275. def test_binary_string(self):
  276. # Binary strings should be cachable
  277. from zlib import compress, decompress
  278. value = 'value_to_be_compressed'
  279. compressed_value = compress(value)
  280. self.cache.set('binary1', compressed_value)
  281. compressed_result = self.cache.get('binary1')
  282. self.assertEqual(compressed_value, compressed_result)
  283. self.assertEqual(value, decompress(compressed_result))
  284. def test_set_many(self):
  285. # Multiple keys can be set using set_many
  286. self.cache.set_many({"key1": "spam", "key2": "eggs"})
  287. self.assertEqual(self.cache.get("key1"), "spam")
  288. self.assertEqual(self.cache.get("key2"), "eggs")
  289. def test_set_many_expiration(self):
  290. # set_many takes a second ``timeout`` parameter
  291. self.cache.set_many({"key1": "spam", "key2": "eggs"}, 1)
  292. time.sleep(2)
  293. self.assertEqual(self.cache.get("key1"), None)
  294. self.assertEqual(self.cache.get("key2"), None)
  295. def test_delete_many(self):
  296. # Multiple keys can be deleted using delete_many
  297. self.cache.set("key1", "spam")
  298. self.cache.set("key2", "eggs")
  299. self.cache.set("key3", "ham")
  300. self.cache.delete_many(["key1", "key2"])
  301. self.assertEqual(self.cache.get("key1"), None)
  302. self.assertEqual(self.cache.get("key2"), None)
  303. self.assertEqual(self.cache.get("key3"), "ham")
  304. def test_clear(self):
  305. # The cache can be emptied using clear
  306. self.cache.set("key1", "spam")
  307. self.cache.set("key2", "eggs")
  308. self.cache.clear()
  309. self.assertEqual(self.cache.get("key1"), None)
  310. self.assertEqual(self.cache.get("key2"), None)
  311. def test_long_timeout(self):
  312. '''
  313. Using a timeout greater than 30 days makes memcached think
  314. it is an absolute expiration timestamp instead of a relative
  315. offset. Test that we honour this convention. Refs #12399.
  316. '''
  317. self.cache.set('key1', 'eggs', 60*60*24*30 + 1) #30 days + 1 second
  318. self.assertEqual(self.cache.get('key1'), 'eggs')
  319. self.cache.add('key2', 'ham', 60*60*24*30 + 1)
  320. self.assertEqual(self.cache.get('key2'), 'ham')
  321. self.cache.set_many({'key3': 'sausage', 'key4': 'lobster bisque'}, 60*60*24*30 + 1)
  322. self.assertEqual(self.cache.get('key3'), 'sausage')
  323. self.assertEqual(self.cache.get('key4'), 'lobster bisque')
  324. def perform_cull_test(self, initial_count, final_count):
  325. """This is implemented as a utility method, because only some of the backends
  326. implement culling. The culling algorithm also varies slightly, so the final
  327. number of entries will vary between backends"""
  328. # Create initial cache key entries. This will overflow the cache, causing a cull
  329. for i in range(1, initial_count):
  330. self.cache.set('cull%d' % i, 'value', 1000)
  331. count = 0
  332. # Count how many keys are left in the cache.
  333. for i in range(1, initial_count):
  334. if self.cache.has_key('cull%d' % i):
  335. count = count + 1
  336. self.assertEqual(count, final_count)
  337. def test_invalid_keys(self):
  338. """
  339. All the builtin backends (except memcached, see below) should warn on
  340. keys that would be refused by memcached. This encourages portable
  341. caching code without making it too difficult to use production backends
  342. with more liberal key rules. Refs #6447.
  343. """
  344. # mimic custom ``make_key`` method being defined since the default will
  345. # never show the below warnings
  346. def func(key, *args):
  347. return key
  348. old_func = self.cache.key_func
  349. self.cache.key_func = func
  350. # On Python 2.6+ we could use the catch_warnings context
  351. # manager to test this warning nicely. Since we can't do that
  352. # yet, the cleanest option is to temporarily ask for
  353. # CacheKeyWarning to be raised as an exception.
  354. _warnings_state = get_warnings_state()
  355. warnings.simplefilter("error", CacheKeyWarning)
  356. try:
  357. # memcached does not allow whitespace or control characters in keys
  358. self.assertRaises(CacheKeyWarning, self.cache.set, 'key with spaces', 'value')
  359. # memcached limits key length to 250
  360. self.assertRaises(CacheKeyWarning, self.cache.set, 'a' * 251, 'value')
  361. finally:
  362. restore_warnings_state(_warnings_state)
  363. self.cache.key_func = old_func
  364. def test_cache_versioning_get_set(self):
  365. # set, using default version = 1
  366. self.cache.set('answer1', 42)
  367. self.assertEqual(self.cache.get('answer1'), 42)
  368. self.assertEqual(self.cache.get('answer1', version=1), 42)
  369. self.assertEqual(self.cache.get('answer1', version=2), None)
  370. self.assertEqual(self.v2_cache.get('answer1'), None)
  371. self.assertEqual(self.v2_cache.get('answer1', version=1), 42)
  372. self.assertEqual(self.v2_cache.get('answer1', version=2), None)
  373. # set, default version = 1, but manually override version = 2
  374. self.cache.set('answer2', 42, version=2)
  375. self.assertEqual(self.cache.get('answer2'), None)
  376. self.assertEqual(self.cache.get('answer2', version=1), None)
  377. self.assertEqual(self.cache.get('answer2', version=2), 42)
  378. self.assertEqual(self.v2_cache.get('answer2'), 42)
  379. self.assertEqual(self.v2_cache.get('answer2', version=1), None)
  380. self.assertEqual(self.v2_cache.get('answer2', version=2), 42)
  381. # v2 set, using default version = 2
  382. self.v2_cache.set('answer3', 42)
  383. self.assertEqual(self.cache.get('answer3'), None)
  384. self.assertEqual(self.cache.get('answer3', version=1), None)
  385. self.assertEqual(self.cache.get('answer3', version=2), 42)
  386. self.assertEqual(self.v2_cache.get('answer3'), 42)
  387. self.assertEqual(self.v2_cache.get('answer3', version=1), None)
  388. self.assertEqual(self.v2_cache.get('answer3', version=2), 42)
  389. # v2 set, default version = 2, but manually override version = 1
  390. self.v2_cache.set('answer4', 42, version=1)
  391. self.assertEqual(self.cache.get('answer4'), 42)
  392. self.assertEqual(self.cache.get('answer4', version=1), 42)
  393. self.assertEqual(self.cache.get('answer4', version=2), None)
  394. self.assertEqual(self.v2_cache.get('answer4'), None)
  395. self.assertEqual(self.v2_cache.get('answer4', version=1), 42)
  396. self.assertEqual(self.v2_cache.get('answer4', version=2), None)
  397. def test_cache_versioning_add(self):
  398. # add, default version = 1, but manually override version = 2
  399. self.cache.add('answer1', 42, version=2)
  400. self.assertEqual(self.cache.get('answer1', version=1), None)
  401. self.assertEqual(self.cache.get('answer1', version=2), 42)
  402. self.cache.add('answer1', 37, version=2)
  403. self.assertEqual(self.cache.get('answer1', version=1), None)
  404. self.assertEqual(self.cache.get('answer1', version=2), 42)
  405. self.cache.add('answer1', 37, version=1)
  406. self.assertEqual(self.cache.get('answer1', version=1), 37)
  407. self.assertEqual(self.cache.get('answer1', version=2), 42)
  408. # v2 add, using default version = 2
  409. self.v2_cache.add('answer2', 42)
  410. self.assertEqual(self.cache.get('answer2', version=1), None)
  411. self.assertEqual(self.cache.get('answer2', version=2), 42)
  412. self.v2_cache.add('answer2', 37)
  413. self.assertEqual(self.cache.get('answer2', version=1), None)
  414. self.assertEqual(self.cache.get('answer2', version=2), 42)
  415. self.v2_cache.add('answer2', 37, version=1)
  416. self.assertEqual(self.cache.get('answer2', version=1), 37)
  417. self.assertEqual(self.cache.get('answer2', version=2), 42)
  418. # v2 add, default version = 2, but manually override version = 1
  419. self.v2_cache.add('answer3', 42, version=1)
  420. self.assertEqual(self.cache.get('answer3', version=1), 42)
  421. self.assertEqual(self.cache.get('answer3', version=2), None)
  422. self.v2_cache.add('answer3', 37, version=1)
  423. self.assertEqual(self.cache.get('answer3', version=1), 42)
  424. self.assertEqual(self.cache.get('answer3', version=2), None)
  425. self.v2_cache.add('answer3', 37)
  426. self.assertEqual(self.cache.get('answer3', version=1), 42)
  427. self.assertEqual(self.cache.get('answer3', version=2), 37)
  428. def test_cache_versioning_has_key(self):
  429. self.cache.set('answer1', 42)
  430. # has_key
  431. self.assertTrue(self.cache.has_key('answer1'))
  432. self.assertTrue(self.cache.has_key('answer1', version=1))
  433. self.assertFalse(self.cache.has_key('answer1', version=2))
  434. self.assertFalse(self.v2_cache.has_key('answer1'))
  435. self.assertTrue(self.v2_cache.has_key('answer1', version=1))
  436. self.assertFalse(self.v2_cache.has_key('answer1', version=2))
  437. def test_cache_versioning_delete(self):
  438. self.cache.set('answer1', 37, version=1)
  439. self.cache.set('answer1', 42, version=2)
  440. self.cache.delete('answer1')
  441. self.assertEqual(self.cache.get('answer1', version=1), None)
  442. self.assertEqual(self.cache.get('answer1', version=2), 42)
  443. self.cache.set('answer2', 37, version=1)
  444. self.cache.set('answer2', 42, version=2)
  445. self.cache.delete('answer2', version=2)
  446. self.assertEqual(self.cache.get('answer2', version=1), 37)
  447. self.assertEqual(self.cache.get('answer2', version=2), None)
  448. self.cache.set('answer3', 37, version=1)
  449. self.cache.set('answer3', 42, version=2)
  450. self.v2_cache.delete('answer3')
  451. self.assertEqual(self.cache.get('answer3', version=1), 37)
  452. self.assertEqual(self.cache.get('answer3', version=2), None)
  453. self.cache.set('answer4', 37, version=1)
  454. self.cache.set('answer4', 42, version=2)
  455. self.v2_cache.delete('answer4', version=1)
  456. self.assertEqual(self.cache.get('answer4', version=1), None)
  457. self.assertEqual(self.cache.get('answer4', version=2), 42)
  458. def test_cache_versioning_incr_decr(self):
  459. self.cache.set('answer1', 37, version=1)
  460. self.cache.set('answer1', 42, version=2)
  461. self.cache.incr('answer1')
  462. self.assertEqual(self.cache.get('answer1', version=1), 38)
  463. self.assertEqual(self.cache.get('answer1', version=2), 42)
  464. self.cache.decr('answer1')
  465. self.assertEqual(self.cache.get('answer1', version=1), 37)
  466. self.assertEqual(self.cache.get('answer1', version=2), 42)
  467. self.cache.set('answer2', 37, version=1)
  468. self.cache.set('answer2', 42, version=2)
  469. self.cache.incr('answer2', version=2)
  470. self.assertEqual(self.cache.get('answer2', version=1), 37)
  471. self.assertEqual(self.cache.get('answer2', version=2), 43)
  472. self.cache.decr('answer2', version=2)
  473. self.assertEqual(self.cache.get('answer2', version=1), 37)
  474. self.assertEqual(self.cache.get('answer2', version=2), 42)
  475. self.cache.set('answer3', 37, version=1)
  476. self.cache.set('answer3', 42, version=2)
  477. self.v2_cache.incr('answer3')
  478. self.assertEqual(self.cache.get('answer3', version=1), 37)
  479. self.assertEqual(self.cache.get('answer3', version=2), 43)
  480. self.v2_cache.decr('answer3')
  481. self.assertEqual(self.cache.get('answer3', version=1), 37)
  482. self.assertEqual(self.cache.get('answer3', version=2), 42)
  483. self.cache.set('answer4', 37, version=1)
  484. self.cache.set('answer4', 42, version=2)
  485. self.v2_cache.incr('answer4', version=1)
  486. self.assertEqual(self.cache.get('answer4', version=1), 38)
  487. self.assertEqual(self.cache.get('answer4', version=2), 42)
  488. self.v2_cache.decr('answer4', version=1)
  489. self.assertEqual(self.cache.get('answer4', version=1), 37)
  490. self.assertEqual(self.cache.get('answer4', version=2), 42)
  491. def test_cache_versioning_get_set_many(self):
  492. # set, using default version = 1
  493. self.cache.set_many({'ford1': 37, 'arthur1': 42})
  494. self.assertEqual(self.cache.get_many(['ford1','arthur1']),
  495. {'ford1': 37, 'arthur1': 42})
  496. self.assertEqual(self.cache.get_many(['ford1','arthur1'], version=1),
  497. {'ford1': 37, 'arthur1': 42})
  498. self.assertEqual(self.cache.get_many(['ford1','arthur1'], version=2), {})
  499. self.assertEqual(self.v2_cache.get_many(['ford1','arthur1']), {})
  500. self.assertEqual(self.v2_cache.get_many(['ford1','arthur1'], version=1),
  501. {'ford1': 37, 'arthur1': 42})
  502. self.assertEqual(self.v2_cache.get_many(['ford1','arthur1'], version=2), {})
  503. # set, default version = 1, but manually override version = 2
  504. self.cache.set_many({'ford2': 37, 'arthur2': 42}, version=2)
  505. self.assertEqual(self.cache.get_many(['ford2','arthur2']), {})
  506. self.assertEqual(self.cache.get_many(['ford2','arthur2'], version=1), {})
  507. self.assertEqual(self.cache.get_many(['ford2','arthur2'], version=2),
  508. {'ford2': 37, 'arthur2': 42})
  509. self.assertEqual(self.v2_cache.get_many(['ford2','arthur2']),
  510. {'ford2': 37, 'arthur2': 42})
  511. self.assertEqual(self.v2_cache.get_many(['ford2','arthur2'], version=1), {})
  512. self.assertEqual(self.v2_cache.get_many(['ford2','arthur2'], version=2),
  513. {'ford2': 37, 'arthur2': 42})
  514. # v2 set, using default version = 2
  515. self.v2_cache.set_many({'ford3': 37, 'arthur3': 42})
  516. self.assertEqual(self.cache.get_many(['ford3','arthur3']), {})
  517. self.assertEqual(self.cache.get_many(['ford3','arthur3'], version=1), {})
  518. self.assertEqual(self.cache.get_many(['ford3','arthur3'], version=2),
  519. {'ford3': 37, 'arthur3': 42})
  520. self.assertEqual(self.v2_cache.get_many(['ford3','arthur3']),
  521. {'ford3': 37, 'arthur3': 42})
  522. self.assertEqual(self.v2_cache.get_many(['ford3','arthur3'], version=1), {})
  523. self.assertEqual(self.v2_cache.get_many(['ford3','arthur3'], version=2),
  524. {'ford3': 37, 'arthur3': 42})
  525. # v2 set, default version = 2, but manually override version = 1
  526. self.v2_cache.set_many({'ford4': 37, 'arthur4': 42}, version=1)
  527. self.assertEqual(self.cache.get_many(['ford4','arthur4']),
  528. {'ford4': 37, 'arthur4': 42})
  529. self.assertEqual(self.cache.get_many(['ford4','arthur4'], version=1),
  530. {'ford4': 37, 'arthur4': 42})
  531. self.assertEqual(self.cache.get_many(['ford4','arthur4'], version=2), {})
  532. self.assertEqual(self.v2_cache.get_many(['ford4','arthur4']), {})
  533. self.assertEqual(self.v2_cache.get_many(['ford4','arthur4'], version=1),
  534. {'ford4': 37, 'arthur4': 42})
  535. self.assertEqual(self.v2_cache.get_many(['ford4','arthur4'], version=2), {})
  536. def test_incr_version(self):
  537. self.cache.set('answer', 42, version=2)
  538. self.assertEqual(self.cache.get('answer'), None)
  539. self.assertEqual(self.cache.get('answer', version=1), None)
  540. self.assertEqual(self.cache.get('answer', version=2), 42)
  541. self.assertEqual(self.cache.get('answer', version=3), None)
  542. self.assertEqual(self.cache.incr_version('answer', version=2), 3)
  543. self.assertEqual(self.cache.get('answer'), None)
  544. self.assertEqual(self.cache.get('answer', version=1), None)
  545. self.assertEqual(self.cache.get('answer', version=2), None)
  546. self.assertEqual(self.cache.get('answer', version=3), 42)
  547. self.v2_cache.set('answer2', 42)
  548. self.assertEqual(self.v2_cache.get('answer2'), 42)
  549. self.assertEqual(self.v2_cache.get('answer2', version=1), None)
  550. self.assertEqual(self.v2_cache.get('answer2', version=2), 42)
  551. self.assertEqual(self.v2_cache.get('answer2', version=3), None)
  552. self.assertEqual(self.v2_cache.incr_version('answer2'), 3)
  553. self.assertEqual(self.v2_cache.get('answer2'), None)
  554. self.assertEqual(self.v2_cache.get('answer2', version=1), None)
  555. self.assertEqual(self.v2_cache.get('answer2', version=2), None)
  556. self.assertEqual(self.v2_cache.get('answer2', version=3), 42)
  557. self.assertRaises(ValueError, self.cache.incr_version, 'does_not_exist')
  558. def test_decr_version(self):
  559. self.cache.set('answer', 42, version=2)
  560. self.assertEqual(self.cache.get('answer'), None)
  561. self.assertEqual(self.cache.get('answer', version=1), None)
  562. self.assertEqual(self.cache.get('answer', version=2), 42)
  563. self.assertEqual(self.cache.decr_version('answer', version=2), 1)
  564. self.assertEqual(self.cache.get('answer'), 42)
  565. self.assertEqual(self.cache.get('answer', version=1), 42)
  566. self.assertEqual(self.cache.get('answer', version=2), None)
  567. self.v2_cache.set('answer2', 42)
  568. self.assertEqual(self.v2_cache.get('answer2'), 42)
  569. self.assertEqual(self.v2_cache.get('answer2', version=1), None)
  570. self.assertEqual(self.v2_cache.get('answer2', version=2), 42)
  571. self.assertEqual(self.v2_cache.decr_version('answer2'), 1)
  572. self.assertEqual(self.v2_cache.get('answer2'), None)
  573. self.assertEqual(self.v2_cache.get('answer2', version=1), 42)
  574. self.assertEqual(self.v2_cache.get('answer2', version=2), None)
  575. self.assertRaises(ValueError, self.cache.decr_version, 'does_not_exist', version=2)
  576. def test_custom_key_func(self):
  577. # Two caches with different key functions aren't visible to each other
  578. self.cache.set('answer1', 42)
  579. self.assertEqual(self.cache.get('answer1'), 42)
  580. self.assertEqual(self.custom_key_cache.get('answer1'), None)
  581. self.assertEqual(self.custom_key_cache2.get('answer1'), None)
  582. self.custom_key_cache.set('answer2', 42)
  583. self.assertEqual(self.cache.get('answer2'), None)
  584. self.assertEqual(self.custom_key_cache.get('answer2'), 42)
  585. self.assertEqual(self.custom_key_cache2.get('answer2'), 42)
  586. def custom_key_func(key, key_prefix, version):
  587. "A customized cache key function"
  588. return 'CUSTOM-' + '-'.join([key_prefix, str(version), key])
  589. class DBCacheTests(unittest.TestCase, BaseCacheTests):
  590. def setUp(self):
  591. # Spaces are used in the table name to ensure quoting/escaping is working
  592. self._table_name = 'test cache table'
  593. management.call_command('createcachetable', self._table_name, verbosity=0, interactive=False)
  594. self.cache = get_cache('django.core.cache.backends.db.DatabaseCache', LOCATION=self._table_name, OPTIONS={'MAX_ENTRIES': 30})
  595. self.prefix_cache = get_cache('django.core.cache.backends.db.DatabaseCache', LOCATION=self._table_name, KEY_PREFIX='cacheprefix')
  596. self.v2_cache = get_cache('django.core.cache.backends.db.DatabaseCache', LOCATION=self._table_name, VERSION=2)
  597. self.custom_key_cache = get_cache('django.core.cache.backends.db.DatabaseCache', LOCATION=self._table_name, KEY_FUNCTION=custom_key_func)
  598. self.custom_key_cache2 = get_cache('django.core.cache.backends.db.DatabaseCache', LOCATION=self._table_name, KEY_FUNCTION='regressiontests.cache.tests.custom_key_func')
  599. def tearDown(self):
  600. from django.db import connection
  601. cursor = connection.cursor()
  602. cursor.execute('DROP TABLE %s' % connection.ops.quote_name(self._table_name))
  603. def test_cull(self):
  604. self.perform_cull_test(50, 29)
  605. def test_zero_cull(self):
  606. self.cache = get_cache('django.core.cache.backends.db.DatabaseCache', LOCATION=self._table_name, OPTIONS={'MAX_ENTRIES': 30, 'CULL_FREQUENCY': 0})
  607. self.perform_cull_test(50, 18)
  608. def test_old_initialization(self):
  609. self.cache = get_cache('db://%s?max_entries=30&cull_frequency=0' % self._table_name)
  610. self.perform_cull_test(50, 18)
  611. class LocMemCacheTests(unittest.TestCase, BaseCacheTests):
  612. def setUp(self):
  613. self.cache = get_cache('django.core.cache.backends.locmem.LocMemCache', OPTIONS={'MAX_ENTRIES': 30})
  614. self.prefix_cache = get_cache('django.core.cache.backends.locmem.LocMemCache', KEY_PREFIX='cacheprefix')
  615. self.v2_cache = get_cache('django.core.cache.backends.locmem.LocMemCache', VERSION=2)
  616. self.custom_key_cache = get_cache('django.core.cache.backends.locmem.LocMemCache', OPTIONS={'MAX_ENTRIES': 30}, KEY_FUNCTION=custom_key_func)
  617. self.custom_key_cache2 = get_cache('django.core.cache.backends.locmem.LocMemCache', OPTIONS={'MAX_ENTRIES': 30}, KEY_FUNCTION='regressiontests.cache.tests.custom_key_func')
  618. # LocMem requires a hack to make the other caches
  619. # share a data store with the 'normal' cache.
  620. self.prefix_cache._cache = self.cache._cache
  621. self.prefix_cache._expire_info = self.cache._expire_info
  622. self.v2_cache._cache = self.cache._cache
  623. self.v2_cache._expire_info = self.cache._expire_info
  624. self.custom_key_cache._cache = self.cache._cache
  625. self.custom_key_cache._expire_info = self.cache._expire_info
  626. self.custom_key_cache2._cache = self.cache._cache
  627. self.custom_key_cache2._expire_info = self.cache._expire_info
  628. def tearDown(self):
  629. self.cache.clear()
  630. def test_cull(self):
  631. self.perform_cull_test(50, 29)
  632. def test_zero_cull(self):
  633. self.cache = get_cache('django.core.cache.backends.locmem.LocMemCache', OPTIONS={'MAX_ENTRIES': 30, 'CULL_FREQUENCY': 0})
  634. self.perform_cull_test(50, 19)
  635. def test_old_initialization(self):
  636. self.cache = get_cache('locmem://?max_entries=30&cull_frequency=0')
  637. self.perform_cull_test(50, 19)
  638. def test_multiple_caches(self):
  639. "Check that multiple locmem caches are isolated"
  640. mirror_cache = get_cache('django.core.cache.backends.locmem.LocMemCache')
  641. other_cache = get_cache('django.core.cache.backends.locmem.LocMemCache', LOCATION='other')
  642. self.cache.set('value1', 42)
  643. self.assertEqual(mirror_cache.get('value1'), 42)
  644. self.assertEqual(other_cache.get('value1'), None)
  645. # memcached backend isn't guaranteed to be available.
  646. # To check the memcached backend, the test settings file will
  647. # need to contain a cache backend setting that points at
  648. # your memcache server.
  649. class MemcachedCacheTests(unittest.TestCase, BaseCacheTests):
  650. def setUp(self):
  651. name = settings.CACHES[DEFAULT_CACHE_ALIAS]['LOCATION']
  652. self.cache = get_cache('django.core.cache.backends.memcached.MemcachedCache', LOCATION=name)
  653. self.prefix_cache = get_cache('django.core.cache.backends.memcached.MemcachedCache', LOCATION=name, KEY_PREFIX='cacheprefix')
  654. self.v2_cache = get_cache('django.core.cache.backends.memcached.MemcachedCache', LOCATION=name, VERSION=2)
  655. self.custom_key_cache = get_cache('django.core.cache.backends.memcached.MemcachedCache', LOCATION=name, KEY_FUNCTION=custom_key_func)
  656. self.custom_key_cache2 = get_cache('django.core.cache.backends.memcached.MemcachedCache', LOCATION=name, KEY_FUNCTION='regressiontests.cache.tests.custom_key_func')
  657. def tearDown(self):
  658. self.cache.clear()
  659. def test_invalid_keys(self):
  660. """
  661. On memcached, we don't introduce a duplicate key validation
  662. step (for speed reasons), we just let the memcached API
  663. library raise its own exception on bad keys. Refs #6447.
  664. In order to be memcached-API-library agnostic, we only assert
  665. that a generic exception of some kind is raised.
  666. """
  667. # memcached does not allow whitespace or control characters in keys
  668. self.assertRaises(Exception, self.cache.set, 'key with spaces', 'value')
  669. # memcached limits key length to 250
  670. self.assertRaises(Exception, self.cache.set, 'a' * 251, 'value')
  671. MemcachedCacheTests = unittest.skipUnless(settings.CACHES[DEFAULT_CACHE_ALIAS]['BACKEND'].startswith('django.core.cache.backends.memcached.'), "memcached not available")(MemcachedCacheTests)
  672. class FileBasedCacheTests(unittest.TestCase, BaseCacheTests):
  673. """
  674. Specific test cases for the file-based cache.
  675. """
  676. def setUp(self):
  677. self.dirname = tempfile.mkdtemp()
  678. self.cache = get_cache('django.core.cache.backends.filebased.FileBasedCache', LOCATION=self.dirname, OPTIONS={'MAX_ENTRIES': 30})
  679. self.prefix_cache = get_cache('django.core.cache.backends.filebased.FileBasedCache', LOCATION=self.dirname, KEY_PREFIX='cacheprefix')
  680. self.v2_cache = get_cache('django.core.cache.backends.filebased.FileBasedCache', LOCATION=self.dirname, VERSION=2)
  681. self.custom_key_cache = get_cache('django.core.cache.backends.filebased.FileBasedCache', LOCATION=self.dirname, KEY_FUNCTION=custom_key_func)
  682. self.custom_key_cache2 = get_cache('django.core.cache.backends.filebased.FileBasedCache', LOCATION=self.dirname, KEY_FUNCTION='regressiontests.cache.tests.custom_key_func')
  683. def tearDown(self):
  684. self.cache.clear()
  685. def test_hashing(self):
  686. """Test that keys are hashed into subdirectories correctly"""
  687. self.cache.set("foo", "bar")
  688. key = self.cache.make_key("foo")
  689. keyhash = md5_constructor(key).hexdigest()
  690. keypath = os.path.join(self.dirname, keyhash[:2], keyhash[2:4], keyhash[4:])
  691. self.assertTrue(os.path.exists(keypath))
  692. def test_subdirectory_removal(self):
  693. """
  694. Make sure that the created subdirectories are correctly removed when empty.
  695. """
  696. self.cache.set("foo", "bar")
  697. key = self.cache.make_key("foo")
  698. keyhash = md5_constructor(key).hexdigest()
  699. keypath = os.path.join(self.dirname, keyhash[:2], keyhash[2:4], keyhash[4:])
  700. self.assertTrue(os.path.exists(keypath))
  701. self.cache.delete("foo")
  702. self.assertTrue(not os.path.exists(keypath))
  703. self.assertTrue(not os.path.exists(os.path.dirname(keypath)))
  704. self.assertTrue(not os.path.exists(os.path.dirname(os.path.dirname(keypath))))
  705. def test_cull(self):
  706. self.perform_cull_test(50, 29)
  707. def test_old_initialization(self):
  708. self.cache = get_cache('file://%s?max_entries=30' % self.dirname)
  709. self.perform_cull_test(50, 29)
  710. class CustomCacheKeyValidationTests(unittest.TestCase):
  711. """
  712. Tests for the ability to mixin a custom ``validate_key`` method to
  713. a custom cache backend that otherwise inherits from a builtin
  714. backend, and override the default key validation. Refs #6447.
  715. """
  716. def test_custom_key_validation(self):
  717. cache = get_cache('regressiontests.cache.liberal_backend://')
  718. # this key is both longer than 250 characters, and has spaces
  719. key = 'some key with spaces' * 15
  720. val = 'a value'
  721. cache.set(key, val)
  722. self.assertEqual(cache.get(key), val)
  723. class CacheUtils(unittest.TestCase):
  724. """TestCase for django.utils.cache functions."""
  725. def setUp(self):
  726. self.path = '/cache/test/'
  727. self.old_cache_middleware_key_prefix = settings.CACHE_MIDDLEWARE_KEY_PREFIX
  728. self.old_cache_middleware_seconds = settings.CACHE_MIDDLEWARE_SECONDS
  729. self.orig_use_i18n = settings.USE_I18N
  730. settings.CACHE_MIDDLEWARE_KEY_PREFIX = 'settingsprefix'
  731. settings.CACHE_MIDDLEWARE_SECONDS = 1
  732. settings.USE_I18N = False
  733. def tearDown(self):
  734. settings.CACHE_MIDDLEWARE_KEY_PREFIX = self.old_cache_middleware_key_prefix
  735. settings.CACHE_MIDDLEWARE_SECONDS = self.old_cache_middleware_seconds
  736. settings.USE_I18N = self.orig_use_i18n
  737. def _get_request(self, path, method='GET'):
  738. request = HttpRequest()
  739. request.META = {
  740. 'SERVER_NAME': 'testserver',
  741. 'SERVER_PORT': 80,
  742. }
  743. request.method = method
  744. request.path = request.path_info = "/cache/%s" % path
  745. return request
  746. def test_patch_vary_headers(self):
  747. headers = (
  748. # Initial vary, new headers, resulting vary.
  749. (None, ('Accept-Encoding',), 'Accept-Encoding'),
  750. ('Accept-Encoding', ('accept-encoding',), 'Accept-Encoding'),
  751. ('Accept-Encoding', ('ACCEPT-ENCODING',), 'Accept-Encoding'),
  752. ('Cookie', ('Accept-Encoding',), 'Cookie, Accept-Encoding'),
  753. ('Cookie, Accept-Encoding', ('Accept-Encoding',), 'Cookie, Accept-Encoding'),
  754. ('Cookie, Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  755. (None, ('Accept-Encoding', 'COOKIE'), 'Accept-Encoding, COOKIE'),
  756. ('Cookie, Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  757. ('Cookie , Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  758. )
  759. for initial_vary, newheaders, resulting_vary in headers:
  760. response = HttpResponse()
  761. if initial_vary is not None:
  762. response['Vary'] = initial_vary
  763. patch_vary_headers(response, newheaders)
  764. self.assertEqual(response['Vary'], resulting_vary)
  765. def test_get_cache_key(self):
  766. request = self._get_request(self.path)
  767. response = HttpResponse()
  768. key_prefix = 'localprefix'
  769. # Expect None if no headers have been set yet.
  770. self.assertEqual(get_cache_key(request), None)
  771. # Set headers to an empty list.
  772. learn_cache_key(request, response)
  773. self.assertEqual(get_cache_key(request), 'views.decorators.cache.cache_page.settingsprefix.GET.a8c87a3d8c44853d7f79474f7ffe4ad5.d41d8cd98f00b204e9800998ecf8427e')
  774. # Verify that a specified key_prefix is taken into account.
  775. learn_cache_key(request, response, key_prefix=key_prefix)
  776. self.assertEqual(get_cache_key(request, key_prefix=key_prefix), 'views.decorators.cache.cache_page.localprefix.GET.a8c87a3d8c44853d7f79474f7ffe4ad5.d41d8cd98f00b204e9800998ecf8427e')
  777. def test_get_cache_key_with_query(self):
  778. request = self._get_request(self.path + '?test=1')
  779. response = HttpResponse()
  780. # Expect None if no headers have been set yet.
  781. self.assertEqual(get_cache_key(request), None)
  782. # Set headers to an empty list.
  783. learn_cache_key(request, response)
  784. # Verify that the querystring is taken into account.
  785. self.assertEqual(get_cache_key(request), 'views.decorators.cache.cache_page.settingsprefix.GET.bd889c5a59603af44333ed21504db3cd.d41d8cd98f00b204e9800998ecf8427e')
  786. def test_learn_cache_key(self):
  787. request = self._get_request(self.path, 'HEAD')
  788. response = HttpResponse()
  789. response['Vary'] = 'Pony'
  790. # Make sure that the Vary header is added to the key hash
  791. learn_cache_key(request, response)
  792. self.assertEqual(get_cache_key(request), 'views.decorators.cache.cache_page.settingsprefix.HEAD.a8c87a3d8c44853d7f79474f7ffe4ad5.d41d8cd98f00b204e9800998ecf8427e')
  793. class PrefixedCacheUtils(CacheUtils):
  794. def setUp(self):
  795. super(PrefixedCacheUtils, self).setUp()
  796. self.old_cache_key_prefix = settings.CACHES['default'].get('KEY_PREFIX', None)
  797. settings.CACHES['default']['KEY_PREFIX'] = 'cacheprefix'
  798. def tearDown(self):
  799. super(PrefixedCacheUtils, self).tearDown()
  800. if self.old_cache_key_prefix is None:
  801. del settings.CACHES['default']['KEY_PREFIX']
  802. else:
  803. settings.CACHES['default']['KEY_PREFIX'] = self.old_cache_key_prefix
  804. class CacheHEADTest(unittest.TestCase):
  805. def setUp(self):
  806. self.orig_cache_middleware_seconds = settings.CACHE_MIDDLEWARE_SECONDS
  807. self.orig_cache_middleware_key_prefix = settings.CACHE_MIDDLEWARE_KEY_PREFIX
  808. self.orig_caches = settings.CACHES
  809. settings.CACHE_MIDDLEWARE_SECONDS = 60
  810. settings.CACHE_MIDDLEWARE_KEY_PREFIX = 'test'
  811. settings.CACHES = {
  812. 'default': {
  813. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache'
  814. }
  815. }
  816. self.path = '/cache/test/'
  817. def tearDown(self):
  818. settings.CACHE_MIDDLEWARE_SECONDS = self.orig_cache_middleware_seconds
  819. settings.CACHE_MIDDLEWARE_KEY_PREFIX = self.orig_cache_middleware_key_prefix
  820. settings.CACHES = self.orig_caches
  821. def _get_request(self, method):
  822. request = HttpRequest()
  823. request.META = {
  824. 'SERVER_NAME': 'testserver',
  825. 'SERVER_PORT': 80,
  826. }
  827. request.method = method
  828. request.path = request.path_info = self.path
  829. return request
  830. def _get_request_cache(self, method):
  831. request = self._get_request(method)
  832. request._cache_update_cache = True
  833. return request
  834. def _set_cache(self, request, msg):
  835. response = HttpResponse()
  836. response.content = msg
  837. return UpdateCacheMiddleware().process_response(request, response)
  838. def test_head_caches_correctly(self):
  839. test_content = 'test content'
  840. request = self._get_request_cache('HEAD')
  841. self._set_cache(request, test_content)
  842. request = self._get_request('HEAD')
  843. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  844. self.assertNotEqual(get_cache_data, None)
  845. self.assertEqual(test_content, get_cache_data.content)
  846. def test_head_with_cached_get(self):
  847. test_content = 'test content'
  848. request = self._get_request_cache('GET')
  849. self._set_cache(request, test_content)
  850. request = self._get_request('HEAD')
  851. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  852. self.assertNotEqual(get_cache_data, None)
  853. self.assertEqual(test_content, get_cache_data.content)
  854. class CacheI18nTest(unittest.TestCase):
  855. def setUp(self):
  856. self.orig_cache_middleware_seconds = settings.CACHE_MIDDLEWARE_SECONDS
  857. self.orig_cache_middleware_key_prefix = settings.CACHE_MIDDLEWARE_KEY_PREFIX
  858. self.orig_caches = settings.CACHES
  859. self.orig_use_i18n = settings.USE_I18N
  860. self.orig_languages = settings.LANGUAGES
  861. settings.LANGUAGES = (
  862. ('en', 'English'),
  863. ('es', 'Spanish'),
  864. )
  865. settings.CACHE_MIDDLEWARE_KEY_PREFIX = 'settingsprefix'
  866. self.path = '/cache/test/'
  867. def tearDown(self):
  868. settings.CACHE_MIDDLEWARE_SECONDS = self.orig_cache_middleware_seconds
  869. settings.CACHE_MIDDLEWARE_KEY_PREFIX = self.orig_cache_middleware_key_prefix
  870. settings.CACHES = self.orig_caches
  871. settings.USE_I18N = self.orig_use_i18n
  872. settings.LANGUAGES = self.orig_languages
  873. translation.deactivate()
  874. def _get_request(self):
  875. request = HttpRequest()
  876. request.META = {
  877. 'SERVER_NAME': 'testserver',
  878. 'SERVER_PORT': 80,
  879. }
  880. request.path = request.path_info = self.path
  881. return request
  882. def _get_request_cache(self, query_string=None):
  883. request = HttpRequest()
  884. request.META = {
  885. 'SERVER_NAME': 'testserver',
  886. 'SERVER_PORT': 80,
  887. }
  888. if query_string:
  889. request.META['QUERY_STRING'] = query_string
  890. request.GET = QueryDict(query_string)
  891. request.path = request.path_info = self.path
  892. request._cache_update_cache = True
  893. request.method = 'GET'
  894. request.session = {}
  895. return request
  896. def test_cache_key_i18n(self):
  897. settings.USE_I18N = True
  898. request = self._get_request()
  899. lang = translation.get_language()
  900. response = HttpResponse()
  901. key = learn_cache_key(request, response)
  902. self.assertTrue(key.endswith(lang), "Cache keys should include the language name when i18n is active")
  903. key2 = get_cache_key(request)
  904. self.assertEqual(key, key2)
  905. def test_cache_key_no_i18n (self):
  906. settings.USE_I18N = False
  907. request = self._get_request()
  908. lang = translation.get_language()
  909. response = HttpResponse()
  910. key = learn_cache_key(request, response)
  911. self.assertFalse(key.endswith(lang), "Cache keys shouldn't include the language name when i18n is inactive")
  912. def test_middleware(self):
  913. def set_cache(request, lang, msg):
  914. translation.activate(lang)
  915. response = HttpResponse()
  916. response.content= msg
  917. return UpdateCacheMiddleware().process_response(request, response)
  918. settings.CACHE_MIDDLEWARE_SECONDS = 60
  919. settings.CACHE_MIDDLEWARE_KEY_PREFIX = "test"
  920. settings.CACHES = {
  921. 'default': {
  922. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache'
  923. }
  924. }
  925. settings.USE_ETAGS = True
  926. settings.USE_I18N = True
  927. # cache with non empty request.GET
  928. request = self._get_request_cache(query_string='foo=bar&other=true')
  929. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  930. # first access, cache must return None
  931. self.assertEqual(get_cache_data, None)
  932. response = HttpResponse()
  933. content = 'Check for cache with QUERY_STRING'
  934. response.content = content
  935. UpdateCacheMiddleware().process_response(request, response)
  936. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  937. # cache must return content
  938. self.assertNotEqual(get_cache_data, None)
  939. self.assertEqual(get_cache_data.content, content)
  940. # different QUERY_STRING, cache must be empty
  941. request = self._get_request_cache(query_string='foo=bar&somethingelse=true')
  942. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  943. self.assertEqual(get_cache_data, None)
  944. # i18n tests
  945. en_message ="Hello world!"
  946. es_message ="Hola mundo!"
  947. request = self._get_request_cache()
  948. set_cache(request, 'en', en_message)
  949. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  950. # Check that we can recover the cache
  951. self.assertNotEqual(get_cache_data.content, None)
  952. self.assertEqual(en_message, get_cache_data.content)
  953. # Check that we use etags
  954. self.assertTrue(get_cache_data.has_header('ETag'))
  955. # Check that we can disable etags
  956. settings.USE_ETAGS = False
  957. request._cache_update_cache = True
  958. set_cache(request, 'en', en_message)
  959. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  960. self.assertFalse(get_cache_data.has_header('ETag'))
  961. # change the session language and set content
  962. request = self._get_request_cache()
  963. set_cache(request, 'es', es_message)
  964. # change again the language
  965. translation.activate('en')
  966. # retrieve the content from cache
  967. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  968. self.assertEqual(get_cache_data.content, en_message)
  969. # change again the language
  970. translation.activate('es')
  971. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  972. self.assertEqual(get_cache_data.content, es_message)
  973. class PrefixedCacheI18nTest(CacheI18nTest):
  974. def setUp(self):
  975. super(PrefixedCacheI18nTest, self).setUp()
  976. self.old_cache_key_prefix = settings.CACHES['default'].get('KEY_PREFIX', None)
  977. settings.CACHES['default']['KEY_PREFIX'] = 'cacheprefix'
  978. def tearDown(self):
  979. super(PrefixedCacheI18nTest, self).tearDown()
  980. if self.old_cache_key_prefix is not None:
  981. del settings.CACHES['default']['KEY_PREFIX']
  982. else:
  983. settings.CACHES['default']['KEY_PREFIX'] = self.old_cache_key_prefix
  984. def hello_world_view(request, value):
  985. return HttpResponse('Hello World %s' % value)
  986. class CacheMiddlewareTest(unittest.TestCase):
  987. def setUp(self):
  988. self.factory = RequestFactory()
  989. self.orig_cache_middleware_alias = settings.CACHE_MIDDLEWARE_ALIAS
  990. self.orig_cache_middleware_key_prefix = settings.CACHE_MIDDLEWARE_KEY_PREFIX
  991. self.orig_cache_middleware_seconds = settings.CACHE_MIDDLEWARE_SECONDS
  992. self.orig_cache_middleware_anonymous_only = getattr(settings, 'CACHE_MIDDLEWARE_ANONYMOUS_ONLY', False)
  993. self.orig_caches = settings.CACHES
  994. settings.CACHE_MIDDLEWARE_ALIAS = 'other'
  995. settings.CACHE_MIDDLEWARE_KEY_PREFIX = 'middlewareprefix'
  996. settings.CACHE_MIDDLEWARE_SECONDS = 30
  997. settings.CACHE_MIDDLEWARE_ANONYMOUS_ONLY = False
  998. settings.CACHES = {
  999. 'default': {
  1000. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache'
  1001. },
  1002. 'other': {
  1003. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1004. 'LOCATION': 'other',
  1005. 'TIMEOUT': '1'
  1006. }
  1007. }
  1008. def tearDown(self):
  1009. settings.CACHE_MIDDLEWARE_ALIAS = self.orig_cache_middleware_alias
  1010. settings.CACHE_MIDDLEWARE_KEY_PREFIX = self.orig_cache_middleware_key_prefix
  1011. settings.CACHE_MIDDLEWARE_SECONDS = self.orig_cache_middleware_seconds
  1012. settings.CACHE_MIDDLEWARE_ANONYMOUS_ONLY = self.orig_cache_middleware_anonymous_only
  1013. settings.CACHES = self.orig_caches
  1014. def test_constructor(self):
  1015. """
  1016. Ensure the constructor is correctly distinguishing between usage of CacheMiddleware as
  1017. Middleware vs. usage of CacheMiddleware as view decorator and setting attributes
  1018. appropriately.
  1019. """
  1020. # If no arguments are passed in construction, it's being used as middleware.
  1021. middleware = CacheMiddleware()
  1022. # Now test object attributes against values defined in setUp above
  1023. self.assertEqual(middleware.cache_timeout, 30)
  1024. self.assertEqual(middleware.key_prefix, 'middlewareprefix')
  1025. self.assertEqual(middleware.cache_alias, 'other')
  1026. self.assertEqual(middleware.cache_anonymous_only, False)
  1027. # If arguments are being passed in construction, it's being used as a decorator.
  1028. # First, test with "defaults":
  1029. as_view_decorator = CacheMiddleware(cache_alias=None, key_prefix=None)
  1030. self.assertEqual(as_view_decorator.cache_timeout, 300) # Timeout value for 'default' cache, i.e. 300
  1031. self.assertEqual(as_view_decorator.key_prefix, '')
  1032. self.assertEqual(as_view_decorator.cache_alias, 'default') # Value of DEFAULT_CACHE_ALIAS from django.core.cache
  1033. self.assertEqual(as_view_decorator.cache_anonymous_only, False)
  1034. # Next, test with custom values:
  1035. as_view_decorator_with_custom = CacheMiddleware(cache_anonymous_only=True, cache_timeout=60, cache_alias='other', key_prefix='foo')
  1036. self.assertEqual(as_view_decorator_with_custom.cache_timeout, 60)
  1037. self.assertEqual(as_view_decorator_with_custom.key_prefix, 'foo')
  1038. self.assertEqual(as_view_decorator_with_custom.cache_alias, 'other')
  1039. self.assertEqual(as_view_decorator_with_custom.cache_anonymous_only, True)
  1040. def test_middleware(self):
  1041. middleware = CacheMiddleware()
  1042. prefix_middleware = CacheMiddleware(key_prefix='prefix1')
  1043. timeout_middleware = CacheMiddleware(cache_timeout=1)
  1044. request = self.factory.get('/view/')
  1045. # Put the request through the request middleware
  1046. result = middleware.process_request(request)
  1047. self.assertEqual(result, None)
  1048. response = hello_world_view(request, '1')
  1049. # Now put the response through the response middleware
  1050. response = middleware.process_response(request, response)
  1051. # Repeating the request should result in a cache hit
  1052. result = middleware.process_request(request)
  1053. self.assertNotEquals(result, None)
  1054. self.assertEqual(result.content, 'Hello World 1')
  1055. # The same request through a different middleware won't hit
  1056. result = prefix_middleware.process_request(request)
  1057. self.assertEqual(result, None)
  1058. # The same request with a timeout _will_ hit
  1059. result = timeout_middleware.process_request(request)
  1060. self.assertNotEquals(result, None)
  1061. self.assertEqual(result.content, 'Hello World 1')
  1062. def test_cache_middleware_anonymous_only_wont_cause_session_access(self):
  1063. """ The cache middleware shouldn't cause a session access due to
  1064. CACHE_MIDDLEWARE_ANONYMOUS_ONLY if nothing else has accessed the
  1065. session. Refs 13283 """
  1066. settings.CACHE_MIDDLEWARE_ANONYMOUS_ONLY = True
  1067. from django.contrib.sessions.middleware import SessionMiddleware
  1068. from django.contrib.auth.middleware import AuthenticationMiddleware
  1069. middleware = CacheMiddleware()
  1070. session_middleware = SessionMiddleware()
  1071. auth_middleware = AuthenticationMiddleware()
  1072. request = self.factory.get('/view_anon/')
  1073. # Put the request through the request middleware
  1074. session_middleware.process_request(request)
  1075. auth_middleware.process_request(request)
  1076. result = middleware.process_request(request)
  1077. self.assertEqual(result, None)
  1078. response = hello_world_view(request, '1')
  1079. # Now put the response through the response middleware
  1080. session_middleware.process_response(request, response)
  1081. response = middleware.process_response(request, response)
  1082. self.assertEqual(request.session.accessed, False)
  1083. def test_cache_middleware_anonymous_only_with_cache_page(self):
  1084. """CACHE_MIDDLEWARE_ANONYMOUS_ONLY should still be effective when used
  1085. with the cache_page decorator: the response to a request from an
  1086. authenticated user should not be cached."""
  1087. settings.CACHE_MIDDLEWARE_ANONYMOUS_ONLY = True
  1088. request = self.factory.get('/view_anon/')
  1089. class MockAuthenticatedUser(object):
  1090. def is_authenticated(self):
  1091. return True
  1092. class MockAccessedSession(object):
  1093. accessed = True
  1094. request.user = MockAuthenticatedUser()
  1095. request.session = MockAccessedSession()
  1096. response = cache_page(hello_world_view)(request, '1')
  1097. self.assertFalse("Cache-Control" in response)
  1098. def test_view_decorator(self):
  1099. # decorate the same view with different cache decorators
  1100. default_view = cache_page(hello_world_view)
  1101. default_with_prefix_view = cache_page(key_prefix='prefix1')(hello_world_view)
  1102. explicit_default_view = cache_page(cache='default')(hello_world_view)
  1103. explicit_default_with_prefix_view = cache_page(cache='default', key_prefix='prefix1')(hello_world_view)
  1104. other_view = cache_page(cache='other')(hello_world_view)
  1105. other_with_prefix_view = cache_page(cache='other', key_prefix='prefix2')(hello_world_view)
  1106. other_with_timeout_view = cache_page(4, cache='other', key_prefix='prefix3')(hello_world_view)
  1107. request = self.factory.get('/view/')
  1108. # Request the view once
  1109. response = default_view(request, '1')
  1110. self.assertEqual(response.content, 'Hello World 1')
  1111. # Request again -- hit the cache
  1112. response = default_view(request, '2')
  1113. self.assertEqual(response.content, 'Hello World 1')
  1114. # Requesting the same view with the explicit cache should yield the same result
  1115. response = explicit_default_view(request, '3')
  1116. self.assertEqual(response.content, 'Hello World 1')
  1117. # Requesting with a prefix will hit a different cache key
  1118. response = explicit_default_with_prefix_view(request, '4')
  1119. self.assertEqual(response.content, 'Hello World 4')
  1120. # Hitting the same view again gives a cache hit
  1121. response = explicit_default_with_prefix_view(request, '5')
  1122. self.assertEqual(response.content, 'Hello World 4')
  1123. # And going back to the implicit cache will hit the same cache
  1124. response = default_with_prefix_view(request, '6')
  1125. self.assertEqual(response.content, 'Hello World 4')
  1126. # Requesting from an alternate cache won't hit cache
  1127. response = other_view(request, '7')
  1128. self.assertEqual(response.content, 'Hello World 7')
  1129. # But a repeated hit will hit cache
  1130. response = other_view(request, '8')
  1131. self.assertEqual(response.content, 'Hello World 7')
  1132. # And prefixing the alternate cache yields yet another cache entry
  1133. response = other_with_prefix_view(request, '9')
  1134. self.assertEqual(response.content, 'Hello World 9')
  1135. # Request from the alternate cache with a new prefix and a custom timeout
  1136. response = other_with_timeout_view(request, '10')
  1137. self.assertEqual(response.content, 'Hello World 10')
  1138. # But if we wait a couple of seconds...
  1139. time.sleep(2)
  1140. # ... the default cache will still hit
  1141. cache = get_cache('default')
  1142. response = default_view(request, '11')
  1143. self.assertEqual(response.content, 'Hello World 1')
  1144. # ... the default cache with a prefix will still hit
  1145. response = default_with_prefix_view(request, '12')
  1146. self.assertEqual(response.content, 'Hello World 4')
  1147. # ... the explicit default cache will still hit
  1148. response = explicit_default_view(request, '13')
  1149. self.assertEqual(response.content, 'Hello World 1')
  1150. # ... the explicit default cache with a prefix will still hit
  1151. response = explicit_default_with_prefix_view(request, '14')
  1152. self.assertEqual(response.content, 'Hello World 4')
  1153. # .. but a rapidly expiring cache won't hit
  1154. response = other_view(request, '15')
  1155. self.assertEqual(response.content, 'Hello World 15')
  1156. # .. even if it has a prefix
  1157. response = other_with_prefix_view(request, '16')
  1158. self.assertEqual(response.content, 'Hello World 16')
  1159. # ... but a view with a custom timeout will still hit
  1160. response = other_with_timeout_view(request, '17')
  1161. self.assertEqual(response.content, 'Hello World 10')
  1162. # And if we wait a few more seconds
  1163. time.sleep(2)
  1164. # the custom timeouot cache will miss
  1165. response = other_with_timeout_view(request, '18')
  1166. self.assertEqual(response.content, 'Hello World 18')
  1167. if __name__ == '__main__':
  1168. unittest.main()