From e0b4e2d22a2454a6e5be0ce567a4aa0131018d27 Mon Sep 17 00:00:00 2001
From: Ferdi Koomen <info@madebyferdi.com>
Date: Mon, 29 Nov 2021 16:52:20 -0500
Subject: [PATCH] - Fixed #771

---
 src/openApi/v2/parser/getModelComposition.ts | 3 +++
 src/openApi/v3/parser/getModelComposition.ts | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/src/openApi/v2/parser/getModelComposition.ts b/src/openApi/v2/parser/getModelComposition.ts
index 1e9619c3..468dbe26 100644
--- a/src/openApi/v2/parser/getModelComposition.ts
+++ b/src/openApi/v2/parser/getModelComposition.ts
@@ -59,6 +59,9 @@ export function getModelComposition(
         modelProperties.forEach(modelProperty => {
             composition.imports.push(...modelProperty.imports);
             composition.enums.push(...modelProperty.enums);
+            if (modelProperty.export === 'enum') {
+                composition.enums.push(modelProperty);
+            }
         });
         properties.push(...modelProperties);
     }
diff --git a/src/openApi/v3/parser/getModelComposition.ts b/src/openApi/v3/parser/getModelComposition.ts
index 1e9619c3..468dbe26 100644
--- a/src/openApi/v3/parser/getModelComposition.ts
+++ b/src/openApi/v3/parser/getModelComposition.ts
@@ -59,6 +59,9 @@ export function getModelComposition(
         modelProperties.forEach(modelProperty => {
             composition.imports.push(...modelProperty.imports);
             composition.enums.push(...modelProperty.enums);
+            if (modelProperty.export === 'enum') {
+                composition.enums.push(modelProperty);
+            }
         });
         properties.push(...modelProperties);
     }