tests.py 83 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126
  1. # -*- coding: utf-8 -*-
  2. # Unit tests for cache framework
  3. # Uses whatever cache backend is set in the test settings file.
  4. from __future__ import unicode_literals
  5. import os
  6. import re
  7. import copy
  8. import shutil
  9. import tempfile
  10. import threading
  11. import time
  12. import unittest
  13. import warnings
  14. from django.conf import settings
  15. from django.core import management
  16. from django.core import signals
  17. from django.core.cache import (cache, caches, CacheKeyWarning,
  18. InvalidCacheBackendError, DEFAULT_CACHE_ALIAS, get_cache,
  19. close_caches)
  20. from django.db import connection, connections, transaction
  21. from django.core.cache.utils import make_template_fragment_key
  22. from django.http import HttpResponse, StreamingHttpResponse
  23. from django.middleware.cache import (FetchFromCacheMiddleware,
  24. UpdateCacheMiddleware, CacheMiddleware)
  25. from django.middleware.csrf import CsrfViewMiddleware
  26. from django.template import Template
  27. from django.template.context_processors import csrf
  28. from django.template.response import TemplateResponse
  29. from django.test import TestCase, TransactionTestCase, RequestFactory, override_settings
  30. from django.test.signals import setting_changed
  31. from django.test.utils import IgnoreDeprecationWarningsMixin
  32. from django.utils import six
  33. from django.utils import timezone
  34. from django.utils import translation
  35. from django.utils.cache import (patch_vary_headers, get_cache_key,
  36. learn_cache_key, patch_cache_control, patch_response_headers)
  37. from django.utils.encoding import force_text
  38. from django.views.decorators.cache import cache_page
  39. try: # Use the same idiom as in cache backends
  40. from django.utils.six.moves import cPickle as pickle
  41. except ImportError:
  42. import pickle
  43. from .models import Poll, expensive_calculation
  44. # functions/classes for complex data type tests
  45. def f():
  46. return 42
  47. class C:
  48. def m(n):
  49. return 24
  50. class Unpickable(object):
  51. def __getstate__(self):
  52. raise pickle.PickleError()
  53. @override_settings(CACHES={
  54. 'default': {
  55. 'BACKEND': 'django.core.cache.backends.dummy.DummyCache',
  56. }
  57. })
  58. class DummyCacheTests(TestCase):
  59. # The Dummy cache backend doesn't really behave like a test backend,
  60. # so it has its own test case.
  61. def test_simple(self):
  62. "Dummy cache backend ignores cache set calls"
  63. cache.set("key", "value")
  64. self.assertIsNone(cache.get("key"))
  65. def test_add(self):
  66. "Add doesn't do anything in dummy cache backend"
  67. cache.add("addkey1", "value")
  68. result = cache.add("addkey1", "newvalue")
  69. self.assertTrue(result)
  70. self.assertIsNone(cache.get("addkey1"))
  71. def test_non_existent(self):
  72. "Non-existent keys aren't found in the dummy cache backend"
  73. self.assertIsNone(cache.get("does_not_exist"))
  74. self.assertEqual(cache.get("does_not_exist", "bang!"), "bang!")
  75. def test_get_many(self):
  76. "get_many returns nothing for the dummy cache backend"
  77. cache.set('a', 'a')
  78. cache.set('b', 'b')
  79. cache.set('c', 'c')
  80. cache.set('d', 'd')
  81. self.assertEqual(cache.get_many(['a', 'c', 'd']), {})
  82. self.assertEqual(cache.get_many(['a', 'b', 'e']), {})
  83. def test_delete(self):
  84. "Cache deletion is transparently ignored on the dummy cache backend"
  85. cache.set("key1", "spam")
  86. cache.set("key2", "eggs")
  87. self.assertIsNone(cache.get("key1"))
  88. cache.delete("key1")
  89. self.assertIsNone(cache.get("key1"))
  90. self.assertIsNone(cache.get("key2"))
  91. def test_has_key(self):
  92. "The has_key method doesn't ever return True for the dummy cache backend"
  93. cache.set("hello1", "goodbye1")
  94. self.assertFalse(cache.has_key("hello1"))
  95. self.assertFalse(cache.has_key("goodbye1"))
  96. def test_in(self):
  97. "The in operator doesn't ever return True for the dummy cache backend"
  98. cache.set("hello2", "goodbye2")
  99. self.assertNotIn("hello2", cache)
  100. self.assertNotIn("goodbye2", cache)
  101. def test_incr(self):
  102. "Dummy cache values can't be incremented"
  103. cache.set('answer', 42)
  104. self.assertRaises(ValueError, cache.incr, 'answer')
  105. self.assertRaises(ValueError, cache.incr, 'does_not_exist')
  106. def test_decr(self):
  107. "Dummy cache values can't be decremented"
  108. cache.set('answer', 42)
  109. self.assertRaises(ValueError, cache.decr, 'answer')
  110. self.assertRaises(ValueError, cache.decr, 'does_not_exist')
  111. def test_data_types(self):
  112. "All data types are ignored equally by the dummy cache"
  113. stuff = {
  114. 'string': 'this is a string',
  115. 'int': 42,
  116. 'list': [1, 2, 3, 4],
  117. 'tuple': (1, 2, 3, 4),
  118. 'dict': {'A': 1, 'B': 2},
  119. 'function': f,
  120. 'class': C,
  121. }
  122. cache.set("stuff", stuff)
  123. self.assertIsNone(cache.get("stuff"))
  124. def test_expiration(self):
  125. "Expiration has no effect on the dummy cache"
  126. cache.set('expire1', 'very quickly', 1)
  127. cache.set('expire2', 'very quickly', 1)
  128. cache.set('expire3', 'very quickly', 1)
  129. time.sleep(2)
  130. self.assertIsNone(cache.get("expire1"))
  131. cache.add("expire2", "newvalue")
  132. self.assertIsNone(cache.get("expire2"))
  133. self.assertFalse(cache.has_key("expire3"))
  134. def test_unicode(self):
  135. "Unicode values are ignored by the dummy cache"
  136. stuff = {
  137. 'ascii': 'ascii_value',
  138. 'unicode_ascii': 'Iñtërnâtiônàlizætiøn1',
  139. 'Iñtërnâtiônàlizætiøn': 'Iñtërnâtiônàlizætiøn2',
  140. 'ascii2': {'x': 1}
  141. }
  142. for (key, value) in stuff.items():
  143. cache.set(key, value)
  144. self.assertIsNone(cache.get(key))
  145. def test_set_many(self):
  146. "set_many does nothing for the dummy cache backend"
  147. cache.set_many({'a': 1, 'b': 2})
  148. cache.set_many({'a': 1, 'b': 2}, timeout=2, version='1')
  149. def test_delete_many(self):
  150. "delete_many does nothing for the dummy cache backend"
  151. cache.delete_many(['a', 'b'])
  152. def test_clear(self):
  153. "clear does nothing for the dummy cache backend"
  154. cache.clear()
  155. def test_incr_version(self):
  156. "Dummy cache versions can't be incremented"
  157. cache.set('answer', 42)
  158. self.assertRaises(ValueError, cache.incr_version, 'answer')
  159. self.assertRaises(ValueError, cache.incr_version, 'does_not_exist')
  160. def test_decr_version(self):
  161. "Dummy cache versions can't be decremented"
  162. cache.set('answer', 42)
  163. self.assertRaises(ValueError, cache.decr_version, 'answer')
  164. self.assertRaises(ValueError, cache.decr_version, 'does_not_exist')
  165. def custom_key_func(key, key_prefix, version):
  166. "A customized cache key function"
  167. return 'CUSTOM-' + '-'.join([key_prefix, str(version), key])
  168. _caches_setting_base = {
  169. 'default': {},
  170. 'prefix': {'KEY_PREFIX': 'cacheprefix{}'.format(os.getpid())},
  171. 'v2': {'VERSION': 2},
  172. 'custom_key': {'KEY_FUNCTION': custom_key_func},
  173. 'custom_key2': {'KEY_FUNCTION': 'cache.tests.custom_key_func'},
  174. 'cull': {'OPTIONS': {'MAX_ENTRIES': 30}},
  175. 'zero_cull': {'OPTIONS': {'CULL_FREQUENCY': 0, 'MAX_ENTRIES': 30}},
  176. }
  177. def caches_setting_for_tests(base=None, **params):
  178. # `base` is used to pull in the memcached config from the original settings,
  179. # `params` are test specific overrides and `_caches_settings_base` is the
  180. # base config for the tests.
  181. # This results in the following search order:
  182. # params -> _caches_setting_base -> base
  183. base = base or {}
  184. setting = {k: base.copy() for k in _caches_setting_base.keys()}
  185. for key, cache_params in setting.items():
  186. cache_params.update(_caches_setting_base[key])
  187. cache_params.update(params)
  188. return setting
  189. class BaseCacheTests(object):
  190. # A common set of tests to apply to all cache backends
  191. def setUp(self):
  192. self.factory = RequestFactory()
  193. def tearDown(self):
  194. cache.clear()
  195. def test_simple(self):
  196. # Simple cache set/get works
  197. cache.set("key", "value")
  198. self.assertEqual(cache.get("key"), "value")
  199. def test_add(self):
  200. # A key can be added to a cache
  201. cache.add("addkey1", "value")
  202. result = cache.add("addkey1", "newvalue")
  203. self.assertFalse(result)
  204. self.assertEqual(cache.get("addkey1"), "value")
  205. def test_prefix(self):
  206. # Test for same cache key conflicts between shared backend
  207. cache.set('somekey', 'value')
  208. # should not be set in the prefixed cache
  209. self.assertFalse(caches['prefix'].has_key('somekey'))
  210. caches['prefix'].set('somekey', 'value2')
  211. self.assertEqual(cache.get('somekey'), 'value')
  212. self.assertEqual(caches['prefix'].get('somekey'), 'value2')
  213. def test_non_existent(self):
  214. # Non-existent cache keys return as None/default
  215. # get with non-existent keys
  216. self.assertIsNone(cache.get("does_not_exist"))
  217. self.assertEqual(cache.get("does_not_exist", "bang!"), "bang!")
  218. def test_get_many(self):
  219. # Multiple cache keys can be returned using get_many
  220. cache.set('a', 'a')
  221. cache.set('b', 'b')
  222. cache.set('c', 'c')
  223. cache.set('d', 'd')
  224. self.assertDictEqual(cache.get_many(['a', 'c', 'd']), {'a': 'a', 'c': 'c', 'd': 'd'})
  225. self.assertDictEqual(cache.get_many(['a', 'b', 'e']), {'a': 'a', 'b': 'b'})
  226. def test_delete(self):
  227. # Cache keys can be deleted
  228. cache.set("key1", "spam")
  229. cache.set("key2", "eggs")
  230. self.assertEqual(cache.get("key1"), "spam")
  231. cache.delete("key1")
  232. self.assertIsNone(cache.get("key1"))
  233. self.assertEqual(cache.get("key2"), "eggs")
  234. def test_has_key(self):
  235. # The cache can be inspected for cache keys
  236. cache.set("hello1", "goodbye1")
  237. self.assertTrue(cache.has_key("hello1"))
  238. self.assertFalse(cache.has_key("goodbye1"))
  239. cache.set("no_expiry", "here", None)
  240. self.assertTrue(cache.has_key("no_expiry"))
  241. def test_in(self):
  242. # The in operator can be used to inspect cache contents
  243. cache.set("hello2", "goodbye2")
  244. self.assertIn("hello2", cache)
  245. self.assertNotIn("goodbye2", cache)
  246. def test_incr(self):
  247. # Cache values can be incremented
  248. cache.set('answer', 41)
  249. self.assertEqual(cache.incr('answer'), 42)
  250. self.assertEqual(cache.get('answer'), 42)
  251. self.assertEqual(cache.incr('answer', 10), 52)
  252. self.assertEqual(cache.get('answer'), 52)
  253. self.assertEqual(cache.incr('answer', -10), 42)
  254. self.assertRaises(ValueError, cache.incr, 'does_not_exist')
  255. def test_decr(self):
  256. # Cache values can be decremented
  257. cache.set('answer', 43)
  258. self.assertEqual(cache.decr('answer'), 42)
  259. self.assertEqual(cache.get('answer'), 42)
  260. self.assertEqual(cache.decr('answer', 10), 32)
  261. self.assertEqual(cache.get('answer'), 32)
  262. self.assertEqual(cache.decr('answer', -10), 42)
  263. self.assertRaises(ValueError, cache.decr, 'does_not_exist')
  264. def test_close(self):
  265. self.assertTrue(hasattr(cache, 'close'))
  266. cache.close()
  267. def test_data_types(self):
  268. # Many different data types can be cached
  269. stuff = {
  270. 'string': 'this is a string',
  271. 'int': 42,
  272. 'list': [1, 2, 3, 4],
  273. 'tuple': (1, 2, 3, 4),
  274. 'dict': {'A': 1, 'B': 2},
  275. 'function': f,
  276. 'class': C,
  277. }
  278. cache.set("stuff", stuff)
  279. self.assertEqual(cache.get("stuff"), stuff)
  280. def test_cache_read_for_model_instance(self):
  281. # Don't want fields with callable as default to be called on cache read
  282. expensive_calculation.num_runs = 0
  283. Poll.objects.all().delete()
  284. my_poll = Poll.objects.create(question="Well?")
  285. self.assertEqual(Poll.objects.count(), 1)
  286. pub_date = my_poll.pub_date
  287. cache.set('question', my_poll)
  288. cached_poll = cache.get('question')
  289. self.assertEqual(cached_poll.pub_date, pub_date)
  290. # We only want the default expensive calculation run once
  291. self.assertEqual(expensive_calculation.num_runs, 1)
  292. def test_cache_write_for_model_instance_with_deferred(self):
  293. # Don't want fields with callable as default to be called on cache write
  294. expensive_calculation.num_runs = 0
  295. Poll.objects.all().delete()
  296. Poll.objects.create(question="What?")
  297. self.assertEqual(expensive_calculation.num_runs, 1)
  298. defer_qs = Poll.objects.all().defer('question')
  299. self.assertEqual(defer_qs.count(), 1)
  300. self.assertEqual(expensive_calculation.num_runs, 1)
  301. cache.set('deferred_queryset', defer_qs)
  302. # cache set should not re-evaluate default functions
  303. self.assertEqual(expensive_calculation.num_runs, 1)
  304. def test_cache_read_for_model_instance_with_deferred(self):
  305. # Don't want fields with callable as default to be called on cache read
  306. expensive_calculation.num_runs = 0
  307. Poll.objects.all().delete()
  308. Poll.objects.create(question="What?")
  309. self.assertEqual(expensive_calculation.num_runs, 1)
  310. defer_qs = Poll.objects.all().defer('question')
  311. self.assertEqual(defer_qs.count(), 1)
  312. cache.set('deferred_queryset', defer_qs)
  313. self.assertEqual(expensive_calculation.num_runs, 1)
  314. runs_before_cache_read = expensive_calculation.num_runs
  315. cache.get('deferred_queryset')
  316. # We only want the default expensive calculation run on creation and set
  317. self.assertEqual(expensive_calculation.num_runs, runs_before_cache_read)
  318. def test_expiration(self):
  319. # Cache values can be set to expire
  320. cache.set('expire1', 'very quickly', 1)
  321. cache.set('expire2', 'very quickly', 1)
  322. cache.set('expire3', 'very quickly', 1)
  323. time.sleep(2)
  324. self.assertIsNone(cache.get("expire1"))
  325. cache.add("expire2", "newvalue")
  326. self.assertEqual(cache.get("expire2"), "newvalue")
  327. self.assertFalse(cache.has_key("expire3"))
  328. def test_unicode(self):
  329. # Unicode values can be cached
  330. stuff = {
  331. 'ascii': 'ascii_value',
  332. 'unicode_ascii': 'Iñtërnâtiônàlizætiøn1',
  333. 'Iñtërnâtiônàlizætiøn': 'Iñtërnâtiônàlizætiøn2',
  334. 'ascii2': {'x': 1}
  335. }
  336. # Test `set`
  337. for (key, value) in stuff.items():
  338. cache.set(key, value)
  339. self.assertEqual(cache.get(key), value)
  340. # Test `add`
  341. for (key, value) in stuff.items():
  342. cache.delete(key)
  343. cache.add(key, value)
  344. self.assertEqual(cache.get(key), value)
  345. # Test `set_many`
  346. for (key, value) in stuff.items():
  347. cache.delete(key)
  348. cache.set_many(stuff)
  349. for (key, value) in stuff.items():
  350. self.assertEqual(cache.get(key), value)
  351. def test_binary_string(self):
  352. # Binary strings should be cacheable
  353. from zlib import compress, decompress
  354. value = 'value_to_be_compressed'
  355. compressed_value = compress(value.encode())
  356. # Test set
  357. cache.set('binary1', compressed_value)
  358. compressed_result = cache.get('binary1')
  359. self.assertEqual(compressed_value, compressed_result)
  360. self.assertEqual(value, decompress(compressed_result).decode())
  361. # Test add
  362. cache.add('binary1-add', compressed_value)
  363. compressed_result = cache.get('binary1-add')
  364. self.assertEqual(compressed_value, compressed_result)
  365. self.assertEqual(value, decompress(compressed_result).decode())
  366. # Test set_many
  367. cache.set_many({'binary1-set_many': compressed_value})
  368. compressed_result = cache.get('binary1-set_many')
  369. self.assertEqual(compressed_value, compressed_result)
  370. self.assertEqual(value, decompress(compressed_result).decode())
  371. def test_set_many(self):
  372. # Multiple keys can be set using set_many
  373. cache.set_many({"key1": "spam", "key2": "eggs"})
  374. self.assertEqual(cache.get("key1"), "spam")
  375. self.assertEqual(cache.get("key2"), "eggs")
  376. def test_set_many_expiration(self):
  377. # set_many takes a second ``timeout`` parameter
  378. cache.set_many({"key1": "spam", "key2": "eggs"}, 1)
  379. time.sleep(2)
  380. self.assertIsNone(cache.get("key1"))
  381. self.assertIsNone(cache.get("key2"))
  382. def test_delete_many(self):
  383. # Multiple keys can be deleted using delete_many
  384. cache.set("key1", "spam")
  385. cache.set("key2", "eggs")
  386. cache.set("key3", "ham")
  387. cache.delete_many(["key1", "key2"])
  388. self.assertIsNone(cache.get("key1"))
  389. self.assertIsNone(cache.get("key2"))
  390. self.assertEqual(cache.get("key3"), "ham")
  391. def test_clear(self):
  392. # The cache can be emptied using clear
  393. cache.set("key1", "spam")
  394. cache.set("key2", "eggs")
  395. cache.clear()
  396. self.assertIsNone(cache.get("key1"))
  397. self.assertIsNone(cache.get("key2"))
  398. def test_long_timeout(self):
  399. '''
  400. Using a timeout greater than 30 days makes memcached think
  401. it is an absolute expiration timestamp instead of a relative
  402. offset. Test that we honour this convention. Refs #12399.
  403. '''
  404. cache.set('key1', 'eggs', 60 * 60 * 24 * 30 + 1) # 30 days + 1 second
  405. self.assertEqual(cache.get('key1'), 'eggs')
  406. cache.add('key2', 'ham', 60 * 60 * 24 * 30 + 1)
  407. self.assertEqual(cache.get('key2'), 'ham')
  408. cache.set_many({'key3': 'sausage', 'key4': 'lobster bisque'}, 60 * 60 * 24 * 30 + 1)
  409. self.assertEqual(cache.get('key3'), 'sausage')
  410. self.assertEqual(cache.get('key4'), 'lobster bisque')
  411. def test_forever_timeout(self):
  412. '''
  413. Passing in None into timeout results in a value that is cached forever
  414. '''
  415. cache.set('key1', 'eggs', None)
  416. self.assertEqual(cache.get('key1'), 'eggs')
  417. cache.add('key2', 'ham', None)
  418. self.assertEqual(cache.get('key2'), 'ham')
  419. added = cache.add('key1', 'new eggs', None)
  420. self.assertEqual(added, False)
  421. self.assertEqual(cache.get('key1'), 'eggs')
  422. cache.set_many({'key3': 'sausage', 'key4': 'lobster bisque'}, None)
  423. self.assertEqual(cache.get('key3'), 'sausage')
  424. self.assertEqual(cache.get('key4'), 'lobster bisque')
  425. def test_zero_timeout(self):
  426. '''
  427. Passing in zero into timeout results in a value that is not cached
  428. '''
  429. cache.set('key1', 'eggs', 0)
  430. self.assertIsNone(cache.get('key1'))
  431. cache.add('key2', 'ham', 0)
  432. self.assertIsNone(cache.get('key2'))
  433. cache.set_many({'key3': 'sausage', 'key4': 'lobster bisque'}, 0)
  434. self.assertIsNone(cache.get('key3'))
  435. self.assertIsNone(cache.get('key4'))
  436. def test_float_timeout(self):
  437. # Make sure a timeout given as a float doesn't crash anything.
  438. cache.set("key1", "spam", 100.2)
  439. self.assertEqual(cache.get("key1"), "spam")
  440. def _perform_cull_test(self, cull_cache, initial_count, final_count):
  441. # Create initial cache key entries. This will overflow the cache,
  442. # causing a cull.
  443. for i in range(1, initial_count):
  444. cull_cache.set('cull%d' % i, 'value', 1000)
  445. count = 0
  446. # Count how many keys are left in the cache.
  447. for i in range(1, initial_count):
  448. if cull_cache.has_key('cull%d' % i):
  449. count = count + 1
  450. self.assertEqual(count, final_count)
  451. def test_cull(self):
  452. self._perform_cull_test(caches['cull'], 50, 29)
  453. def test_zero_cull(self):
  454. self._perform_cull_test(caches['zero_cull'], 50, 19)
  455. def test_invalid_keys(self):
  456. """
  457. All the builtin backends (except memcached, see below) should warn on
  458. keys that would be refused by memcached. This encourages portable
  459. caching code without making it too difficult to use production backends
  460. with more liberal key rules. Refs #6447.
  461. """
  462. # mimic custom ``make_key`` method being defined since the default will
  463. # never show the below warnings
  464. def func(key, *args):
  465. return key
  466. old_func = cache.key_func
  467. cache.key_func = func
  468. try:
  469. with warnings.catch_warnings(record=True) as w:
  470. warnings.simplefilter("always")
  471. # memcached does not allow whitespace or control characters in keys
  472. cache.set('key with spaces', 'value')
  473. self.assertEqual(len(w), 2)
  474. self.assertIsInstance(w[0].message, CacheKeyWarning)
  475. with warnings.catch_warnings(record=True) as w:
  476. warnings.simplefilter("always")
  477. # memcached limits key length to 250
  478. cache.set('a' * 251, 'value')
  479. self.assertEqual(len(w), 1)
  480. self.assertIsInstance(w[0].message, CacheKeyWarning)
  481. finally:
  482. cache.key_func = old_func
  483. def test_cache_versioning_get_set(self):
  484. # set, using default version = 1
  485. cache.set('answer1', 42)
  486. self.assertEqual(cache.get('answer1'), 42)
  487. self.assertEqual(cache.get('answer1', version=1), 42)
  488. self.assertIsNone(cache.get('answer1', version=2))
  489. self.assertIsNone(caches['v2'].get('answer1'))
  490. self.assertEqual(caches['v2'].get('answer1', version=1), 42)
  491. self.assertIsNone(caches['v2'].get('answer1', version=2))
  492. # set, default version = 1, but manually override version = 2
  493. cache.set('answer2', 42, version=2)
  494. self.assertIsNone(cache.get('answer2'))
  495. self.assertIsNone(cache.get('answer2', version=1))
  496. self.assertEqual(cache.get('answer2', version=2), 42)
  497. self.assertEqual(caches['v2'].get('answer2'), 42)
  498. self.assertIsNone(caches['v2'].get('answer2', version=1))
  499. self.assertEqual(caches['v2'].get('answer2', version=2), 42)
  500. # v2 set, using default version = 2
  501. caches['v2'].set('answer3', 42)
  502. self.assertIsNone(cache.get('answer3'))
  503. self.assertIsNone(cache.get('answer3', version=1))
  504. self.assertEqual(cache.get('answer3', version=2), 42)
  505. self.assertEqual(caches['v2'].get('answer3'), 42)
  506. self.assertIsNone(caches['v2'].get('answer3', version=1))
  507. self.assertEqual(caches['v2'].get('answer3', version=2), 42)
  508. # v2 set, default version = 2, but manually override version = 1
  509. caches['v2'].set('answer4', 42, version=1)
  510. self.assertEqual(cache.get('answer4'), 42)
  511. self.assertEqual(cache.get('answer4', version=1), 42)
  512. self.assertIsNone(cache.get('answer4', version=2))
  513. self.assertIsNone(caches['v2'].get('answer4'))
  514. self.assertEqual(caches['v2'].get('answer4', version=1), 42)
  515. self.assertIsNone(caches['v2'].get('answer4', version=2))
  516. def test_cache_versioning_add(self):
  517. # add, default version = 1, but manually override version = 2
  518. cache.add('answer1', 42, version=2)
  519. self.assertIsNone(cache.get('answer1', version=1))
  520. self.assertEqual(cache.get('answer1', version=2), 42)
  521. cache.add('answer1', 37, version=2)
  522. self.assertIsNone(cache.get('answer1', version=1))
  523. self.assertEqual(cache.get('answer1', version=2), 42)
  524. cache.add('answer1', 37, version=1)
  525. self.assertEqual(cache.get('answer1', version=1), 37)
  526. self.assertEqual(cache.get('answer1', version=2), 42)
  527. # v2 add, using default version = 2
  528. caches['v2'].add('answer2', 42)
  529. self.assertIsNone(cache.get('answer2', version=1))
  530. self.assertEqual(cache.get('answer2', version=2), 42)
  531. caches['v2'].add('answer2', 37)
  532. self.assertIsNone(cache.get('answer2', version=1))
  533. self.assertEqual(cache.get('answer2', version=2), 42)
  534. caches['v2'].add('answer2', 37, version=1)
  535. self.assertEqual(cache.get('answer2', version=1), 37)
  536. self.assertEqual(cache.get('answer2', version=2), 42)
  537. # v2 add, default version = 2, but manually override version = 1
  538. caches['v2'].add('answer3', 42, version=1)
  539. self.assertEqual(cache.get('answer3', version=1), 42)
  540. self.assertIsNone(cache.get('answer3', version=2))
  541. caches['v2'].add('answer3', 37, version=1)
  542. self.assertEqual(cache.get('answer3', version=1), 42)
  543. self.assertIsNone(cache.get('answer3', version=2))
  544. caches['v2'].add('answer3', 37)
  545. self.assertEqual(cache.get('answer3', version=1), 42)
  546. self.assertEqual(cache.get('answer3', version=2), 37)
  547. def test_cache_versioning_has_key(self):
  548. cache.set('answer1', 42)
  549. # has_key
  550. self.assertTrue(cache.has_key('answer1'))
  551. self.assertTrue(cache.has_key('answer1', version=1))
  552. self.assertFalse(cache.has_key('answer1', version=2))
  553. self.assertFalse(caches['v2'].has_key('answer1'))
  554. self.assertTrue(caches['v2'].has_key('answer1', version=1))
  555. self.assertFalse(caches['v2'].has_key('answer1', version=2))
  556. def test_cache_versioning_delete(self):
  557. cache.set('answer1', 37, version=1)
  558. cache.set('answer1', 42, version=2)
  559. cache.delete('answer1')
  560. self.assertIsNone(cache.get('answer1', version=1))
  561. self.assertEqual(cache.get('answer1', version=2), 42)
  562. cache.set('answer2', 37, version=1)
  563. cache.set('answer2', 42, version=2)
  564. cache.delete('answer2', version=2)
  565. self.assertEqual(cache.get('answer2', version=1), 37)
  566. self.assertIsNone(cache.get('answer2', version=2))
  567. cache.set('answer3', 37, version=1)
  568. cache.set('answer3', 42, version=2)
  569. caches['v2'].delete('answer3')
  570. self.assertEqual(cache.get('answer3', version=1), 37)
  571. self.assertIsNone(cache.get('answer3', version=2))
  572. cache.set('answer4', 37, version=1)
  573. cache.set('answer4', 42, version=2)
  574. caches['v2'].delete('answer4', version=1)
  575. self.assertIsNone(cache.get('answer4', version=1))
  576. self.assertEqual(cache.get('answer4', version=2), 42)
  577. def test_cache_versioning_incr_decr(self):
  578. cache.set('answer1', 37, version=1)
  579. cache.set('answer1', 42, version=2)
  580. cache.incr('answer1')
  581. self.assertEqual(cache.get('answer1', version=1), 38)
  582. self.assertEqual(cache.get('answer1', version=2), 42)
  583. cache.decr('answer1')
  584. self.assertEqual(cache.get('answer1', version=1), 37)
  585. self.assertEqual(cache.get('answer1', version=2), 42)
  586. cache.set('answer2', 37, version=1)
  587. cache.set('answer2', 42, version=2)
  588. cache.incr('answer2', version=2)
  589. self.assertEqual(cache.get('answer2', version=1), 37)
  590. self.assertEqual(cache.get('answer2', version=2), 43)
  591. cache.decr('answer2', version=2)
  592. self.assertEqual(cache.get('answer2', version=1), 37)
  593. self.assertEqual(cache.get('answer2', version=2), 42)
  594. cache.set('answer3', 37, version=1)
  595. cache.set('answer3', 42, version=2)
  596. caches['v2'].incr('answer3')
  597. self.assertEqual(cache.get('answer3', version=1), 37)
  598. self.assertEqual(cache.get('answer3', version=2), 43)
  599. caches['v2'].decr('answer3')
  600. self.assertEqual(cache.get('answer3', version=1), 37)
  601. self.assertEqual(cache.get('answer3', version=2), 42)
  602. cache.set('answer4', 37, version=1)
  603. cache.set('answer4', 42, version=2)
  604. caches['v2'].incr('answer4', version=1)
  605. self.assertEqual(cache.get('answer4', version=1), 38)
  606. self.assertEqual(cache.get('answer4', version=2), 42)
  607. caches['v2'].decr('answer4', version=1)
  608. self.assertEqual(cache.get('answer4', version=1), 37)
  609. self.assertEqual(cache.get('answer4', version=2), 42)
  610. def test_cache_versioning_get_set_many(self):
  611. # set, using default version = 1
  612. cache.set_many({'ford1': 37, 'arthur1': 42})
  613. self.assertDictEqual(cache.get_many(['ford1', 'arthur1']),
  614. {'ford1': 37, 'arthur1': 42})
  615. self.assertDictEqual(cache.get_many(['ford1', 'arthur1'], version=1),
  616. {'ford1': 37, 'arthur1': 42})
  617. self.assertDictEqual(cache.get_many(['ford1', 'arthur1'], version=2), {})
  618. self.assertDictEqual(caches['v2'].get_many(['ford1', 'arthur1']), {})
  619. self.assertDictEqual(caches['v2'].get_many(['ford1', 'arthur1'], version=1),
  620. {'ford1': 37, 'arthur1': 42})
  621. self.assertDictEqual(caches['v2'].get_many(['ford1', 'arthur1'], version=2), {})
  622. # set, default version = 1, but manually override version = 2
  623. cache.set_many({'ford2': 37, 'arthur2': 42}, version=2)
  624. self.assertDictEqual(cache.get_many(['ford2', 'arthur2']), {})
  625. self.assertDictEqual(cache.get_many(['ford2', 'arthur2'], version=1), {})
  626. self.assertDictEqual(cache.get_many(['ford2', 'arthur2'], version=2),
  627. {'ford2': 37, 'arthur2': 42})
  628. self.assertDictEqual(caches['v2'].get_many(['ford2', 'arthur2']),
  629. {'ford2': 37, 'arthur2': 42})
  630. self.assertDictEqual(caches['v2'].get_many(['ford2', 'arthur2'], version=1), {})
  631. self.assertDictEqual(caches['v2'].get_many(['ford2', 'arthur2'], version=2),
  632. {'ford2': 37, 'arthur2': 42})
  633. # v2 set, using default version = 2
  634. caches['v2'].set_many({'ford3': 37, 'arthur3': 42})
  635. self.assertDictEqual(cache.get_many(['ford3', 'arthur3']), {})
  636. self.assertDictEqual(cache.get_many(['ford3', 'arthur3'], version=1), {})
  637. self.assertDictEqual(cache.get_many(['ford3', 'arthur3'], version=2),
  638. {'ford3': 37, 'arthur3': 42})
  639. self.assertDictEqual(caches['v2'].get_many(['ford3', 'arthur3']),
  640. {'ford3': 37, 'arthur3': 42})
  641. self.assertDictEqual(caches['v2'].get_many(['ford3', 'arthur3'], version=1), {})
  642. self.assertDictEqual(caches['v2'].get_many(['ford3', 'arthur3'], version=2),
  643. {'ford3': 37, 'arthur3': 42})
  644. # v2 set, default version = 2, but manually override version = 1
  645. caches['v2'].set_many({'ford4': 37, 'arthur4': 42}, version=1)
  646. self.assertDictEqual(cache.get_many(['ford4', 'arthur4']),
  647. {'ford4': 37, 'arthur4': 42})
  648. self.assertDictEqual(cache.get_many(['ford4', 'arthur4'], version=1),
  649. {'ford4': 37, 'arthur4': 42})
  650. self.assertDictEqual(cache.get_many(['ford4', 'arthur4'], version=2), {})
  651. self.assertDictEqual(caches['v2'].get_many(['ford4', 'arthur4']), {})
  652. self.assertDictEqual(caches['v2'].get_many(['ford4', 'arthur4'], version=1),
  653. {'ford4': 37, 'arthur4': 42})
  654. self.assertDictEqual(caches['v2'].get_many(['ford4', 'arthur4'], version=2), {})
  655. def test_incr_version(self):
  656. cache.set('answer', 42, version=2)
  657. self.assertIsNone(cache.get('answer'))
  658. self.assertIsNone(cache.get('answer', version=1))
  659. self.assertEqual(cache.get('answer', version=2), 42)
  660. self.assertIsNone(cache.get('answer', version=3))
  661. self.assertEqual(cache.incr_version('answer', version=2), 3)
  662. self.assertIsNone(cache.get('answer'))
  663. self.assertIsNone(cache.get('answer', version=1))
  664. self.assertIsNone(cache.get('answer', version=2))
  665. self.assertEqual(cache.get('answer', version=3), 42)
  666. caches['v2'].set('answer2', 42)
  667. self.assertEqual(caches['v2'].get('answer2'), 42)
  668. self.assertIsNone(caches['v2'].get('answer2', version=1))
  669. self.assertEqual(caches['v2'].get('answer2', version=2), 42)
  670. self.assertIsNone(caches['v2'].get('answer2', version=3))
  671. self.assertEqual(caches['v2'].incr_version('answer2'), 3)
  672. self.assertIsNone(caches['v2'].get('answer2'))
  673. self.assertIsNone(caches['v2'].get('answer2', version=1))
  674. self.assertIsNone(caches['v2'].get('answer2', version=2))
  675. self.assertEqual(caches['v2'].get('answer2', version=3), 42)
  676. self.assertRaises(ValueError, cache.incr_version, 'does_not_exist')
  677. def test_decr_version(self):
  678. cache.set('answer', 42, version=2)
  679. self.assertIsNone(cache.get('answer'))
  680. self.assertIsNone(cache.get('answer', version=1))
  681. self.assertEqual(cache.get('answer', version=2), 42)
  682. self.assertEqual(cache.decr_version('answer', version=2), 1)
  683. self.assertEqual(cache.get('answer'), 42)
  684. self.assertEqual(cache.get('answer', version=1), 42)
  685. self.assertIsNone(cache.get('answer', version=2))
  686. caches['v2'].set('answer2', 42)
  687. self.assertEqual(caches['v2'].get('answer2'), 42)
  688. self.assertIsNone(caches['v2'].get('answer2', version=1))
  689. self.assertEqual(caches['v2'].get('answer2', version=2), 42)
  690. self.assertEqual(caches['v2'].decr_version('answer2'), 1)
  691. self.assertIsNone(caches['v2'].get('answer2'))
  692. self.assertEqual(caches['v2'].get('answer2', version=1), 42)
  693. self.assertIsNone(caches['v2'].get('answer2', version=2))
  694. self.assertRaises(ValueError, cache.decr_version, 'does_not_exist', version=2)
  695. def test_custom_key_func(self):
  696. # Two caches with different key functions aren't visible to each other
  697. cache.set('answer1', 42)
  698. self.assertEqual(cache.get('answer1'), 42)
  699. self.assertIsNone(caches['custom_key'].get('answer1'))
  700. self.assertIsNone(caches['custom_key2'].get('answer1'))
  701. caches['custom_key'].set('answer2', 42)
  702. self.assertIsNone(cache.get('answer2'))
  703. self.assertEqual(caches['custom_key'].get('answer2'), 42)
  704. self.assertEqual(caches['custom_key2'].get('answer2'), 42)
  705. def test_cache_write_unpickable_object(self):
  706. update_middleware = UpdateCacheMiddleware()
  707. update_middleware.cache = cache
  708. fetch_middleware = FetchFromCacheMiddleware()
  709. fetch_middleware.cache = cache
  710. request = self.factory.get('/cache/test')
  711. request._cache_update_cache = True
  712. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  713. self.assertIsNone(get_cache_data)
  714. response = HttpResponse()
  715. content = 'Testing cookie serialization.'
  716. response.content = content
  717. response.set_cookie('foo', 'bar')
  718. update_middleware.process_response(request, response)
  719. get_cache_data = fetch_middleware.process_request(request)
  720. self.assertIsNotNone(get_cache_data)
  721. self.assertEqual(get_cache_data.content, content.encode('utf-8'))
  722. self.assertEqual(get_cache_data.cookies, response.cookies)
  723. update_middleware.process_response(request, get_cache_data)
  724. get_cache_data = fetch_middleware.process_request(request)
  725. self.assertIsNotNone(get_cache_data)
  726. self.assertEqual(get_cache_data.content, content.encode('utf-8'))
  727. self.assertEqual(get_cache_data.cookies, response.cookies)
  728. def test_add_fail_on_pickleerror(self):
  729. "See https://code.djangoproject.com/ticket/21200"
  730. with self.assertRaises(pickle.PickleError):
  731. cache.add('unpickable', Unpickable())
  732. def test_set_fail_on_pickleerror(self):
  733. "See https://code.djangoproject.com/ticket/21200"
  734. with self.assertRaises(pickle.PickleError):
  735. cache.set('unpickable', Unpickable())
  736. @override_settings(CACHES=caches_setting_for_tests(
  737. BACKEND='django.core.cache.backends.db.DatabaseCache',
  738. # Spaces are used in the table name to ensure quoting/escaping is working
  739. LOCATION='test cache table'
  740. ))
  741. class DBCacheTests(BaseCacheTests, TransactionTestCase):
  742. available_apps = ['cache']
  743. def setUp(self):
  744. # The super calls needs to happen first for the settings override.
  745. super(DBCacheTests, self).setUp()
  746. self.create_table()
  747. def tearDown(self):
  748. # The super call needs to happen first because it uses the database.
  749. super(DBCacheTests, self).tearDown()
  750. self.drop_table()
  751. def create_table(self):
  752. management.call_command('createcachetable', verbosity=0, interactive=False)
  753. def drop_table(self):
  754. with connection.cursor() as cursor:
  755. table_name = connection.ops.quote_name('test cache table')
  756. cursor.execute('DROP TABLE %s' % table_name)
  757. def test_zero_cull(self):
  758. self._perform_cull_test(caches['zero_cull'], 50, 18)
  759. def test_second_call_doesnt_crash(self):
  760. out = six.StringIO()
  761. management.call_command('createcachetable', stdout=out)
  762. self.assertEqual(out.getvalue(),
  763. "Cache table 'test cache table' already exists.\n" * len(settings.CACHES))
  764. def test_createcachetable_with_table_argument(self):
  765. """
  766. Delete and recreate cache table with legacy behavior (explicitly
  767. specifying the table name).
  768. """
  769. self.drop_table()
  770. out = six.StringIO()
  771. management.call_command(
  772. 'createcachetable',
  773. 'test cache table',
  774. verbosity=2,
  775. stdout=out,
  776. )
  777. self.assertEqual(out.getvalue(),
  778. "Cache table 'test cache table' created.\n")
  779. def test_clear_commits_transaction(self):
  780. # Ensure the database transaction is committed (#19896)
  781. cache.set("key1", "spam")
  782. cache.clear()
  783. transaction.rollback()
  784. self.assertIsNone(cache.get("key1"))
  785. @override_settings(USE_TZ=True)
  786. class DBCacheWithTimeZoneTests(DBCacheTests):
  787. pass
  788. class DBCacheRouter(object):
  789. """A router that puts the cache table on the 'other' database."""
  790. def db_for_read(self, model, **hints):
  791. if model._meta.app_label == 'django_cache':
  792. return 'other'
  793. def db_for_write(self, model, **hints):
  794. if model._meta.app_label == 'django_cache':
  795. return 'other'
  796. def allow_migrate(self, db, model):
  797. if model._meta.app_label == 'django_cache':
  798. return db == 'other'
  799. @override_settings(
  800. CACHES={
  801. 'default': {
  802. 'BACKEND': 'django.core.cache.backends.db.DatabaseCache',
  803. 'LOCATION': 'my_cache_table',
  804. },
  805. },
  806. )
  807. class CreateCacheTableForDBCacheTests(TestCase):
  808. multi_db = True
  809. @override_settings(DATABASE_ROUTERS=[DBCacheRouter()])
  810. def test_createcachetable_observes_database_router(self):
  811. # cache table should not be created on 'default'
  812. with self.assertNumQueries(0, using='default'):
  813. management.call_command('createcachetable',
  814. database='default',
  815. verbosity=0, interactive=False)
  816. # cache table should be created on 'other'
  817. # Queries:
  818. # 1: check table doesn't already exist
  819. # 2: create savepoint (if transactional DDL is supported)
  820. # 3: create the table
  821. # 4: create the index
  822. # 5: release savepoint (if transactional DDL is supported)
  823. num = 5 if connections['other'].features.can_rollback_ddl else 3
  824. with self.assertNumQueries(num, using='other'):
  825. management.call_command('createcachetable',
  826. database='other',
  827. verbosity=0, interactive=False)
  828. class PicklingSideEffect(object):
  829. def __init__(self, cache):
  830. self.cache = cache
  831. self.locked = False
  832. def __getstate__(self):
  833. if self.cache._lock.active_writers:
  834. self.locked = True
  835. return {}
  836. @override_settings(CACHES=caches_setting_for_tests(
  837. BACKEND='django.core.cache.backends.locmem.LocMemCache',
  838. ))
  839. class LocMemCacheTests(BaseCacheTests, TestCase):
  840. def setUp(self):
  841. super(LocMemCacheTests, self).setUp()
  842. # LocMem requires a hack to make the other caches
  843. # share a data store with the 'normal' cache.
  844. caches['prefix']._cache = cache._cache
  845. caches['prefix']._expire_info = cache._expire_info
  846. caches['v2']._cache = cache._cache
  847. caches['v2']._expire_info = cache._expire_info
  848. caches['custom_key']._cache = cache._cache
  849. caches['custom_key']._expire_info = cache._expire_info
  850. caches['custom_key2']._cache = cache._cache
  851. caches['custom_key2']._expire_info = cache._expire_info
  852. @override_settings(CACHES={
  853. 'default': {'BACKEND': 'django.core.cache.backends.locmem.LocMemCache'},
  854. 'other': {
  855. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  856. 'LOCATION': 'other'
  857. },
  858. })
  859. def test_multiple_caches(self):
  860. "Check that multiple locmem caches are isolated"
  861. cache.set('value', 42)
  862. self.assertEqual(caches['default'].get('value'), 42)
  863. self.assertIsNone(caches['other'].get('value'))
  864. def test_locking_on_pickle(self):
  865. """#20613/#18541 -- Ensures pickling is done outside of the lock."""
  866. bad_obj = PicklingSideEffect(cache)
  867. cache.set('set', bad_obj)
  868. self.assertFalse(bad_obj.locked, "Cache was locked during pickling")
  869. cache.add('add', bad_obj)
  870. self.assertFalse(bad_obj.locked, "Cache was locked during pickling")
  871. def test_incr_decr_timeout(self):
  872. """incr/decr does not modify expiry time (matches memcached behavior)"""
  873. key = 'value'
  874. _key = cache.make_key(key)
  875. cache.set(key, 1, timeout=cache.default_timeout * 10)
  876. expire = cache._expire_info[_key]
  877. cache.incr(key)
  878. self.assertEqual(expire, cache._expire_info[_key])
  879. cache.decr(key)
  880. self.assertEqual(expire, cache._expire_info[_key])
  881. # memcached backend isn't guaranteed to be available.
  882. # To check the memcached backend, the test settings file will
  883. # need to contain at least one cache backend setting that points at
  884. # your memcache server.
  885. memcached_params = {}
  886. for _cache_params in settings.CACHES.values():
  887. if _cache_params['BACKEND'].startswith('django.core.cache.backends.memcached.'):
  888. memcached_params = _cache_params
  889. memcached_never_expiring_params = memcached_params.copy()
  890. memcached_never_expiring_params['TIMEOUT'] = None
  891. memcached_far_future_params = memcached_params.copy()
  892. memcached_far_future_params['TIMEOUT'] = 31536000 # 60*60*24*365, 1 year
  893. @unittest.skipUnless(memcached_params, "memcached not available")
  894. @override_settings(CACHES=caches_setting_for_tests(base=memcached_params))
  895. class MemcachedCacheTests(BaseCacheTests, TestCase):
  896. def test_invalid_keys(self):
  897. """
  898. On memcached, we don't introduce a duplicate key validation
  899. step (for speed reasons), we just let the memcached API
  900. library raise its own exception on bad keys. Refs #6447.
  901. In order to be memcached-API-library agnostic, we only assert
  902. that a generic exception of some kind is raised.
  903. """
  904. # memcached does not allow whitespace or control characters in keys
  905. self.assertRaises(Exception, cache.set, 'key with spaces', 'value')
  906. # memcached limits key length to 250
  907. self.assertRaises(Exception, cache.set, 'a' * 251, 'value')
  908. # Explicitly display a skipped test if no configured cache uses MemcachedCache
  909. @unittest.skipUnless(
  910. memcached_params.get('BACKEND') == 'django.core.cache.backends.memcached.MemcachedCache',
  911. "cache with python-memcached library not available")
  912. def test_memcached_uses_highest_pickle_version(self):
  913. # Regression test for #19810
  914. for cache_key, cache_config in settings.CACHES.items():
  915. if cache_config['BACKEND'] == 'django.core.cache.backends.memcached.MemcachedCache':
  916. self.assertEqual(caches[cache_key]._cache.pickleProtocol,
  917. pickle.HIGHEST_PROTOCOL)
  918. @override_settings(CACHES=caches_setting_for_tests(base=memcached_never_expiring_params))
  919. def test_default_never_expiring_timeout(self):
  920. # Regression test for #22845
  921. cache.set('infinite_foo', 'bar')
  922. self.assertEqual(cache.get('infinite_foo'), 'bar')
  923. @override_settings(CACHES=caches_setting_for_tests(base=memcached_far_future_params))
  924. def test_default_far_future_timeout(self):
  925. # Regression test for #22845
  926. cache.set('future_foo', 'bar')
  927. self.assertEqual(cache.get('future_foo'), 'bar')
  928. def test_cull(self):
  929. # culling isn't implemented, memcached deals with it.
  930. pass
  931. def test_zero_cull(self):
  932. # culling isn't implemented, memcached deals with it.
  933. pass
  934. def test_memcached_deletes_key_on_failed_set(self):
  935. # By default memcached allows objects up to 1MB. For the cache_db session
  936. # backend to always use the current session, memcached needs to delete
  937. # the old key if it fails to set.
  938. # pylibmc doesn't seem to have SERVER_MAX_VALUE_LENGTH as far as I can
  939. # tell from a quick check of its source code. This is falling back to
  940. # the default value exposed by python-memcached on my system.
  941. max_value_length = getattr(cache._lib, 'SERVER_MAX_VALUE_LENGTH', 1048576)
  942. cache.set('small_value', 'a')
  943. self.assertEqual(cache.get('small_value'), 'a')
  944. large_value = 'a' * (max_value_length + 1)
  945. cache.set('small_value', large_value)
  946. # small_value should be deleted, or set if configured to accept larger values
  947. value = cache.get('small_value')
  948. self.assertTrue(value is None or value == large_value)
  949. @override_settings(CACHES=caches_setting_for_tests(
  950. BACKEND='django.core.cache.backends.filebased.FileBasedCache',
  951. ))
  952. class FileBasedCacheTests(BaseCacheTests, TestCase):
  953. """
  954. Specific test cases for the file-based cache.
  955. """
  956. def setUp(self):
  957. super(FileBasedCacheTests, self).setUp()
  958. self.dirname = tempfile.mkdtemp()
  959. # Caches location cannot be modified through override_settings / modify_settings,
  960. # hence settings are manipulated directly here and the setting_changed signal
  961. # is triggered manually.
  962. for cache_params in settings.CACHES.values():
  963. cache_params.update({'LOCATION': self.dirname})
  964. setting_changed.send(self.__class__, setting='CACHES', enter=False)
  965. def tearDown(self):
  966. super(FileBasedCacheTests, self).tearDown()
  967. # Call parent first, as cache.clear() may recreate cache base directory
  968. shutil.rmtree(self.dirname)
  969. def test_ignores_non_cache_files(self):
  970. fname = os.path.join(self.dirname, 'not-a-cache-file')
  971. with open(fname, 'w'):
  972. os.utime(fname, None)
  973. cache.clear()
  974. self.assertTrue(os.path.exists(fname),
  975. 'Expected cache.clear to ignore non cache files')
  976. os.remove(fname)
  977. def test_clear_does_not_remove_cache_dir(self):
  978. cache.clear()
  979. self.assertTrue(os.path.exists(self.dirname),
  980. 'Expected cache.clear to keep the cache dir')
  981. def test_creates_cache_dir_if_nonexistent(self):
  982. os.rmdir(self.dirname)
  983. cache.set('foo', 'bar')
  984. os.path.exists(self.dirname)
  985. @override_settings(CACHES={
  986. 'default': {
  987. 'BACKEND': 'cache.liberal_backend.CacheClass',
  988. },
  989. })
  990. class CustomCacheKeyValidationTests(TestCase):
  991. """
  992. Tests for the ability to mixin a custom ``validate_key`` method to
  993. a custom cache backend that otherwise inherits from a builtin
  994. backend, and override the default key validation. Refs #6447.
  995. """
  996. def test_custom_key_validation(self):
  997. # this key is both longer than 250 characters, and has spaces
  998. key = 'some key with spaces' * 15
  999. val = 'a value'
  1000. cache.set(key, val)
  1001. self.assertEqual(cache.get(key), val)
  1002. @override_settings(
  1003. CACHES={
  1004. 'default': {
  1005. 'BACKEND': 'cache.closeable_cache.CacheClass',
  1006. }
  1007. }
  1008. )
  1009. class GetCacheTests(IgnoreDeprecationWarningsMixin, TestCase):
  1010. def test_simple(self):
  1011. self.assertIsInstance(
  1012. caches[DEFAULT_CACHE_ALIAS],
  1013. get_cache('default').__class__
  1014. )
  1015. cache = get_cache(
  1016. 'django.core.cache.backends.dummy.DummyCache',
  1017. **{'TIMEOUT': 120}
  1018. )
  1019. self.assertEqual(cache.default_timeout, 120)
  1020. self.assertRaises(InvalidCacheBackendError, get_cache, 'does_not_exist')
  1021. def test_close(self):
  1022. self.assertFalse(cache.closed)
  1023. signals.request_finished.send(self.__class__)
  1024. self.assertTrue(cache.closed)
  1025. def test_close_deprecated(self):
  1026. cache = get_cache('cache.closeable_cache.CacheClass')
  1027. self.assertFalse(cache.closed)
  1028. # Ensure that we don't close the global cache instances.
  1029. signals.request_finished.disconnect(close_caches)
  1030. try:
  1031. signals.request_finished.send(self.__class__)
  1032. self.assertTrue(cache.closed)
  1033. finally:
  1034. signals.request_finished.connect(close_caches)
  1035. DEFAULT_MEMORY_CACHES_SETTINGS = {
  1036. 'default': {
  1037. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1038. 'LOCATION': 'unique-snowflake',
  1039. }
  1040. }
  1041. NEVER_EXPIRING_CACHES_SETTINGS = copy.deepcopy(DEFAULT_MEMORY_CACHES_SETTINGS)
  1042. NEVER_EXPIRING_CACHES_SETTINGS['default']['TIMEOUT'] = None
  1043. class DefaultNonExpiringCacheKeyTests(TestCase):
  1044. """Tests that verify that settings having Cache arguments with a TIMEOUT
  1045. set to `None` will create Caches that will set non-expiring keys.
  1046. This fixes ticket #22085.
  1047. """
  1048. def setUp(self):
  1049. # The 5 minute (300 seconds) default expiration time for keys is
  1050. # defined in the implementation of the initializer method of the
  1051. # BaseCache type.
  1052. self.DEFAULT_TIMEOUT = caches[DEFAULT_CACHE_ALIAS].default_timeout
  1053. def tearDown(self):
  1054. del(self.DEFAULT_TIMEOUT)
  1055. def test_default_expiration_time_for_keys_is_5_minutes(self):
  1056. """The default expiration time of a cache key is 5 minutes.
  1057. This value is defined inside the __init__() method of the
  1058. :class:`django.core.cache.backends.base.BaseCache` type.
  1059. """
  1060. self.assertEqual(300, self.DEFAULT_TIMEOUT)
  1061. def test_caches_with_unset_timeout_has_correct_default_timeout(self):
  1062. """Caches that have the TIMEOUT parameter undefined in the default
  1063. settings will use the default 5 minute timeout.
  1064. """
  1065. cache = caches[DEFAULT_CACHE_ALIAS]
  1066. self.assertEqual(self.DEFAULT_TIMEOUT, cache.default_timeout)
  1067. @override_settings(CACHES=NEVER_EXPIRING_CACHES_SETTINGS)
  1068. def test_caches_set_with_timeout_as_none_has_correct_default_timeout(self):
  1069. """Memory caches that have the TIMEOUT parameter set to `None` in the
  1070. default settings with have `None` as the default timeout.
  1071. This means "no timeout".
  1072. """
  1073. cache = caches[DEFAULT_CACHE_ALIAS]
  1074. self.assertIsNone(cache.default_timeout)
  1075. self.assertIsNone(cache.get_backend_timeout())
  1076. @override_settings(CACHES=DEFAULT_MEMORY_CACHES_SETTINGS)
  1077. def test_caches_with_unset_timeout_set_expiring_key(self):
  1078. """Memory caches that have the TIMEOUT parameter unset will set cache
  1079. keys having the default 5 minute timeout.
  1080. """
  1081. key = "my-key"
  1082. value = "my-value"
  1083. cache = caches[DEFAULT_CACHE_ALIAS]
  1084. cache.set(key, value)
  1085. cache_key = cache.make_key(key)
  1086. self.assertIsNotNone(cache._expire_info[cache_key])
  1087. @override_settings(CACHES=NEVER_EXPIRING_CACHES_SETTINGS)
  1088. def text_caches_set_with_timeout_as_none_set_non_expiring_key(self):
  1089. """Memory caches that have the TIMEOUT parameter set to `None` will set
  1090. a non expiring key by default.
  1091. """
  1092. key = "another-key"
  1093. value = "another-value"
  1094. cache = caches[DEFAULT_CACHE_ALIAS]
  1095. cache.set(key, value)
  1096. cache_key = cache.make_key(key)
  1097. self.assertIsNone(cache._expire_info[cache_key])
  1098. @override_settings(
  1099. CACHE_MIDDLEWARE_KEY_PREFIX='settingsprefix',
  1100. CACHE_MIDDLEWARE_SECONDS=1,
  1101. CACHES={
  1102. 'default': {
  1103. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1104. },
  1105. },
  1106. USE_I18N=False,
  1107. )
  1108. class CacheUtils(TestCase):
  1109. """TestCase for django.utils.cache functions."""
  1110. def setUp(self):
  1111. self.host = 'www.example.com'
  1112. self.path = '/cache/test/'
  1113. self.factory = RequestFactory(HTTP_HOST=self.host)
  1114. def _get_request_cache(self, method='GET', query_string=None, update_cache=None):
  1115. request = self._get_request(self.host, self.path,
  1116. method, query_string=query_string)
  1117. request._cache_update_cache = True if not update_cache else update_cache
  1118. return request
  1119. def _set_cache(self, request, msg):
  1120. response = HttpResponse()
  1121. response.content = msg
  1122. return UpdateCacheMiddleware().process_response(request, response)
  1123. def test_patch_vary_headers(self):
  1124. headers = (
  1125. # Initial vary, new headers, resulting vary.
  1126. (None, ('Accept-Encoding',), 'Accept-Encoding'),
  1127. ('Accept-Encoding', ('accept-encoding',), 'Accept-Encoding'),
  1128. ('Accept-Encoding', ('ACCEPT-ENCODING',), 'Accept-Encoding'),
  1129. ('Cookie', ('Accept-Encoding',), 'Cookie, Accept-Encoding'),
  1130. ('Cookie, Accept-Encoding', ('Accept-Encoding',), 'Cookie, Accept-Encoding'),
  1131. ('Cookie, Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  1132. (None, ('Accept-Encoding', 'COOKIE'), 'Accept-Encoding, COOKIE'),
  1133. ('Cookie, Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  1134. ('Cookie , Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  1135. )
  1136. for initial_vary, newheaders, resulting_vary in headers:
  1137. response = HttpResponse()
  1138. if initial_vary is not None:
  1139. response['Vary'] = initial_vary
  1140. patch_vary_headers(response, newheaders)
  1141. self.assertEqual(response['Vary'], resulting_vary)
  1142. def test_get_cache_key(self):
  1143. request = self.factory.get(self.path)
  1144. response = HttpResponse()
  1145. # Expect None if no headers have been set yet.
  1146. self.assertIsNone(get_cache_key(request, response))
  1147. # Set headers to an empty list.
  1148. learn_cache_key(request, response)
  1149. self.assertEqual(
  1150. get_cache_key(request),
  1151. 'views.decorators.cache.cache_page.settingsprefix.GET.'
  1152. '18a03f9c9649f7d684af5db3524f5c99.d41d8cd98f00b204e9800998ecf8427e'
  1153. )
  1154. # Verify that a specified key_prefix is taken into account.
  1155. key_prefix = 'localprefix'
  1156. learn_cache_key(request, response, key_prefix=key_prefix)
  1157. self.assertEqual(
  1158. get_cache_key(request, key_prefix=key_prefix),
  1159. 'views.decorators.cache.cache_page.localprefix.GET.'
  1160. '18a03f9c9649f7d684af5db3524f5c99.d41d8cd98f00b204e9800998ecf8427e'
  1161. )
  1162. def test_get_cache_key_with_query(self):
  1163. request = self.factory.get(self.path, {'test': 1})
  1164. response = HttpResponse()
  1165. # Expect None if no headers have been set yet.
  1166. self.assertIsNone(get_cache_key(request))
  1167. # Set headers to an empty list.
  1168. learn_cache_key(request, response)
  1169. # Verify that the querystring is taken into account.
  1170. self.assertEqual(
  1171. get_cache_key(request),
  1172. 'views.decorators.cache.cache_page.settingsprefix.GET.'
  1173. 'beaf87a9a99ee81c673ea2d67ccbec2a.d41d8cd98f00b204e9800998ecf8427e'
  1174. )
  1175. def test_cache_key_varies_by_url(self):
  1176. """
  1177. get_cache_key keys differ by fully-qualified URL instead of path
  1178. """
  1179. request1 = self.factory.get(self.path, HTTP_HOST='sub-1.example.com')
  1180. learn_cache_key(request1, HttpResponse())
  1181. request2 = self.factory.get(self.path, HTTP_HOST='sub-2.example.com')
  1182. learn_cache_key(request2, HttpResponse())
  1183. self.assertNotEqual(get_cache_key(request1), get_cache_key(request2))
  1184. def test_learn_cache_key(self):
  1185. request = self.factory.head(self.path)
  1186. response = HttpResponse()
  1187. response['Vary'] = 'Pony'
  1188. # Make sure that the Vary header is added to the key hash
  1189. learn_cache_key(request, response)
  1190. self.assertEqual(
  1191. get_cache_key(request),
  1192. 'views.decorators.cache.cache_page.settingsprefix.GET.'
  1193. '18a03f9c9649f7d684af5db3524f5c99.d41d8cd98f00b204e9800998ecf8427e'
  1194. )
  1195. def test_patch_cache_control(self):
  1196. tests = (
  1197. # Initial Cache-Control, kwargs to patch_cache_control, expected Cache-Control parts
  1198. (None, {'private': True}, {'private'}),
  1199. # Test whether private/public attributes are mutually exclusive
  1200. ('private', {'private': True}, {'private'}),
  1201. ('private', {'public': True}, {'public'}),
  1202. ('public', {'public': True}, {'public'}),
  1203. ('public', {'private': True}, {'private'}),
  1204. ('must-revalidate,max-age=60,private', {'public': True}, {'must-revalidate', 'max-age=60', 'public'}),
  1205. ('must-revalidate,max-age=60,public', {'private': True}, {'must-revalidate', 'max-age=60', 'private'}),
  1206. ('must-revalidate,max-age=60', {'public': True}, {'must-revalidate', 'max-age=60', 'public'}),
  1207. )
  1208. cc_delim_re = re.compile(r'\s*,\s*')
  1209. for initial_cc, newheaders, expected_cc in tests:
  1210. response = HttpResponse()
  1211. if initial_cc is not None:
  1212. response['Cache-Control'] = initial_cc
  1213. patch_cache_control(response, **newheaders)
  1214. parts = set(cc_delim_re.split(response['Cache-Control']))
  1215. self.assertEqual(parts, expected_cc)
  1216. @override_settings(
  1217. CACHES={
  1218. 'default': {
  1219. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1220. 'KEY_PREFIX': 'cacheprefix',
  1221. },
  1222. },
  1223. )
  1224. class PrefixedCacheUtils(CacheUtils):
  1225. pass
  1226. @override_settings(
  1227. CACHE_MIDDLEWARE_SECONDS=60,
  1228. CACHE_MIDDLEWARE_KEY_PREFIX='test',
  1229. CACHES={
  1230. 'default': {
  1231. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1232. },
  1233. },
  1234. )
  1235. class CacheHEADTest(TestCase):
  1236. def setUp(self):
  1237. self.path = '/cache/test/'
  1238. self.factory = RequestFactory()
  1239. def tearDown(self):
  1240. cache.clear()
  1241. def _set_cache(self, request, msg):
  1242. response = HttpResponse()
  1243. response.content = msg
  1244. return UpdateCacheMiddleware().process_response(request, response)
  1245. def test_head_caches_correctly(self):
  1246. test_content = 'test content'
  1247. request = self.factory.head(self.path)
  1248. request._cache_update_cache = True
  1249. self._set_cache(request, test_content)
  1250. request = self.factory.head(self.path)
  1251. request._cache_update_cache = True
  1252. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1253. self.assertIsNotNone(get_cache_data)
  1254. self.assertEqual(test_content.encode(), get_cache_data.content)
  1255. def test_head_with_cached_get(self):
  1256. test_content = 'test content'
  1257. request = self.factory.get(self.path)
  1258. request._cache_update_cache = True
  1259. self._set_cache(request, test_content)
  1260. request = self.factory.head(self.path)
  1261. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1262. self.assertIsNotNone(get_cache_data)
  1263. self.assertEqual(test_content.encode(), get_cache_data.content)
  1264. @override_settings(
  1265. CACHE_MIDDLEWARE_KEY_PREFIX='settingsprefix',
  1266. CACHES={
  1267. 'default': {
  1268. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1269. },
  1270. },
  1271. LANGUAGES=(
  1272. ('en', 'English'),
  1273. ('es', 'Spanish'),
  1274. ),
  1275. )
  1276. class CacheI18nTest(TestCase):
  1277. def setUp(self):
  1278. self.path = '/cache/test/'
  1279. self.factory = RequestFactory()
  1280. def tearDown(self):
  1281. cache.clear()
  1282. @override_settings(USE_I18N=True, USE_L10N=False, USE_TZ=False)
  1283. def test_cache_key_i18n_translation(self):
  1284. request = self.factory.get(self.path)
  1285. lang = translation.get_language()
  1286. response = HttpResponse()
  1287. key = learn_cache_key(request, response)
  1288. self.assertIn(lang, key, "Cache keys should include the language name when translation is active")
  1289. key2 = get_cache_key(request)
  1290. self.assertEqual(key, key2)
  1291. def check_accept_language_vary(self, accept_language, vary, reference_key):
  1292. request = self.factory.get(self.path)
  1293. request.META['HTTP_ACCEPT_LANGUAGE'] = accept_language
  1294. request.META['HTTP_ACCEPT_ENCODING'] = 'gzip;q=1.0, identity; q=0.5, *;q=0'
  1295. response = HttpResponse()
  1296. response['Vary'] = vary
  1297. key = learn_cache_key(request, response)
  1298. key2 = get_cache_key(request)
  1299. self.assertEqual(key, reference_key)
  1300. self.assertEqual(key2, reference_key)
  1301. @override_settings(USE_I18N=True, USE_L10N=False, USE_TZ=False)
  1302. def test_cache_key_i18n_translation_accept_language(self):
  1303. lang = translation.get_language()
  1304. self.assertEqual(lang, 'en')
  1305. request = self.factory.get(self.path)
  1306. request.META['HTTP_ACCEPT_ENCODING'] = 'gzip;q=1.0, identity; q=0.5, *;q=0'
  1307. response = HttpResponse()
  1308. response['Vary'] = 'accept-encoding'
  1309. key = learn_cache_key(request, response)
  1310. self.assertIn(lang, key, "Cache keys should include the language name when translation is active")
  1311. self.check_accept_language_vary(
  1312. 'en-us',
  1313. 'cookie, accept-language, accept-encoding',
  1314. key
  1315. )
  1316. self.check_accept_language_vary(
  1317. 'en-US',
  1318. 'cookie, accept-encoding, accept-language',
  1319. key
  1320. )
  1321. self.check_accept_language_vary(
  1322. 'en-US,en;q=0.8',
  1323. 'accept-encoding, accept-language, cookie',
  1324. key
  1325. )
  1326. self.check_accept_language_vary(
  1327. 'en-US,en;q=0.8,ko;q=0.6',
  1328. 'accept-language, cookie, accept-encoding',
  1329. key
  1330. )
  1331. self.check_accept_language_vary(
  1332. 'ko-kr,ko;q=0.8,en-us;q=0.5,en;q=0.3 ',
  1333. 'accept-encoding, cookie, accept-language',
  1334. key
  1335. )
  1336. self.check_accept_language_vary(
  1337. 'ko-KR,ko;q=0.8,en-US;q=0.6,en;q=0.4',
  1338. 'accept-language, accept-encoding, cookie',
  1339. key
  1340. )
  1341. self.check_accept_language_vary(
  1342. 'ko;q=1.0,en;q=0.5',
  1343. 'cookie, accept-language, accept-encoding',
  1344. key
  1345. )
  1346. self.check_accept_language_vary(
  1347. 'ko, en',
  1348. 'cookie, accept-encoding, accept-language',
  1349. key
  1350. )
  1351. self.check_accept_language_vary(
  1352. 'ko-KR, en-US',
  1353. 'accept-encoding, accept-language, cookie',
  1354. key
  1355. )
  1356. @override_settings(USE_I18N=False, USE_L10N=True, USE_TZ=False)
  1357. def test_cache_key_i18n_formatting(self):
  1358. request = self.factory.get(self.path)
  1359. lang = translation.get_language()
  1360. response = HttpResponse()
  1361. key = learn_cache_key(request, response)
  1362. self.assertIn(lang, key, "Cache keys should include the language name when formatting is active")
  1363. key2 = get_cache_key(request)
  1364. self.assertEqual(key, key2)
  1365. @override_settings(USE_I18N=False, USE_L10N=False, USE_TZ=True)
  1366. def test_cache_key_i18n_timezone(self):
  1367. request = self.factory.get(self.path)
  1368. # This is tightly coupled to the implementation,
  1369. # but it's the most straightforward way to test the key.
  1370. tz = force_text(timezone.get_current_timezone_name(), errors='ignore')
  1371. tz = tz.encode('ascii', 'ignore').decode('ascii').replace(' ', '_')
  1372. response = HttpResponse()
  1373. key = learn_cache_key(request, response)
  1374. self.assertIn(tz, key, "Cache keys should include the time zone name when time zones are active")
  1375. key2 = get_cache_key(request)
  1376. self.assertEqual(key, key2)
  1377. @override_settings(USE_I18N=False, USE_L10N=False)
  1378. def test_cache_key_no_i18n(self):
  1379. request = self.factory.get(self.path)
  1380. lang = translation.get_language()
  1381. tz = force_text(timezone.get_current_timezone_name(), errors='ignore')
  1382. tz = tz.encode('ascii', 'ignore').decode('ascii').replace(' ', '_')
  1383. response = HttpResponse()
  1384. key = learn_cache_key(request, response)
  1385. self.assertNotIn(lang, key, "Cache keys shouldn't include the language name when i18n isn't active")
  1386. self.assertNotIn(tz, key, "Cache keys shouldn't include the time zone name when i18n isn't active")
  1387. @override_settings(USE_I18N=False, USE_L10N=False, USE_TZ=True)
  1388. def test_cache_key_with_non_ascii_tzname(self):
  1389. # Regression test for #17476
  1390. class CustomTzName(timezone.UTC):
  1391. name = ''
  1392. def tzname(self, dt):
  1393. return self.name
  1394. request = self.factory.get(self.path)
  1395. response = HttpResponse()
  1396. with timezone.override(CustomTzName()):
  1397. CustomTzName.name = 'Hora estándar de Argentina'.encode('UTF-8') # UTF-8 string
  1398. sanitized_name = 'Hora_estndar_de_Argentina'
  1399. self.assertIn(sanitized_name, learn_cache_key(request, response),
  1400. "Cache keys should include the time zone name when time zones are active")
  1401. CustomTzName.name = 'Hora estándar de Argentina' # unicode
  1402. sanitized_name = 'Hora_estndar_de_Argentina'
  1403. self.assertIn(sanitized_name, learn_cache_key(request, response),
  1404. "Cache keys should include the time zone name when time zones are active")
  1405. @override_settings(
  1406. CACHE_MIDDLEWARE_KEY_PREFIX="test",
  1407. CACHE_MIDDLEWARE_SECONDS=60,
  1408. USE_ETAGS=True,
  1409. USE_I18N=True,
  1410. )
  1411. def test_middleware(self):
  1412. def set_cache(request, lang, msg):
  1413. translation.activate(lang)
  1414. response = HttpResponse()
  1415. response.content = msg
  1416. return UpdateCacheMiddleware().process_response(request, response)
  1417. # cache with non empty request.GET
  1418. request = self.factory.get(self.path, {'foo': 'bar', 'other': 'true'})
  1419. request._cache_update_cache = True
  1420. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1421. # first access, cache must return None
  1422. self.assertIsNone(get_cache_data)
  1423. response = HttpResponse()
  1424. content = 'Check for cache with QUERY_STRING'
  1425. response.content = content
  1426. UpdateCacheMiddleware().process_response(request, response)
  1427. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1428. # cache must return content
  1429. self.assertIsNotNone(get_cache_data)
  1430. self.assertEqual(get_cache_data.content, content.encode())
  1431. # different QUERY_STRING, cache must be empty
  1432. request = self.factory.get(self.path, {'foo': 'bar', 'somethingelse': 'true'})
  1433. request._cache_update_cache = True
  1434. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1435. self.assertIsNone(get_cache_data)
  1436. # i18n tests
  1437. en_message = "Hello world!"
  1438. es_message = "Hola mundo!"
  1439. request = self.factory.get(self.path)
  1440. request._cache_update_cache = True
  1441. set_cache(request, 'en', en_message)
  1442. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1443. # Check that we can recover the cache
  1444. self.assertIsNotNone(get_cache_data)
  1445. self.assertEqual(get_cache_data.content, en_message.encode())
  1446. # Check that we use etags
  1447. self.assertTrue(get_cache_data.has_header('ETag'))
  1448. # Check that we can disable etags
  1449. with self.settings(USE_ETAGS=False):
  1450. request._cache_update_cache = True
  1451. set_cache(request, 'en', en_message)
  1452. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1453. self.assertFalse(get_cache_data.has_header('ETag'))
  1454. # change the session language and set content
  1455. request = self.factory.get(self.path)
  1456. request._cache_update_cache = True
  1457. set_cache(request, 'es', es_message)
  1458. # change again the language
  1459. translation.activate('en')
  1460. # retrieve the content from cache
  1461. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1462. self.assertEqual(get_cache_data.content, en_message.encode())
  1463. # change again the language
  1464. translation.activate('es')
  1465. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1466. self.assertEqual(get_cache_data.content, es_message.encode())
  1467. # reset the language
  1468. translation.deactivate()
  1469. @override_settings(
  1470. CACHE_MIDDLEWARE_KEY_PREFIX="test",
  1471. CACHE_MIDDLEWARE_SECONDS=60,
  1472. USE_ETAGS=True,
  1473. )
  1474. def test_middleware_doesnt_cache_streaming_response(self):
  1475. request = self.factory.get(self.path)
  1476. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1477. self.assertIsNone(get_cache_data)
  1478. # This test passes on Python < 3.3 even without the corresponding code
  1479. # in UpdateCacheMiddleware, because pickling a StreamingHttpResponse
  1480. # fails (http://bugs.python.org/issue14288). LocMemCache silently
  1481. # swallows the exception and doesn't store the response in cache.
  1482. content = ['Check for cache with streaming content.']
  1483. response = StreamingHttpResponse(content)
  1484. UpdateCacheMiddleware().process_response(request, response)
  1485. get_cache_data = FetchFromCacheMiddleware().process_request(request)
  1486. self.assertIsNone(get_cache_data)
  1487. @override_settings(
  1488. CACHES={
  1489. 'default': {
  1490. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1491. 'KEY_PREFIX': 'cacheprefix'
  1492. },
  1493. },
  1494. )
  1495. class PrefixedCacheI18nTest(CacheI18nTest):
  1496. pass
  1497. def hello_world_view(request, value):
  1498. return HttpResponse('Hello World %s' % value)
  1499. def csrf_view(request):
  1500. return HttpResponse(csrf(request)['csrf_token'])
  1501. @override_settings(
  1502. CACHE_MIDDLEWARE_ALIAS='other',
  1503. CACHE_MIDDLEWARE_KEY_PREFIX='middlewareprefix',
  1504. CACHE_MIDDLEWARE_SECONDS=30,
  1505. CACHES={
  1506. 'default': {
  1507. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1508. },
  1509. 'other': {
  1510. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1511. 'LOCATION': 'other',
  1512. 'TIMEOUT': '1',
  1513. },
  1514. },
  1515. )
  1516. class CacheMiddlewareTest(TestCase):
  1517. def setUp(self):
  1518. super(CacheMiddlewareTest, self).setUp()
  1519. self.factory = RequestFactory()
  1520. self.default_cache = caches['default']
  1521. self.other_cache = caches['other']
  1522. def tearDown(self):
  1523. self.default_cache.clear()
  1524. self.other_cache.clear()
  1525. super(CacheMiddlewareTest, self).tearDown()
  1526. def test_constructor(self):
  1527. """
  1528. Ensure the constructor is correctly distinguishing between usage of CacheMiddleware as
  1529. Middleware vs. usage of CacheMiddleware as view decorator and setting attributes
  1530. appropriately.
  1531. """
  1532. # If no arguments are passed in construction, it's being used as middleware.
  1533. middleware = CacheMiddleware()
  1534. # Now test object attributes against values defined in setUp above
  1535. self.assertEqual(middleware.cache_timeout, 30)
  1536. self.assertEqual(middleware.key_prefix, 'middlewareprefix')
  1537. self.assertEqual(middleware.cache_alias, 'other')
  1538. # If arguments are being passed in construction, it's being used as a decorator.
  1539. # First, test with "defaults":
  1540. as_view_decorator = CacheMiddleware(cache_alias=None, key_prefix=None)
  1541. self.assertEqual(as_view_decorator.cache_timeout, 30) # Timeout value for 'default' cache, i.e. 30
  1542. self.assertEqual(as_view_decorator.key_prefix, '')
  1543. self.assertEqual(as_view_decorator.cache_alias, 'default') # Value of DEFAULT_CACHE_ALIAS from django.core.cache
  1544. # Next, test with custom values:
  1545. as_view_decorator_with_custom = CacheMiddleware(cache_timeout=60, cache_alias='other', key_prefix='foo')
  1546. self.assertEqual(as_view_decorator_with_custom.cache_timeout, 60)
  1547. self.assertEqual(as_view_decorator_with_custom.key_prefix, 'foo')
  1548. self.assertEqual(as_view_decorator_with_custom.cache_alias, 'other')
  1549. def test_middleware(self):
  1550. middleware = CacheMiddleware()
  1551. prefix_middleware = CacheMiddleware(key_prefix='prefix1')
  1552. timeout_middleware = CacheMiddleware(cache_timeout=1)
  1553. request = self.factory.get('/view/')
  1554. # Put the request through the request middleware
  1555. result = middleware.process_request(request)
  1556. self.assertIsNone(result)
  1557. response = hello_world_view(request, '1')
  1558. # Now put the response through the response middleware
  1559. response = middleware.process_response(request, response)
  1560. # Repeating the request should result in a cache hit
  1561. result = middleware.process_request(request)
  1562. self.assertIsNotNone(result)
  1563. self.assertEqual(result.content, b'Hello World 1')
  1564. # The same request through a different middleware won't hit
  1565. result = prefix_middleware.process_request(request)
  1566. self.assertIsNone(result)
  1567. # The same request with a timeout _will_ hit
  1568. result = timeout_middleware.process_request(request)
  1569. self.assertIsNotNone(result)
  1570. self.assertEqual(result.content, b'Hello World 1')
  1571. def test_view_decorator(self):
  1572. # decorate the same view with different cache decorators
  1573. default_view = cache_page(3)(hello_world_view)
  1574. default_with_prefix_view = cache_page(3, key_prefix='prefix1')(hello_world_view)
  1575. explicit_default_view = cache_page(3, cache='default')(hello_world_view)
  1576. explicit_default_with_prefix_view = cache_page(3, cache='default', key_prefix='prefix1')(hello_world_view)
  1577. other_view = cache_page(1, cache='other')(hello_world_view)
  1578. other_with_prefix_view = cache_page(1, cache='other', key_prefix='prefix2')(hello_world_view)
  1579. request = self.factory.get('/view/')
  1580. # Request the view once
  1581. response = default_view(request, '1')
  1582. self.assertEqual(response.content, b'Hello World 1')
  1583. # Request again -- hit the cache
  1584. response = default_view(request, '2')
  1585. self.assertEqual(response.content, b'Hello World 1')
  1586. # Requesting the same view with the explicit cache should yield the same result
  1587. response = explicit_default_view(request, '3')
  1588. self.assertEqual(response.content, b'Hello World 1')
  1589. # Requesting with a prefix will hit a different cache key
  1590. response = explicit_default_with_prefix_view(request, '4')
  1591. self.assertEqual(response.content, b'Hello World 4')
  1592. # Hitting the same view again gives a cache hit
  1593. response = explicit_default_with_prefix_view(request, '5')
  1594. self.assertEqual(response.content, b'Hello World 4')
  1595. # And going back to the implicit cache will hit the same cache
  1596. response = default_with_prefix_view(request, '6')
  1597. self.assertEqual(response.content, b'Hello World 4')
  1598. # Requesting from an alternate cache won't hit cache
  1599. response = other_view(request, '7')
  1600. self.assertEqual(response.content, b'Hello World 7')
  1601. # But a repeated hit will hit cache
  1602. response = other_view(request, '8')
  1603. self.assertEqual(response.content, b'Hello World 7')
  1604. # And prefixing the alternate cache yields yet another cache entry
  1605. response = other_with_prefix_view(request, '9')
  1606. self.assertEqual(response.content, b'Hello World 9')
  1607. # But if we wait a couple of seconds...
  1608. time.sleep(2)
  1609. # ... the default cache will still hit
  1610. caches['default']
  1611. response = default_view(request, '11')
  1612. self.assertEqual(response.content, b'Hello World 1')
  1613. # ... the default cache with a prefix will still hit
  1614. response = default_with_prefix_view(request, '12')
  1615. self.assertEqual(response.content, b'Hello World 4')
  1616. # ... the explicit default cache will still hit
  1617. response = explicit_default_view(request, '13')
  1618. self.assertEqual(response.content, b'Hello World 1')
  1619. # ... the explicit default cache with a prefix will still hit
  1620. response = explicit_default_with_prefix_view(request, '14')
  1621. self.assertEqual(response.content, b'Hello World 4')
  1622. # .. but a rapidly expiring cache won't hit
  1623. response = other_view(request, '15')
  1624. self.assertEqual(response.content, b'Hello World 15')
  1625. # .. even if it has a prefix
  1626. response = other_with_prefix_view(request, '16')
  1627. self.assertEqual(response.content, b'Hello World 16')
  1628. def test_sensitive_cookie_not_cached(self):
  1629. """
  1630. Django must prevent caching of responses that set a user-specific (and
  1631. maybe security sensitive) cookie in response to a cookie-less request.
  1632. """
  1633. csrf_middleware = CsrfViewMiddleware()
  1634. cache_middleware = CacheMiddleware()
  1635. request = self.factory.get('/view/')
  1636. self.assertIsNone(cache_middleware.process_request(request))
  1637. csrf_middleware.process_view(request, csrf_view, (), {})
  1638. response = csrf_view(request)
  1639. response = csrf_middleware.process_response(request, response)
  1640. response = cache_middleware.process_response(request, response)
  1641. # Inserting a CSRF cookie in a cookie-less request prevented caching.
  1642. self.assertIsNone(cache_middleware.process_request(request))
  1643. @override_settings(
  1644. CACHE_MIDDLEWARE_KEY_PREFIX='settingsprefix',
  1645. CACHE_MIDDLEWARE_SECONDS=1,
  1646. CACHES={
  1647. 'default': {
  1648. 'BACKEND': 'django.core.cache.backends.locmem.LocMemCache',
  1649. },
  1650. },
  1651. USE_I18N=False,
  1652. )
  1653. class TestWithTemplateResponse(TestCase):
  1654. """
  1655. Tests various headers w/ TemplateResponse.
  1656. Most are probably redundant since they manipulate the same object
  1657. anyway but the Etag header is 'special' because it relies on the
  1658. content being complete (which is not necessarily always the case
  1659. with a TemplateResponse)
  1660. """
  1661. def setUp(self):
  1662. self.path = '/cache/test/'
  1663. self.factory = RequestFactory()
  1664. def tearDown(self):
  1665. cache.clear()
  1666. def test_patch_vary_headers(self):
  1667. headers = (
  1668. # Initial vary, new headers, resulting vary.
  1669. (None, ('Accept-Encoding',), 'Accept-Encoding'),
  1670. ('Accept-Encoding', ('accept-encoding',), 'Accept-Encoding'),
  1671. ('Accept-Encoding', ('ACCEPT-ENCODING',), 'Accept-Encoding'),
  1672. ('Cookie', ('Accept-Encoding',), 'Cookie, Accept-Encoding'),
  1673. ('Cookie, Accept-Encoding', ('Accept-Encoding',), 'Cookie, Accept-Encoding'),
  1674. ('Cookie, Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  1675. (None, ('Accept-Encoding', 'COOKIE'), 'Accept-Encoding, COOKIE'),
  1676. ('Cookie, Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  1677. ('Cookie , Accept-Encoding', ('Accept-Encoding', 'cookie'), 'Cookie, Accept-Encoding'),
  1678. )
  1679. for initial_vary, newheaders, resulting_vary in headers:
  1680. response = TemplateResponse(HttpResponse(), Template("This is a test"))
  1681. if initial_vary is not None:
  1682. response['Vary'] = initial_vary
  1683. patch_vary_headers(response, newheaders)
  1684. self.assertEqual(response['Vary'], resulting_vary)
  1685. def test_get_cache_key(self):
  1686. request = self.factory.get(self.path)
  1687. response = TemplateResponse(HttpResponse(), Template("This is a test"))
  1688. key_prefix = 'localprefix'
  1689. # Expect None if no headers have been set yet.
  1690. self.assertIsNone(get_cache_key(request))
  1691. # Set headers to an empty list.
  1692. learn_cache_key(request, response)
  1693. self.assertEqual(
  1694. get_cache_key(request),
  1695. 'views.decorators.cache.cache_page.settingsprefix.GET.'
  1696. '58a0a05c8a5620f813686ff969c26853.d41d8cd98f00b204e9800998ecf8427e'
  1697. )
  1698. # Verify that a specified key_prefix is taken into account.
  1699. learn_cache_key(request, response, key_prefix=key_prefix)
  1700. self.assertEqual(
  1701. get_cache_key(request, key_prefix=key_prefix),
  1702. 'views.decorators.cache.cache_page.localprefix.GET.'
  1703. '58a0a05c8a5620f813686ff969c26853.d41d8cd98f00b204e9800998ecf8427e'
  1704. )
  1705. def test_get_cache_key_with_query(self):
  1706. request = self.factory.get(self.path, {'test': 1})
  1707. response = TemplateResponse(HttpResponse(), Template("This is a test"))
  1708. # Expect None if no headers have been set yet.
  1709. self.assertIsNone(get_cache_key(request))
  1710. # Set headers to an empty list.
  1711. learn_cache_key(request, response)
  1712. # Verify that the querystring is taken into account.
  1713. self.assertEqual(
  1714. get_cache_key(request),
  1715. 'views.decorators.cache.cache_page.settingsprefix.GET.'
  1716. '0f1c2d56633c943073c4569d9a9502fe.d41d8cd98f00b204e9800998ecf8427e'
  1717. )
  1718. @override_settings(USE_ETAGS=False)
  1719. def test_without_etag(self):
  1720. response = TemplateResponse(HttpResponse(), Template("This is a test"))
  1721. self.assertFalse(response.has_header('ETag'))
  1722. patch_response_headers(response)
  1723. self.assertFalse(response.has_header('ETag'))
  1724. response = response.render()
  1725. self.assertFalse(response.has_header('ETag'))
  1726. @override_settings(USE_ETAGS=True)
  1727. def test_with_etag(self):
  1728. response = TemplateResponse(HttpResponse(), Template("This is a test"))
  1729. self.assertFalse(response.has_header('ETag'))
  1730. patch_response_headers(response)
  1731. self.assertFalse(response.has_header('ETag'))
  1732. response = response.render()
  1733. self.assertTrue(response.has_header('ETag'))
  1734. class TestMakeTemplateFragmentKey(TestCase):
  1735. def test_without_vary_on(self):
  1736. key = make_template_fragment_key('a.fragment')
  1737. self.assertEqual(key, 'template.cache.a.fragment.d41d8cd98f00b204e9800998ecf8427e')
  1738. def test_with_one_vary_on(self):
  1739. key = make_template_fragment_key('foo', ['abc'])
  1740. self.assertEqual(key,
  1741. 'template.cache.foo.900150983cd24fb0d6963f7d28e17f72')
  1742. def test_with_many_vary_on(self):
  1743. key = make_template_fragment_key('bar', ['abc', 'def'])
  1744. self.assertEqual(key,
  1745. 'template.cache.bar.4b35f12ab03cec09beec4c21b2d2fa88')
  1746. def test_proper_escaping(self):
  1747. key = make_template_fragment_key('spam', ['abc:def%'])
  1748. self.assertEqual(key,
  1749. 'template.cache.spam.f27688177baec990cdf3fbd9d9c3f469')
  1750. class CacheHandlerTest(TestCase):
  1751. def test_same_instance(self):
  1752. """
  1753. Attempting to retrieve the same alias should yield the same instance.
  1754. """
  1755. cache1 = caches['default']
  1756. cache2 = caches['default']
  1757. self.assertIs(cache1, cache2)
  1758. def test_per_thread(self):
  1759. """
  1760. Requesting the same alias from separate threads should yield separate
  1761. instances.
  1762. """
  1763. c = []
  1764. def runner():
  1765. c.append(caches['default'])
  1766. for x in range(2):
  1767. t = threading.Thread(target=runner)
  1768. t.start()
  1769. t.join()
  1770. self.assertIsNot(c[0], c[1])