add test for encoding an integer field; catches format regression

zbyte64 · zbyte64 · commit 9e11a718183b · 2017-11-02T18:42:07.000-07:00
diff --git a/openapi_codec/encode.py b/openapi_codec/encode.py
@@ -129,7 +129,7 @@ def _get_field_type(field):
 
 
 def _get_field_format(field):
-    return field.schema and field.schema.format
+    return field.schema and getattr(field.schema, 'format', None)
 
 
 def _get_parameters(link, encoding):
diff --git a/tests/test_encode.py b/tests/test_encode.py
@@ -129,3 +129,25 @@ def test_expected_fields(self):
             'format': 'date'
         }
         self.assertEquals(self.swagger[0], expected)
+
+
+class TestIntegerField(TestCase):
+    def setUp(self):
+        self.field = coreapi.Field(
+            name='page',
+            required=True,
+            location='query',
+            schema=coreschema.Integer(description='A page number.')
+        )
+        self.swagger = _get_parameters(coreapi.Link(fields=[self.field]), encoding='')
+
+    def test_expected_fields(self):
+        self.assertEquals(len(self.swagger), 1)
+        expected = {
+            'name': self.field.name,
+            'required': self.field.required,
+            'in': 'query',
+            'description': self.field.schema.description,
+            'type': 'integer'
+        }
+        self.assertEquals(self.swagger[0], expected)
