mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-24 02:24:03 +03:00
263 lines
8.7 KiB
Python
263 lines
8.7 KiB
Python
from __future__ import unicode_literals
|
|
from django.db import models
|
|
from django.test import TestCase
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from rest_framework import serializers, viewsets, permissions
|
|
from rest_framework.compat import include, patterns, url
|
|
from rest_framework.decorators import link, action, collection_link, collection_action
|
|
from rest_framework.response import Response
|
|
from rest_framework.routers import SimpleRouter, DefaultRouter
|
|
from rest_framework.test import APIRequestFactory
|
|
|
|
factory = APIRequestFactory()
|
|
|
|
urlpatterns = patterns('',)
|
|
|
|
|
|
class BasicViewSet(viewsets.ViewSet):
|
|
def list(self, request, *args, **kwargs):
|
|
return Response({'method': 'list'})
|
|
|
|
@action()
|
|
def action1(self, request, *args, **kwargs):
|
|
return Response({'method': 'action1'})
|
|
|
|
@action()
|
|
def action2(self, request, *args, **kwargs):
|
|
return Response({'method': 'action2'})
|
|
|
|
@action(methods=['post', 'delete'])
|
|
def action3(self, request, *args, **kwargs):
|
|
return Response({'method': 'action2'})
|
|
|
|
@link()
|
|
def link1(self, request, *args, **kwargs):
|
|
return Response({'method': 'link1'})
|
|
|
|
@link()
|
|
def link2(self, request, *args, **kwargs):
|
|
return Response({'method': 'link2'})
|
|
|
|
|
|
class TestSimpleRouter(TestCase):
|
|
def setUp(self):
|
|
self.router = SimpleRouter()
|
|
|
|
def test_link_and_action_decorator(self):
|
|
routes = self.router.get_routes(BasicViewSet)
|
|
decorator_routes = routes[2:]
|
|
# Make sure all these endpoints exist and none have been clobbered
|
|
for i, endpoint in enumerate(['action1', 'action2', 'action3', 'link1', 'link2']):
|
|
route = decorator_routes[i]
|
|
# check url listing
|
|
self.assertEqual(route.url,
|
|
'^{{prefix}}/{{lookup}}/{0}{{trailing_slash}}$'.format(endpoint))
|
|
# check method to function mapping
|
|
if endpoint == 'action3':
|
|
methods_map = ['post', 'delete']
|
|
elif endpoint.startswith('action'):
|
|
methods_map = ['post']
|
|
else:
|
|
methods_map = ['get']
|
|
for method in methods_map:
|
|
self.assertEqual(route.mapping[method], endpoint)
|
|
|
|
|
|
class RouterTestModel(models.Model):
|
|
uuid = models.CharField(max_length=20)
|
|
text = models.CharField(max_length=200)
|
|
|
|
|
|
class TestCustomLookupFields(TestCase):
|
|
"""
|
|
Ensure that custom lookup fields are correctly routed.
|
|
"""
|
|
urls = 'rest_framework.tests.test_routers'
|
|
|
|
def setUp(self):
|
|
class NoteSerializer(serializers.HyperlinkedModelSerializer):
|
|
class Meta:
|
|
model = RouterTestModel
|
|
lookup_field = 'uuid'
|
|
fields = ('url', 'uuid', 'text')
|
|
|
|
class NoteViewSet(viewsets.ModelViewSet):
|
|
queryset = RouterTestModel.objects.all()
|
|
serializer_class = NoteSerializer
|
|
lookup_field = 'uuid'
|
|
|
|
RouterTestModel.objects.create(uuid='123', text='foo bar')
|
|
|
|
self.router = SimpleRouter()
|
|
self.router.register(r'notes', NoteViewSet)
|
|
|
|
from rest_framework.tests import test_routers
|
|
urls = getattr(test_routers, 'urlpatterns')
|
|
urls += patterns('',
|
|
url(r'^', include(self.router.urls)),
|
|
)
|
|
|
|
def test_custom_lookup_field_route(self):
|
|
detail_route = self.router.urls[-1]
|
|
detail_url_pattern = detail_route.regex.pattern
|
|
self.assertIn('<uuid>', detail_url_pattern)
|
|
|
|
def test_retrieve_lookup_field_list_view(self):
|
|
response = self.client.get('/notes/')
|
|
self.assertEqual(response.data,
|
|
[{
|
|
"url": "http://testserver/notes/123/",
|
|
"uuid": "123", "text": "foo bar"
|
|
}]
|
|
)
|
|
|
|
def test_retrieve_lookup_field_detail_view(self):
|
|
response = self.client.get('/notes/123/')
|
|
self.assertEqual(response.data,
|
|
{
|
|
"url": "http://testserver/notes/123/",
|
|
"uuid": "123", "text": "foo bar"
|
|
}
|
|
)
|
|
|
|
|
|
class TestTrailingSlashIncluded(TestCase):
|
|
def setUp(self):
|
|
class NoteViewSet(viewsets.ModelViewSet):
|
|
model = RouterTestModel
|
|
|
|
self.router = SimpleRouter()
|
|
self.router.register(r'notes', NoteViewSet)
|
|
self.urls = self.router.urls
|
|
|
|
def test_urls_have_trailing_slash_by_default(self):
|
|
expected = ['^notes/$', '^notes/(?P<pk>[^/]+)/$']
|
|
for idx in range(len(expected)):
|
|
self.assertEqual(expected[idx], self.urls[idx].regex.pattern)
|
|
|
|
|
|
class TestTrailingSlashRemoved(TestCase):
|
|
def setUp(self):
|
|
class NoteViewSet(viewsets.ModelViewSet):
|
|
model = RouterTestModel
|
|
|
|
self.router = SimpleRouter(trailing_slash=False)
|
|
self.router.register(r'notes', NoteViewSet)
|
|
self.urls = self.router.urls
|
|
|
|
def test_urls_can_have_trailing_slash_removed(self):
|
|
expected = ['^notes$', '^notes/(?P<pk>[^/]+)$']
|
|
for idx in range(len(expected)):
|
|
self.assertEqual(expected[idx], self.urls[idx].regex.pattern)
|
|
|
|
|
|
class TestNameableRoot(TestCase):
|
|
def setUp(self):
|
|
class NoteViewSet(viewsets.ModelViewSet):
|
|
model = RouterTestModel
|
|
self.router = DefaultRouter()
|
|
self.router.root_view_name = 'nameable-root'
|
|
self.router.register(r'notes', NoteViewSet)
|
|
self.urls = self.router.urls
|
|
|
|
def test_router_has_custom_name(self):
|
|
expected = 'nameable-root'
|
|
self.assertEqual(expected, self.urls[0].name)
|
|
|
|
|
|
class TestActionKeywordArgs(TestCase):
|
|
"""
|
|
Ensure keyword arguments passed in the `@action` decorator
|
|
are properly handled. Refs #940.
|
|
"""
|
|
|
|
def setUp(self):
|
|
class TestViewSet(viewsets.ModelViewSet):
|
|
permission_classes = []
|
|
|
|
@action(permission_classes=[permissions.AllowAny])
|
|
def custom(self, request, *args, **kwargs):
|
|
return Response({
|
|
'permission_classes': self.permission_classes
|
|
})
|
|
|
|
self.router = SimpleRouter()
|
|
self.router.register(r'test', TestViewSet, base_name='test')
|
|
self.view = self.router.urls[-1].callback
|
|
|
|
def test_action_kwargs(self):
|
|
request = factory.post('/test/0/custom/')
|
|
response = self.view(request)
|
|
self.assertEqual(
|
|
response.data,
|
|
{'permission_classes': [permissions.AllowAny]}
|
|
)
|
|
|
|
|
|
class TestActionAppliedToExistingRoute(TestCase):
|
|
"""
|
|
Ensure `@action` decorator raises an except when applied
|
|
to an existing route
|
|
"""
|
|
|
|
def test_exception_raised_when_action_applied_to_existing_route(self):
|
|
class TestViewSet(viewsets.ModelViewSet):
|
|
|
|
@action()
|
|
def retrieve(self, request, *args, **kwargs):
|
|
return Response({
|
|
'hello': 'world'
|
|
})
|
|
|
|
self.router = SimpleRouter()
|
|
self.router.register(r'test', TestViewSet, base_name='test')
|
|
|
|
with self.assertRaises(ImproperlyConfigured):
|
|
self.router.urls
|
|
|
|
|
|
class StaticAndDynamicViewSet(viewsets.ViewSet):
|
|
def list(self, request, *args, **kwargs):
|
|
return Response({'method': 'list'})
|
|
|
|
@collection_action()
|
|
def collection_action(self, request, *args, **kwargs):
|
|
return Response({'method': 'action1'})
|
|
|
|
@action()
|
|
def dynamic_action(self, request, *args, **kwargs):
|
|
return Response({'method': 'action2'})
|
|
|
|
@collection_link()
|
|
def collection_link(self, request, *args, **kwargs):
|
|
return Response({'method': 'link1'})
|
|
|
|
@link()
|
|
def dynamic_link(self, request, *args, **kwargs):
|
|
return Response({'method': 'link2'})
|
|
|
|
|
|
class TestStaticAndDynamicRouter(TestCase):
|
|
def setUp(self):
|
|
self.router = SimpleRouter()
|
|
|
|
def test_link_and_action_decorator(self):
|
|
routes = self.router.get_routes(StaticAndDynamicViewSet)
|
|
decorator_routes = [r for r in routes if not (r.name.endswith('-list') or r.name.endswith('-detail'))]
|
|
# Make sure all these endpoints exist and none have been clobbered
|
|
for i, endpoint in enumerate(['collection_action', 'collection_link', 'dynamic_action', 'dynamic_link']):
|
|
route = decorator_routes[i]
|
|
# check url listing
|
|
if endpoint.startswith('collection_'):
|
|
self.assertEqual(route.url,
|
|
'^{{prefix}}/{0}{{trailing_slash}}$'.format(endpoint))
|
|
else:
|
|
self.assertEqual(route.url,
|
|
'^{{prefix}}/{{lookup}}/{0}{{trailing_slash}}$'.format(endpoint))
|
|
# check method to function mapping
|
|
if endpoint.endswith('action'):
|
|
method_map = 'post'
|
|
else:
|
|
method_map = 'get'
|
|
self.assertEqual(route.mapping[method_map], endpoint)
|