aboutsummaryrefslogtreecommitdiffstats
path: root/rest_framework/tests/testcases.py
diff options
context:
space:
mode:
authorJamie Matthews2012-09-26 13:05:21 +0100
committerJamie Matthews2012-09-26 13:05:21 +0100
commit01770c53cd9045e6ea054f32b1e40b5d2ff7fe44 (patch)
tree657cb66f92d78add3b2f587754387832043168e6 /rest_framework/tests/testcases.py
parentf6488cb0589d3b11fb8d831e00d1389f3fff74b6 (diff)
parent09a445b257532be69ffab69a3f62b84bfa90463d (diff)
downloaddjango-rest-framework-01770c53cd9045e6ea054f32b1e40b5d2ff7fe44.tar.bz2
Merge branch 'restframework2' of git://github.com/tomchristie/django-rest-framework into improved-view-decorators
* 'restframework2' of git://github.com/tomchristie/django-rest-framework: (56 commits) Bits of cleanup Add request.QUERY_PARAMS Add readonly 'id' field Tweak browseable API Don't display readonly fields Fix some bits of serialization Add csrf note Fix incorrect bit of tutorial Added tox.ini Tweak media_type -> accepted_media_type. Need to document, but marginally less confusing Tweak media_type -> accepted_media_type. Need to document, but marginally less confusing Tweak media_type -> accepted_media_type. Need to document, but marginally less confusing Clean up bits of templates etc Hack out bunch of unneccesary private methods on View class Clean up template tags Remove dumbass __all__ variables Remove old 'djangorestframework directories Change package name: djangorestframework -> rest_framework Dont strip final '/' Use get_script_prefix to play nicely if not installed at the root. ... Conflicts: rest_framework/decorators.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)