support unit tests for models with schema type as array krakenjs#69

subeeshcbabu-zz · Feb 11, 2016 · 7ed6487 · 7ed6487
1 parent ee8cd66
commit 7ed6487
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/app/templates/_test_express.js b/app/templates/_test_express.js
@@ -25,6 +25,7 @@ test('api', function (t) {
         <%
         var path = operation.path;
         var body;
+        var ref;
         var responseCode = operation.responses && Object.keys(operation.responses)[0];
         var response = responseCode && operation.responses[responseCode];
         var responseSchema = response && response.schema;
@@ -46,8 +47,16 @@ test('api', function (t) {
                         }
                     });
                 }
-                if (param.in === 'body' && param.schema && param.schema.$ref) {
-                    body = models[param.schema.$ref.slice(param.schema.$ref.lastIndexOf('/') + 1)];
+                if (param.in === 'body' && param.schema ) {
+                    ref = param.schema.$ref;
+                    //If the schema type is array then the definition $ref will be items
+                    if (param.schema.type === 'array' && param.schema.items) {
+                        ref = param.schema.items.$ref;
+                    }
+                    //Skip if ref is undefined
+                    if (ref) {
+                        body = models[param.schema.$ref.slice(param.schema.$ref.lastIndexOf('/') + 1)];
+                    }
                 }
             });
         }