feat: support type array

Redocly · May 25, 2021 · 7bdeccd · 7bdeccd
1 parent 8ba3c31
commit 7bdeccd
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 1 deletion.
diff --git a/src/infer.js b/src/infer.js
@@ -26,7 +26,7 @@ const schemaKeywordTypes = {
 
 export function inferType(schema) {
   if (schema.type !== undefined) {
-    return schema.type;
+    return Array.isArray(schema.type) ? schema.type.length === 0 ? null : schema.type[0] : schema.type;
   }
   const keywords = Object.keys(schemaKeywordTypes);
   for (var i = 0; i < keywords.length; i++) {

diff --git a/src/traverse.js b/src/traverse.js
@@ -98,6 +98,9 @@ export function traverse(schema, options, spec, context) {
     example = schema.examples[0];
   } else {
     type = schema.type;
+    if (Array.isArray(type) && schema.type.length > 0) {
+      type = schema.type[0];
+    }
     if (!type) {
       type = inferType(schema);
     }

diff --git a/test/integration.spec.js b/test/integration.spec.js
@@ -44,13 +44,31 @@ describe('Integration', function() {
       expect(result).to.deep.equal(expected);
     });
 
+    it('should support type array', function() {
+      schema = {
+        'type': ['string', 'number']
+      };
+      result = JSONSchemaSampler.sample(schema);
+      expected = 'string';
+      expect(result).to.deep.equal(expected);
+    });
+
     it('should use null if type is not specified', function() {
       schema = {
       };
       result = OpenAPISampler.sample(schema);
       expected = null;
       expect(result).to.deep.equal(expected);
     });
+
+    it('should use null if type array is empty', function() {
+      schema = {
+        type: []
+      };
+      result = JSONSchemaSampler.sample(schema);
+      expected = null;
+      expect(result).to.deep.equal(expected);
+    });
   });
 
   describe('Objects', function() {