diff options
Diffstat (limited to 'rest_framework/tests/routers.py')
| -rw-r--r-- | rest_framework/tests/routers.py | 19 |
1 files changed, 12 insertions, 7 deletions
diff --git a/rest_framework/tests/routers.py b/rest_framework/tests/routers.py index 4e4765cb..9f9ced0f 100644 --- a/rest_framework/tests/routers.py +++ b/rest_framework/tests/routers.py @@ -23,6 +23,10 @@ class BasicViewSet(viewsets.ViewSet): 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'}) @@ -40,16 +44,17 @@ class TestSimpleRouter(TestCase): 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', 'link1', 'link2']): + for i, endpoint in enumerate(['action1', 'action2', 'action3', 'link1', 'link2']): route = decorator_routes[i] # check url listing self.assertEqual(route.url, '^{{prefix}}/{{lookup}}/{0}/$'.format(endpoint)) # check method to function mapping - if endpoint.startswith('action'): - method_map = 'post' + if endpoint == 'action3': + methods_map = ['post', 'delete'] + elif endpoint.startswith('action'): + methods_map = ['post'] else: - method_map = 'get' - self.assertEqual(route.mapping[method_map], endpoint) - - + methods_map = ['get'] + for method in methods_map: + self.assertEqual(route.mapping[method], endpoint) |