# 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[0-9]+)/$', InstanceModelView.as_view(resource=UserResource)), # url(r'^customusers/$', ListOrCreateModelView.as_view(resource=CustomUserResource), name='customusers'), # url(r'^customusers/(?P[0-9]+)/$', InstanceModelView.as_view(resource=CustomUserResource)), # url(r'^groups/$', ListOrCreateModelView.as_view(resource=GroupResource), name='groups'), # url(r'^groups/(?P[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)