aboutsummaryrefslogtreecommitdiffstats
path: root/rest_framework/tests/testcases.py
diff options
context:
space:
mode:
authorTom Christie2012-12-19 23:12:27 +0000
committerTom Christie2012-12-19 23:12:27 +0000
commita493c83248535c9fa7f78815b16bce7e88bf7966 (patch)
treeb52e34b8b6ecfcc08f0135bc0a503b939e53a3ec /rest_framework/tests/testcases.py
parentc097bcef586da4513d1d6f357d9eb3d7b4b0fffb (diff)
downloaddjango-rest-framework-a493c83248535c9fa7f78815b16bce7e88bf7966.tar.bz2
urls, patterns, include imports move to compat to support incoming 1.3 thru 1.6 import compatability
Diffstat (limited to 'rest_framework/tests/testcases.py')
-rw-r--r--rest_framework/tests/testcases.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/rest_framework/tests/testcases.py b/rest_framework/tests/testcases.py
index c90224aa..97f492ff 100644
--- a/rest_framework/tests/testcases.py
+++ b/rest_framework/tests/testcases.py
@@ -6,6 +6,7 @@ from django.test import TestCase
NO_SETTING = ('!', None)
+
class TestSettingsManager(object):
"""
A class which can modify some Django settings temporarily for a
@@ -19,7 +20,7 @@ class TestSettingsManager(object):
self._original_settings = {}
def set(self, **kwargs):
- for k,v in kwargs.iteritems():
+ for k, v in kwargs.iteritems():
self._original_settings.setdefault(k, getattr(settings, k,
NO_SETTING))
setattr(settings, k, v)
@@ -31,7 +32,7 @@ class TestSettingsManager(object):
call_command('syncdb', verbosity=0)
def revert(self):
- for k,v in self._original_settings.iteritems():
+ for k, v in self._original_settings.iteritems():
if v == NO_SETTING:
delattr(settings, k)
else:
@@ -57,6 +58,7 @@ class SettingsTestCase(TestCase):
def tearDown(self):
self.settings_manager.revert()
+
class TestModelsTestCase(SettingsTestCase):
def setUp(self, *args, **kwargs):
installed_apps = tuple(settings.INSTALLED_APPS) + ('rest_framework.tests',)