Adds more tests

Author: Daniel Alvarez

tests/schemas/test_openapi.py

@@ -258,6 +258,28 @@ def test_serializer_datefield(self):
         assert response_schema['date']['format'] == 'date'
         assert response_schema['datetime']['format'] == 'date-time'
 
+    def test_serializer_listfield(self):
+        patterns = [
+            url(r'^example/?$', views.ExampleGenericViewSet.as_view({"get": "get"})),
+        ]
+        generator = SchemaGenerator(patterns=patterns)
+
+        request = create_request('/')
+        schema = generator.get_schema(request=request)
+
+        response = schema['paths']['/example/']['get']['responses']
+        response_schema = response['200']['content']['application/json']['schema']['properties']
+
+        assert response_schema['float_list']['type'] == 'array'
+        assert response_schema['float_list']['items']['type'] == 'number'
+
+        assert response_schema['string_list']['type'] == 'array'
+        assert response_schema['string_list']['items']['type'] == 'string'
+
+        assert response_schema['boolean_list']['type'] == 'array'
+        assert response_schema['boolean_list']['items']['type'] == 'boolean'
+        assert response_schema['boolean_list']['default'] == [True, False]
+
     def test_serializer_validators(self):
         patterns = [
             url(r'^example/?$', views.ExampleValidatedAPIView.as_view()),
@@ -308,13 +330,6 @@ def test_serializer_validators(self):
         assert response_schema['ip']['type'] == 'string'
         assert 'format' not in response_schema['ip']
 
-        assert response_schema['float_list']['type'] == 'array'
-        assert response_schema['float_list']['items']['type'] == 'number'
-
-        assert response_schema['string_list']['type'] == 'array'
-        assert response_schema['string_list']['items']['type'] == 'string'
-
         assert response_schema['boolean_list']['type'] == 'array'
         assert response_schema['boolean_list']['items']['type'] == 'boolean'
         assert response_schema['boolean_list']['default'] == [True, False]
-

tests/schemas/views.py

@@ -33,6 +33,9 @@ def get(self, *args, **kwargs):
 class ExampleSerializer(serializers.Serializer):
     date = serializers.DateField()
     datetime = serializers.DateTimeField()
+    float_list = serializers.ListField(child=serializers.FloatField())
+    string_list = serializers.ListField(child=serializers.CharField())
+    boolean_list = serializers.ListField(child=serializers.BooleanField(), default=[True, False])
 
 
 class ExampleGenericAPIView(generics.GenericAPIView):
@@ -94,8 +97,6 @@ class ExampleValidatedSerializer(serializers.Serializer):
     ip4 = serializers.IPAddressField(protocol='ipv4')
     ip6 = serializers.IPAddressField(protocol='ipv6')
     ip = serializers.IPAddressField()
-    float_list = serializers.ListField(child=serializers.FloatField())
-    string_list = serializers.ListField(child=serializers.CharField())
     boolean_list = serializers.ListField(child=serializers.BooleanField(), default=[True, False])
 
 
@@ -106,7 +107,6 @@ def get(self, *args, **kwargs):
         serializer = self.get_serializer(
             integer=33, string='hello', regex='foo', decimal1=3.55, decimal2=5.33,
             email='[email protected]', url='http://localhost', uuid=uuid.uuid4(), ip4='127.0.0.1',
-            ip6='::1', ip='192.168.1.1', float_list=[1.0, 2.0], string_list=['a', 'b', 'c'],
-            boolean_list=[True]
+            ip6='::1', ip='192.168.1.1', decimal_list=[1.0, 2.5]
         )
         return Response(serializer.data)