Extract headers for Exceptions (#712)

Azure · Aug 11, 2020 · a5e65b8 · a5e65b8
1 parent 17b4080
commit a5e65b8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/generators/operationGenerator.ts b/src/generators/operationGenerator.ts
@@ -958,7 +958,7 @@ function hasMediaType(
   operationDetails: OperationDetails,
   mediaType: KnownMediaType
 ) {
-  if (!operationDetails.requests) {
+  if (!operationDetails.requests.some(r => !!r.mediaType)) {
     return operationDetails.mediaTypes.has(mediaType);
   }
 

diff --git a/src/utils/extractHeaders.ts b/src/utils/extractHeaders.ts
@@ -9,8 +9,12 @@ export function extractHeaders(
   let responseHeaders: ObjectSchema[] = [];
 
   operationGroups.forEach(operationGroup =>
-    operationGroup.operations.forEach(operation =>
-      operation.responses?.forEach(response => {
+    operationGroup.operations.forEach(operation => {
+      const responsesAndExceptions = [
+        ...(operation.responses || []),
+        ...(operation.exceptions || [])
+      ];
+      responsesAndExceptions.forEach(response => {
         const operationName = getOperationFullName(
           operationGroup,
           operation,
@@ -21,8 +25,8 @@ export function extractHeaders(
           const headerSchema = headersToSchema(headers, operationName);
           headerSchema && responseHeaders.push(headerSchema);
         }
-      })
-    )
+      });
+    })
   );
 
   return responseHeaders;