summaryrefslogtreecommitdiff
path: root/tests/template_tests/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/template_tests/utils.py')
-rw-r--r--tests/template_tests/utils.py10
1 files changed, 6 insertions, 4 deletions
diff --git a/tests/template_tests/utils.py b/tests/template_tests/utils.py
index 168eca73f8..ba8dcdd46f 100644
--- a/tests/template_tests/utils.py
+++ b/tests/template_tests/utils.py
@@ -6,7 +6,7 @@ import functools
from django import template
from django.template import Library
-from django.template.base import Context
+from django.template.base import Context, libraries
from django.template.engine import Engine
from django.template.loader import get_template
from django.test.utils import override_settings
@@ -100,9 +100,11 @@ def upper(value):
def register_test_tags(func):
@functools.wraps(func)
def inner(self):
- template.libraries['testtags'] = register
- func(self)
- del template.libraries['testtags']
+ libraries['testtags'] = register
+ try:
+ func(self)
+ finally:
+ del libraries['testtags']
return inner