Merge remote-tracking branch 'origin/master'

KissPeter · Feb 7, 2021 · f9cb9cf · f9cb9cf
2 parents ccc4f5d + e609a1c
commit f9cb9cf
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/apifuzzer/move_json_parts.py b/apifuzzer/move_json_parts.py
@@ -2,6 +2,9 @@
 
 
 class JsonSectionAbove:
+    """
+    By organizing the API definition OpenAPI v3 and Swagger v2 API definitions can be processed in a similar way
+    """
 
     def __init__(self, api_definition, section_to_up='schema'):
         self.logger = get_logger(f'{self.__class__.__name__}-{section_to_up}')
@@ -28,9 +31,9 @@ def _resolve(self, data):
                 elif isinstance(value, list):
                     if not return_data.get(key):
                         return_data[key] = list()
-                    for _iter in range(len(value)):
+                    for _iter, val in enumerate(value):
                         self.logger.debug(f'Process {key} list elem: {_iter}')
-                        return_data[key].append(self.resolve(data=value[_iter]))
+                        return_data[key].append(self.resolve(data=val))
 
                 else:
                     return_data[key] = value

diff --git a/apifuzzer/resolve_json_reference.py b/apifuzzer/resolve_json_reference.py
@@ -13,6 +13,9 @@ class FailedToResolveReference(Exception):
 
 
 class ResolveReferences:
+    """
+    Resolve API references in order to get one single JSON object with all data available in place
+    """
 
     def __init__(self, api_definition_path=None, api_definition_url=""):
         self.logger = get_logger(self.__class__.__name__)
@@ -156,7 +159,7 @@ def _resolve(self, data):
                 elif isinstance(value, list):
                     if not return_data.get(key):
                         return_data[key] = list()
-                    for _iter in range(len(value)):
+                    for _iter, _ in enumerate(value):
                         self.logger.debug(f'Process {key} list elem: {_iter}')
                         return_data[key].append(self.resolve(data=data[key][_iter]))
                 elif key == '$ref' and value: