Add null check for schema and operation group for special case (#87)

Co-authored-by: xichen <[email protected]>
Azure · Jul 29, 2020 · 41818df · 41818df
1 parent 3d98a22
commit 41818df
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/helper.ts b/src/helper.ts
@@ -439,6 +439,9 @@ export class Helper {
     public static enumrateSchemas(schemas: ObjectSchema[], action: (nodeDescriptor: CliCommonSchema.CodeModel.NodeDescriptor) => void, flag: CliCommonSchema.CodeModel.NodeTypeFlag): void {
         const enumObjectSchema = isNullOrUndefined(flag) || ((flag & CliCommonSchema.CodeModel.NodeTypeFlag.objectSchema) > 0);
         const cliKeyMissing = '<clikey-missing>';
+        if (isNullOrUndefined(schemas)) {
+            return;
+        }
 
         for (let i = schemas.length - 1; i >= 0; i--) {
             const schema = schemas[i];
@@ -477,6 +480,9 @@ export class Helper {
     public static enumrateOperationGroups(groups: OperationGroup[], action: (nodeDescriptor: CliCommonSchema.CodeModel.NodeDescriptor) => void, flag: CliCommonSchema.CodeModel.NodeTypeFlag): void {
         const enumGroup = isNullOrUndefined(flag) || ((flag & CliCommonSchema.CodeModel.NodeTypeFlag.operationGroup) > 0);
         const cliKeyMissing = '<clikey-missing>';
+        if (isNullOrUndefined(groups)) {
+            return;
+        }
 
         for (let i = groups.length - 1; i >= 0; i--) {
             const group = groups[i];