Hopiu/django-cachalot
1
0
Fork 0
mirror of https://github.com/Hopiu/django-cachalot.git synced 2026-05-27 05:13:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Tries to fix cache flushes in tests.

Browse source
This commit is contained in:
Bertrand Bordage 2014-10-14 01:40:08 +02:00
parent 511ab26e84
commit 0f6e7d95e5
1 changed files with 19 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
cachalot/monkey_patch.py
View file

@ -257,25 +257,38 @@ def _patch_atomic():
Atomic.__exit__ = patch_exit(Atomic.__exit__)
def _patch_test_teardown():
def patch_teardown(original):
def _patch_tests():
def patch_before(original):
@wraps(original)
def inner(*args, **kwargs):
original(*args, **kwargs)
clear_all_caches()
return original(*args, **kwargs)
inner.original = original
return inner
TransactionTestCase._fixture_setup = patch_teardown(
def patch_after(original):
@wraps(original)
def inner(*args, **kwargs):
out = original(*args, **kwargs)
clear_all_caches()
return out
inner.original = original
return inner
creation = connection.creation
creation.create_test_db = patch_after(creation.create_test_db)
creation.destroy_test_db = patch_before(creation.destroy_test_db)
TransactionTestCase._fixture_setup = patch_after(
TransactionTestCase._fixture_setup)
TransactionTestCase._fixture_teardown = patch_teardown(
TransactionTestCase._fixture_teardown = patch_after(
TransactionTestCase._fixture_teardown)
def patch():
global PATCHED
_patch_test_teardown()
_patch_tests()
_patch_orm_write()
_patch_orm_read()
_patch_atomic()