encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-01-24 08:14:16 +03:00
Code Issues Packages Projects Releases Wiki Activity

Updated tests to use url_path attribute in list and detail decorators

Browse Source
This commit is contained in:
tanwanirahul 2014-12-19 19:53:48 +05:30
parent 92ebeaa040
commit 2448cc8e85
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_routers.py
View File

@ -261,11 +261,11 @@ class DynamicListAndDetailViewSet(viewsets.ViewSet):
def detail_route_get(self, request, *args, **kwargs): def detail_route_get(self, request, *args, **kwargs):
return Response({'method': 'link2'}) return Response({'method': 'link2'})
@list_route(custom_method_name="list_custom-route") @list_route(url_path="list_custom-route")
def list_custom_route_get(self, request, *args, **kwargs): def list_custom_route_get(self, request, *args, **kwargs):
return Response({'method': 'link1'}) return Response({'method': 'link1'})
@detail_route(custom_method_name="detail_custom-route") @detail_route(url_path="detail_custom-route")
def detail_custom_route_get(self, request, *args, **kwargs): def detail_custom_route_get(self, request, *args, **kwargs):
return Response({'method': 'link2'}) return Response({'method': 'link2'})
@ -278,7 +278,7 @@ class TestDynamicListAndDetailRouter(TestCase):
routes = self.router.get_routes(DynamicListAndDetailViewSet) routes = self.router.get_routes(DynamicListAndDetailViewSet)
decorator_routes = [r for r in routes if not (r.name.endswith('-list') or r.name.endswith('-detail'))] decorator_routes = [r for r in routes if not (r.name.endswith('-list') or r.name.endswith('-detail'))]
MethodNamesMap = namedtuple('MethodNamesMap', 'method_name custom_method_name') MethodNamesMap = namedtuple('MethodNamesMap', 'method_name url_path')
# Make sure all these endpoints exist and none have been clobbered # Make sure all these endpoints exist and none have been clobbered
for i, endpoint in enumerate([MethodNamesMap('list_custom_route_get', 'list_custom-route'), for i, endpoint in enumerate([MethodNamesMap('list_custom_route_get', 'list_custom-route'),
MethodNamesMap('list_route_get', 'list_route_get'), MethodNamesMap('list_route_get', 'list_route_get'),
@ -290,14 +290,14 @@ class TestDynamicListAndDetailRouter(TestCase):
route = decorator_routes[i] route = decorator_routes[i]
# check url listing # check url listing
method_name = endpoint.method_name method_name = endpoint.method_name
custom_method_name = endpoint.custom_method_name url_path = endpoint.url_path
if method_name.startswith('list_'): if method_name.startswith('list_'):
self.assertEqual(route.url, self.assertEqual(route.url,
'^{{prefix}}/{0}{{trailing_slash}}$'.format(custom_method_name)) '^{{prefix}}/{0}{{trailing_slash}}$'.format(url_path))
else: else:
self.assertEqual(route.url, self.assertEqual(route.url,
'^{{prefix}}/{{lookup}}/{0}{{trailing_slash}}$'.format(custom_method_name)) '^{{prefix}}/{{lookup}}/{0}{{trailing_slash}}$'.format(url_path))
# check method to function mapping # check method to function mapping
if method_name.endswith('_post'): if method_name.endswith('_post'):
method_map = 'post' method_map = 'post'