Adding tests to for many=True

Author: i.donchev

rest_framework_docs/templates/rest_framework_docs/components/fields_list.html

@@ -7,7 +7,7 @@
     {% endif %}
 
     {% if field.to_many_relation %}
-      <span class="label label-primary label-required" title="Required">many = True</span>
+      <span class="label label-primary label-required" title="Support sending several objects of this type in a request ({{obj}, {obj}, ... })">Array of objects</span>
     {% endif %}
 
     {% if field.sub_fields %}

tests/serializers.py

@@ -34,11 +34,26 @@ class Meta:
         extra_kwargs = {'password': {'write_only': True}}
 
 
+class MembershipSerializer(serializers.ModelSerializer):
+    class Meta:
+        model = Membership
+        fields = ('joined', 'is_owner', 'role')
+
+
 class CreateOrganisationSerializer(serializers.ModelSerializer):
+    membership_set = MembershipSerializer(many=True)
+
+    class Meta:
+        model = Organisation
+        fields = ('name', 'slug', 'membership_set')
+
+
+class RetrieveOrganisationSerializer(serializers.ModelSerializer):
+    membership_set = MembershipSerializer()
 
     class Meta:
         model = Organisation
-        fields = ('name', 'slug',)
+        fields = ('name', 'slug', 'is_active', 'membership_set')
 
 
 class OrganisationMembersSerializer(serializers.ModelSerializer):

tests/tests.py

@@ -27,7 +27,7 @@ def test_index_view_with_endpoints(self):
         response = self.client.get(reverse('drfdocs'))
 
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(len(response.context["endpoints"]), 14)
+        self.assertEqual(len(response.context["endpoints"]), 15)
 
         # Test the login view
         self.assertEqual(response.context["endpoints"][0].name_parent, "accounts")
@@ -72,9 +72,12 @@ def test_model_viewset(self):
         response = self.client.get(reverse('drfdocs'))
 
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(response.context["endpoints"][10].path, '/organisation-model-viewsets/')
-        self.assertEqual(response.context["endpoints"][11].path, '/organisation-model-viewsets/<pk>/')
-        self.assertEqual(response.context["endpoints"][10].allowed_methods, ['GET', 'POST', 'OPTIONS'])
-        self.assertEqual(response.context["endpoints"][11].allowed_methods, ['GET', 'PUT', 'PATCH', 'DELETE', 'OPTIONS'])
-        self.assertEqual(response.context["endpoints"][12].allowed_methods, ['POST', 'OPTIONS'])
-        self.assertEqual(response.context["endpoints"][12].docstring, 'This is a test.')
+
+        self.assertEqual(response.context["endpoints"][10].path, '/organisations/<slug>/')
+        self.assertEqual(response.context['endpoints'][6].fields[2]['to_many_relation'], True)
+        self.assertEqual(response.context["endpoints"][11].path, '/organisation-model-viewsets/')
+        self.assertEqual(response.context["endpoints"][12].path, '/organisation-model-viewsets/<pk>/')
+        self.assertEqual(response.context["endpoints"][11].allowed_methods, ['GET', 'POST', 'OPTIONS'])
+        self.assertEqual(response.context["endpoints"][12].allowed_methods, ['GET', 'PUT', 'PATCH', 'DELETE', 'OPTIONS'])
+        self.assertEqual(response.context["endpoints"][13].allowed_methods, ['POST', 'OPTIONS'])
+        self.assertEqual(response.context["endpoints"][13].docstring, 'This is a test.')

tests/urls.py

@@ -22,7 +22,8 @@
     url(r'^create/$', view=views.CreateOrganisationView.as_view(), name="create"),
     url(r'^(?P<slug>[\w-]+)/members/$', view=views.OrganisationMembersView.as_view(), name="members"),
     url(r'^(?P<slug>[\w-]+)/leave/$', view=views.LeaveOrganisationView.as_view(), name="leave"),
-    url(r'^(?P<slug>[\w-]+)/errored/$', view=views.OrganisationErroredView.as_view(), name="errored")
+    url(r'^(?P<slug>[\w-]+)/errored/$', view=views.OrganisationErroredView.as_view(), name="errored"),
+    url(r'^(?P<slug>[\w-]+)/$', view=views.RetrieveOrganisationView.as_view(), name="organisation"),
 ]
 
 router = SimpleRouter()

tests/views.py

@@ -144,3 +144,11 @@ class TestModelViewSet(ModelViewSet):
     def test_route(self, request):
         """This is a test."""
         return Response()
+
+
+class RetrieveOrganisationView(generics.RetrieveAPIView):
+
+    serializer_class = serializers.RetrieveOrganisationSerializer
+
+    def get_object(self):
+        pass