Test decorated extra action

Author: rpkilby

tests/test_viewsets.py

@@ -33,6 +33,12 @@ class Action(models.Model):
     pass
 
 
+def decorate(fn):
+    def wrapper(self, request, *args, **kwargs):
+        return fn(self, request, *args, **kwargs)
+    return wrapper
+
+
 class ActionViewSet(GenericViewSet):
     queryset = Action.objects.all()
 
@@ -62,6 +68,11 @@ def custom_detail_action(self, request, *args, **kwargs):
     def unresolvable_detail_action(self, request, *args, **kwargs):
         raise NotImplementedError
 
+    @action(detail=False)
+    @decorate
+    def wrapped(self, request, *args, **kwargs):
+        raise NotImplementedError
+
 
 class ActionNamesViewSet(GenericViewSet):
 
@@ -157,6 +168,7 @@ def test_extra_actions(self):
             'detail_action',
             'list_action',
             'unresolvable_detail_action',
+            'wrapped',
         ]
 
         self.assertEqual(actual, expected)
@@ -172,6 +184,7 @@ def test_list_view(self):
         expected = OrderedDict([
             ('Custom list action', 'http://testserver/api/actions/custom_list_action/'),
             ('List action', 'http://testserver/api/actions/list_action/'),
+            ('Wrapped', 'http://testserver/api/actions/wrapped/'),
         ])
 
         self.assertEqual(view.get_extra_action_url_map(), expected)