mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-12-04 23:44:07 +03:00
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
|
# from django.conf.urls.defaults 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)
|