diff options
Diffstat (limited to 'djangorestframework/tests/modelviews.py')
| -rw-r--r-- | djangorestframework/tests/modelviews.py | 90 |
1 files changed, 0 insertions, 90 deletions
diff --git a/djangorestframework/tests/modelviews.py b/djangorestframework/tests/modelviews.py deleted file mode 100644 index ccd8513f..00000000 --- a/djangorestframework/tests/modelviews.py +++ /dev/null @@ -1,90 +0,0 @@ -from django.conf.urls.defaults import patterns, url -from django.forms import ModelForm -from django.contrib.auth.models import Group, User -from djangorestframework.resources import ModelResource -from djangorestframework.views import ListOrCreateModelView, InstanceModelView -from djangorestframework.tests.models import CustomUser -from djangorestframework.tests.testcases import TestModelsTestCase - - -class GroupResource(ModelResource): - model = Group - - -class UserForm(ModelForm): - class Meta: - model = User - exclude = ('last_login', 'date_joined') - - -class UserResource(ModelResource): - model = User - form = UserForm - - -class CustomUserResource(ModelResource): - model = CustomUser - -urlpatterns = patterns('', - url(r'^users/$', ListOrCreateModelView.as_view(resource=UserResource), name='users'), - url(r'^users/(?P<id>[0-9]+)/$', InstanceModelView.as_view(resource=UserResource)), - url(r'^customusers/$', ListOrCreateModelView.as_view(resource=CustomUserResource), name='customusers'), - url(r'^customusers/(?P<id>[0-9]+)/$', InstanceModelView.as_view(resource=CustomUserResource)), - url(r'^groups/$', ListOrCreateModelView.as_view(resource=GroupResource), name='groups'), - url(r'^groups/(?P<id>[0-9]+)/$', InstanceModelView.as_view(resource=GroupResource)), -) - - -class ModelViewTests(TestModelsTestCase): - """Test the model views djangorestframework provides""" - urls = 'djangorestframework.tests.modelviews' - - def test_creation(self): - """Ensure that a model object can be created""" - self.assertEqual(0, Group.objects.count()) - - response = self.client.post('/groups/', {'name': 'foo'}) - - self.assertEqual(response.status_code, 201) - self.assertEqual(1, Group.objects.count()) - self.assertEqual('foo', Group.objects.all()[0].name) - - def test_creation_with_m2m_relation(self): - """Ensure that a model object with a m2m relation can be created""" - group = Group(name='foo') - group.save() - self.assertEqual(0, User.objects.count()) - - response = self.client.post('/users/', {'username': 'bar', 'password': 'baz', 'groups': [group.id]}) - - self.assertEqual(response.status_code, 201) - self.assertEqual(1, User.objects.count()) - - user = User.objects.all()[0] - self.assertEqual('bar', user.username) - self.assertEqual('baz', user.password) - self.assertEqual(1, user.groups.count()) - - group = user.groups.all()[0] - self.assertEqual('foo', group.name) - - def test_creation_with_m2m_relation_through(self): - """ - Ensure that a model object with a m2m relation can be created where that - relation uses a through table - """ - group = Group(name='foo') - group.save() - self.assertEqual(0, User.objects.count()) - - response = self.client.post('/customusers/', {'username': 'bar', 'groups': [group.id]}) - - self.assertEqual(response.status_code, 201) - self.assertEqual(1, CustomUser.objects.count()) - - user = CustomUser.objects.all()[0] - self.assertEqual('bar', user.username) - self.assertEqual(1, user.groups.count()) - - group = user.groups.all()[0] - self.assertEqual('foo', group.name) |
