entrouvert
/
debian-django-cachalot
17
0
Fork 0
Code Issues Pull Requests Releases Activity

Clears the cache on test DB creation and destruction.

This commit is contained in:
Bertrand Bordage 2014-09-29 22:03:05 +02:00
parent e39b47515f
commit fa0a44e0fc
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
cachalot/monkey_patch.py
View File

@ -234,8 +234,29 @@ def _unpatch_atomic():
Atomic.__exit__ = Atomic.__exit__.original
def _patch_test_db():
def patch(original):
def inner(*args, **kwargs):
django_cache.clear()
return original(*args, **kwargs)
inner.original = original
return inner
creation = connection.creation
creation.create_test_db = patch(creation.create_test_db)
creation.destroy_test_db = patch(creation.destroy_test_db)
def _unpatch_test_db():
creation = connection.creation
creation.create_test_db = creation.create_test_db.original
creation.destroy_test_db = creation.destroy_test_db.original
def patch():
global PATCHED
_patch_test_db()
_patch_orm_write()
_patch_orm_read()
_patch_atomic()
@ -244,6 +265,7 @@ def patch():
def unpatch():
global PATCHED
_unpatch_test_db()
_unpatch_orm_read()
_unpatch_orm_write()
_unpatch_atomic()