feat: Add mapping asyncfunction to method in elementKind (#20)

* feat: Add mapping asyncfunction to method in elementKind

* feat: Update test

* feat: Add async function with annotation

Author: lfvdavid

src/reqstool_python_decorators/processors/decorator_processor.py

@@ -9,6 +9,7 @@
 @unique
 class DECORATOR_TYPES(Enum):
     FUNCTION = ("FUNCTION", "METHOD")
+    ASYNCFUNCTION = ("ASYNCFUNCTION", "METHOD")
 
     def __init__(self, from_value, to_value):
         self.from_value = from_value
@@ -173,7 +174,7 @@ def format_results(self, results):
 
         return formatted_data
 
-    def create_dir_from_path(self, filepath: str):
+    def create_dir_from_path(self, filepath: str) -> None:
         """
         Creates directory of provided filepath if it does not exists
 
@@ -184,7 +185,9 @@ def create_dir_from_path(self, filepath: str):
         if not os.path.exists(directory):
             os.makedirs(directory)
 
-    def process_decorated_data(self, path_to_python_files, output_file="build/reqstool/annotations.yml"):
+    def process_decorated_data(
+        self, path_to_python_files: str, output_file: str = "build/reqstool/annotations.yml"
+    ) -> None:
         """
         "Main" function, runs all functions resulting in  a yaml file containing decorated data.
 

tests/resources/test_decorators/requirements_decorators.py

@@ -7,6 +7,12 @@ class RequirementsClass:
 
 
 @Requirements("REQ_333")
-def requirements_function():
+def requirements_function() -> None:
     # Test function for Requirements decorator
     pass
+
+
+@Requirements("REQ_444")
+async def asyncrequirements_function() -> None:
+    # Test async function for Requirements decorator
+    pass

tests/unit/reqstool_decorators/processors/test_processors.py

@@ -25,8 +25,10 @@ def test_get_functions_and_classes(process_decorator_instance: DecoratorProcesso
 
 
 def test_map_type_known_type(process_decorator_instance: DecoratorProcessor):
-    result = process_decorator_instance.map_type("FUNCTION")
-    assert result == "METHOD"
+    map_funcion = process_decorator_instance.map_type("FUNCTION")
+    map_asyncfunction = process_decorator_instance.map_type("ASYNCFUNCTION")
+    assert map_funcion == "METHOD"
+    assert map_asyncfunction == "METHOD"
 
 
 def test_map_type_unknown_type(process_decorator_instance: DecoratorProcessor):