d558a36127
Reset the correct URLs module to fix test failures. Fix third party auth to reconfigure the registry on test cleanup.
57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
import sys
|
|
|
|
from django.conf import settings
|
|
from django.core.urlresolvers import clear_url_caches, resolve
|
|
|
|
|
|
class UrlResetMixin(object):
|
|
"""Mixin to reset urls.py before and after a test
|
|
|
|
Django memoizes the function that reads the urls module (whatever module
|
|
urlconf names). The module itself is also stored by python in sys.modules.
|
|
To fully reload it, we need to reload the python module, and also clear django's
|
|
cache of the parsed urls.
|
|
|
|
However, the order in which we do this doesn't matter, because neither one will
|
|
get reloaded until the next request
|
|
|
|
Doing this is expensive, so it should only be added to tests that modify settings
|
|
that affect the contents of urls.py
|
|
"""
|
|
|
|
def _reset_urls(self, urlconf_modules):
|
|
for urlconf in urlconf_modules:
|
|
if urlconf in sys.modules:
|
|
reload(sys.modules[urlconf])
|
|
clear_url_caches()
|
|
|
|
# Resolve a URL so that the new urlconf gets loaded
|
|
resolve('/')
|
|
|
|
def setUp(self, *args, **kwargs):
|
|
"""Reset Django urls before tests and after tests
|
|
|
|
If you need to reset `urls.py` from a particular Django app (or apps),
|
|
specify these modules in *args.
|
|
|
|
Examples:
|
|
|
|
# Reload only the root urls.py
|
|
super(MyTestCase, self).setUp()
|
|
|
|
# Reload urls from my_app
|
|
super(MyTestCase, self).setUp("my_app.urls")
|
|
|
|
# Reload urls from my_app and another_app
|
|
super(MyTestCase, self).setUp("my_app.urls", "another_app.urls")
|
|
|
|
"""
|
|
super(UrlResetMixin, self).setUp(**kwargs)
|
|
|
|
urlconf_modules = [settings.ROOT_URLCONF]
|
|
if args:
|
|
urlconf_modules.extend(args)
|
|
|
|
self._reset_urls(urlconf_modules)
|
|
self.addCleanup(lambda: self._reset_urls(urlconf_modules))
|