django-rest-framework/rest_framework/tests/modelviews.py

# from rest_framework.compat import patterns, url
# from django.forms import ModelForm
# from django.contrib.auth.models import Group, User
# from rest_framework.resources import ModelResource
# from rest_framework.views import ListOrCreateModelView, InstanceModelView
# from rest_framework.tests.models import CustomUser
# from rest_framework.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 rest_framework provides"""
#     urls = 'rest_framework.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)