Commit e93d112c by Calen Pennington

Make UrlResetMixin.reset_urls a public method (so that it can be used inside test methods)

parent acc9bca3
...@@ -31,8 +31,14 @@ class UrlResetMixin(object): ...@@ -31,8 +31,14 @@ class UrlResetMixin(object):
URLCONF_MODULES = None URLCONF_MODULES = None
def _reset_urls(self, urlconf_modules): def reset_urls(self, urlconf_modules=None):
"""Reset `urls.py` for a set of Django apps.""" """Reset `urls.py` for a set of Django apps."""
if urlconf_modules is None:
urlconf_modules = [settings.ROOT_URLCONF]
if self.URLCONF_MODULES is not None:
urlconf_modules.extend(self.URLCONF_MODULES)
for urlconf in urlconf_modules: for urlconf in urlconf_modules:
if urlconf in sys.modules: if urlconf in sys.modules:
reload(sys.modules[urlconf]) reload(sys.modules[urlconf])
...@@ -61,12 +67,8 @@ class UrlResetMixin(object): ...@@ -61,12 +67,8 @@ class UrlResetMixin(object):
""" """
super(UrlResetMixin, self).setUp() super(UrlResetMixin, self).setUp()
urlconf_modules = [settings.ROOT_URLCONF] self.reset_urls()
if self.URLCONF_MODULES is not None: self.addCleanup(self.reset_urls)
urlconf_modules.extend(self.URLCONF_MODULES)
self._reset_urls(urlconf_modules)
self.addCleanup(lambda: self._reset_urls(urlconf_modules))
class EventTestMixin(object): class EventTestMixin(object):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment