Prepare CurrentUserDefault like DRF would do it

Author: ticosax

rest_framework_swagger/introspectors.py

@@ -18,6 +18,11 @@
 import rest_framework
 from rest_framework import viewsets
 from rest_framework.compat import apply_markdown
+try:
+    from rest_framework.fields import CurrentUserDefault
+except ImportError:
+    # FIXME once we drop support of DRF 2.x .
+    CurrentUserDefault = None
 from rest_framework.utils import formatting
 from django.utils import six
 try:
@@ -43,6 +48,9 @@ def get_default_value(field):
         if default_value == empty:
             default_value = None
     if callable(default_value):
+        if CurrentUserDefault is not None and isinstance(default_value,
+                                                         CurrentUserDefault):
+            default_value.user = None
         default_value = default_value()
     return default_value
 

rest_framework_swagger/tests.py

@@ -5,6 +5,10 @@
 import os.path
 from mock import Mock, patch
 from distutils.version import StrictVersion
+try:
+    from unittest.case import SkipTest
+except ImportError:
+    from unittest2.case import SkipTest
 
 from django.core.urlresolvers import RegexURLResolver, RegexURLPattern
 from django.conf import settings
@@ -485,7 +489,8 @@ class SerializedAPI(ListCreateAPIView):
         delta = datetime.timedelta(seconds=1)
         self.assertAlmostEqual(value, datetime.datetime.now(), delta=delta)
 
-    def test_get_models_ordering(self):
+    def test_get_models_ordering_drf2(self):
+
         class SerializedAPI(ListCreateAPIView):
             serializer_class = CommentSerializer
 
@@ -501,6 +506,35 @@ class SerializedAPI(ListCreateAPIView):
             ["email", "content", "created"],
             list(models['CommentSerializer']['properties'].keys()))
 
+    def test_get_models_ordering_drf3(self):
+        if StrictVersion(rest_framework.VERSION) < StrictVersion('3.0'):
+            raise SkipTest('Only for DRF>=3.0')
+
+        from rest_framework.fields import CurrentUserDefault
+
+        class CommentSerializer(serializers.Serializer):
+            email = serializers.EmailField()
+            content = serializers.CharField(max_length=200)
+            created = serializers.DateTimeField(default=datetime.datetime.now)
+            owner = serializers.PrimaryKeyRelatedField(
+                default=CurrentUserDefault(),
+                queryset=User.objects.all())
+
+        class SerializedAPI(ListCreateAPIView):
+            serializer_class = CommentSerializer
+
+        urlparser = UrlParser()
+        url_patterns = patterns('', url(r'my-api/', SerializedAPI.as_view()))
+        apis = urlparser.get_apis(url_patterns)
+
+        docgen = DocumentationGenerator()
+        models = docgen.get_models(apis)
+
+        self.assertIn('CommentSerializer', models)
+        self.assertEqual(
+            ["email", "content", "created", "owner"],
+            list(models['CommentSerializer']['properties'].keys()))
+
     def test_get_serializer_set(self):
         class SerializedAPI(ListCreateAPIView):
             serializer_class = CommentSerializer

tox.ini

@@ -20,6 +20,7 @@ deps =
     py26: Markdown==2.1.1
     py26: importlib==1.0.1
     py26: ordereddict==1.1
+    py26: unittest2==1.1.0
     py27: functools32==3.2.3-2
     docutils==0.11
     argparse==1.2.1