Merge branch 'support-viewsets' into local.

Author: Marcos Dione

rest_framework_docs/api_endpoint.py

@@ -1,9 +1,17 @@
 import json
 import inspect
+
 from django.contrib.admindocs.views import simplify_regex
 from django.utils.encoding import force_str
+
+from rest_framework.viewsets import ModelViewSet
 from rest_framework.serializers import BaseSerializer
 
+VIEWSET_METHODS = {
+    'List': ['get', 'post'],
+    'Instance': ['get', 'put', 'patch', 'delete'],
+}
+
 
 class ApiEndpoint(object):
 
@@ -31,8 +39,14 @@ def __get_path__(self, parent_regex):
             return "/{0}{1}".format(self.name_parent, simplify_regex(self.pattern.regex.pattern))
         return simplify_regex(self.pattern.regex.pattern)
 
-    def __get_allowed_methods__(self):
+    def is_method_allowed(self, callback_cls, method_name):
+        has_attr = hasattr(callback_cls, method_name)
+        viewset_method = (issubclass(callback_cls, ModelViewSet) and
+                          method_name in VIEWSET_METHODS.get(self.callback.suffix, []))
+
+        return has_attr or viewset_method
 
+    def __get_allowed_methods__(self):
         viewset_methods = []
         if self.drf_router:
             for prefix, viewset, basename in self.drf_router.registry:
@@ -57,14 +71,18 @@ def __get_allowed_methods__(self):
                     )
                     if self.pattern.regex.pattern == regex:
                         funcs, viewset_methods = zip(
-                            *[(mapping[m], m.upper()) for m in self.callback.cls.http_method_names if m in mapping]
+                            *[(mapping[m], m.upper())
+                              for m in self.callback.cls.http_method_names
+                              if m in mapping]
                         )
                         viewset_methods = list(viewset_methods)
                         if len(set(funcs)) == 1:
                             self.docstring = inspect.getdoc(getattr(self.callback.cls, funcs[0]))
 
-        view_methods = [force_str(m).upper() for m in self.callback.cls.http_method_names if hasattr(self.callback.cls, m)]
-        return viewset_methods + view_methods
+        view_methods = [force_str(m).upper()
+                        for m in self.callback.cls.http_method_names
+                        if self.is_method_allowed(self.callback.cls, m)]
+        return sorted(viewset_methods + view_methods)
 
     def __get_docstring__(self):
         return inspect.getdoc(self.callback)

tests/tests.py

@@ -31,7 +31,7 @@ def test_index_view_with_endpoints(self):
 
         # Test the login view
         self.assertEqual(response.context["endpoints"][0].name_parent, "accounts")
-        self.assertEqual(response.context["endpoints"][0].allowed_methods, ['POST', 'OPTIONS'])
+        self.assertEqual(sorted(response.context["endpoints"][0].allowed_methods), sorted(['OPTIONS', 'POST']))
         self.assertEqual(response.context["endpoints"][0].path, "/accounts/login/")
         self.assertEqual(response.context["endpoints"][0].docstring, "A view that allows users to login providing their username and password.")
         self.assertEqual(len(response.context["endpoints"][0].fields), 2)