aboutsummaryrefslogtreecommitdiffstats
path: root/rest_framework/tests/testcases.py
diff options
context:
space:
mode:
authorTom Christie2012-10-30 14:32:31 +0000
committerTom Christie2012-10-30 14:32:31 +0000
commit9b30dab4f772f67a626e176dc4fae0a3ef9c2c81 (patch)
treeca138abf4792f58ffa28684f784f201ee1eef6d7 /rest_framework/tests/testcases.py
parent7e5b1501b5cede61a9391fb1a751d2ebcdb37031 (diff)
parent4e7805cb24d73e7f706318b5e5a27e3f9ba39d14 (diff)
downloaddjango-rest-framework-9b30dab4f772f67a626e176dc4fae0a3ef9c2c81.tar.bz2
Merge branch 'restframework2' into rest-framework-2-merge2.0.0
Conflicts: .gitignore .travis.yml AUTHORS README.rst djangorestframework/mixins.py djangorestframework/renderers.py djangorestframework/resources.py djangorestframework/serializer.py djangorestframework/templates/djangorestframework/base.html djangorestframework/templates/djangorestframework/login.html djangorestframework/templatetags/add_query_param.py djangorestframework/tests/accept.py djangorestframework/tests/authentication.py djangorestframework/tests/content.py djangorestframework/tests/reverse.py djangorestframework/tests/serializer.py djangorestframework/views.py docs/examples.rst docs/examples/blogpost.rst docs/examples/modelviews.rst docs/examples/objectstore.rst docs/examples/permissions.rst docs/examples/pygments.rst docs/examples/views.rst docs/howto/alternativeframeworks.rst docs/howto/mixin.rst docs/howto/reverse.rst docs/howto/usingurllib2.rst docs/index.rst docs/topics/release-notes.md examples/sandbox/views.py rest_framework/__init__.py rest_framework/compat.py rest_framework/utils/breadcrumbs.py setup.py
Diffstat (limited to 'rest_framework/tests/testcases.py')
-rw-r--r--rest_framework/tests/testcases.py63
1 files changed, 63 insertions, 0 deletions
diff --git a/rest_framework/tests/testcases.py b/rest_framework/tests/testcases.py
new file mode 100644
index 00000000..c90224aa
--- /dev/null
+++ b/rest_framework/tests/testcases.py
@@ -0,0 +1,63 @@
+# http://djangosnippets.org/snippets/1011/
+from django.conf import settings
+from django.core.management import call_command
+from django.db.models import loading
+from django.test import TestCase
+
+NO_SETTING = ('!', None)
+
+class TestSettingsManager(object):
+ """
+ A class which can modify some Django settings temporarily for a
+ test and then revert them to their original values later.
+
+ Automatically handles resyncing the DB if INSTALLED_APPS is
+ modified.
+
+ """
+ def __init__(self):
+ self._original_settings = {}
+
+ def set(self, **kwargs):
+ for k,v in kwargs.iteritems():
+ self._original_settings.setdefault(k, getattr(settings, k,
+ NO_SETTING))
+ setattr(settings, k, v)
+ if 'INSTALLED_APPS' in kwargs:
+ self.syncdb()
+
+ def syncdb(self):
+ loading.cache.loaded = False
+ call_command('syncdb', verbosity=0)
+
+ def revert(self):
+ for k,v in self._original_settings.iteritems():
+ if v == NO_SETTING:
+ delattr(settings, k)
+ else:
+ setattr(settings, k, v)
+ if 'INSTALLED_APPS' in self._original_settings:
+ self.syncdb()
+ self._original_settings = {}
+
+
+class SettingsTestCase(TestCase):
+ """
+ A subclass of the Django TestCase with a settings_manager
+ attribute which is an instance of TestSettingsManager.
+
+ Comes with a tearDown() method that calls
+ self.settings_manager.revert().
+
+ """
+ def __init__(self, *args, **kwargs):
+ super(SettingsTestCase, self).__init__(*args, **kwargs)
+ self.settings_manager = TestSettingsManager()
+
+ def tearDown(self):
+ self.settings_manager.revert()
+
+class TestModelsTestCase(SettingsTestCase):
+ def setUp(self, *args, **kwargs):
+ installed_apps = tuple(settings.INSTALLED_APPS) + ('rest_framework.tests',)
+ self.settings_manager.set(INSTALLED_APPS=installed_apps)