fix: add support for $ref when pulling content types (#227)

erunion · web-flow · commit 582968d5794d · 2020-07-17T12:10:40.000-07:00
* fix: add support for $ref when pulling content types

* chore: changing the name of a test method so it makes sense
diff --git a/packages/tooling/__tests__/operation.test.js b/packages/tooling/__tests__/operation.test.js
@@ -66,6 +66,45 @@ describe('#getContentType', () => {
       }).getContentType()
     ).toBe('text/xml');
   });
+
+  it('should handle cases where the requestBody is a $ref', () => {
+    const op = new Operation(
+      {
+        ...petstore,
+        ...{
+          components: {
+            requestBodies: {
+              payload: {
+                required: true,
+                content: {
+                  'multipart/form-data': {
+                    schema: {
+                      type: 'object',
+                      properties: {
+                        'Document file': {
+                          type: 'string',
+                          format: 'binary',
+                        },
+                      },
+                    },
+                  },
+                },
+              },
+            },
+          },
+        },
+      },
+      '/body',
+      'post',
+      {
+        requestBody: {
+          $ref: '#/components/requestBodies/payload',
+        },
+      }
+    );
+
+    expect(op.getContentType()).toBe('multipart/form-data');
+  });
 });
 
 describe('#getSecurity()', () => {
diff --git a/packages/tooling/src/operation.js b/packages/tooling/src/operation.js
@@ -10,7 +10,16 @@ class Operation {
   }
 
   getContentType() {
-    const types = (this.requestBody && this.requestBody.content && Object.keys(this.requestBody.content)) || [];
+    let types = [];
+    if (this.requestBody) {
+      if ('$ref' in this.requestBody) {
+        this.requestBody = findSchemaDefinition(this.requestBody.$ref, this.oas);
+      }
+
+      if ('content' in this.requestBody) {
+        types = Object.keys(this.requestBody.content);
+      }
+    }
 
     let type = 'application/json';
     if (types && types.length) {
