add tests for lib utils

Author: lvrfrc87

netcompare/check_type.py

@@ -1,8 +1,11 @@
 """CheckType Implementation."""
+import sys
 from typing import Mapping, Tuple, Union, List
 from .evaluator import diff_generator
 from .runner import extract_values_from_output
 
+sys.path.append(".")
+
 
 class CheckType:
     """Check Type Class."""

netcompare/runner.py

@@ -2,9 +2,9 @@
 import re
 import jmespath
 from typing import Mapping, List, Union
-from .utils.jmspath.parsers import jmspath_value_parser, jmspath_refkey_parser
-from .utils.data.parsers import exclude_filter, get_values
-from .utils.refkey.utils import keys_cleaner, keys_values_zipper
+from .utils.jmspath_parsers import jmspath_value_parser, jmspath_refkey_parser
+from .utils.filter_parsers import exclude_filter, get_values
+from .utils.refkey_utils import keys_cleaner, keys_values_zipper
 
 
 def extract_values_from_output(value: Mapping, path: Mapping, exclude: List) -> Union[Mapping, List, int, str, bool]:
@@ -49,46 +49,5 @@ def extract_values_from_output(value: Mapping, path: Mapping, exclude: List) ->
         return filtered_value
 
 
-def associate_key_of_my_value(paths: Mapping, wanted_value: List) -> List:
-    """
-    Associate each key defined in path to every value found in output.
-
-    Args:
-        paths: {"path": "global.peers.*.[is_enabled,is_up]"}
-        wanted_value: [[True, False], [True, False], [True, False], [True, False]]
-
-    Return:
-        [{'is_enabled': True, 'is_up': False}, ...
-
-    Example:
-        >>> from runner import associate_key_of_my_value
-        >>> path = {"path": "global.peers.*.[is_enabled,is_up]"}
-        >>> wanted_value = [[True, False], [True, False], [True, False], [True, False]]
-        {'is_enabled': True, 'is_up': False}, {'is_enabled': True, 'is_up': False}, ...
-    """
-
-    # global.peers.*.[is_enabled,is_up] / result.[*].state
-    find_the_key_of_my_values = paths.split(".")[-1]
-
-    # [is_enabled,is_up]
-    if find_the_key_of_my_values.startswith("[") and find_the_key_of_my_values.endswith("]"):
-        # ['is_enabled', 'is_up']
-        my_key_value_list = find_the_key_of_my_values.strip("[]").split(",")
-    # state
-    else:
-        my_key_value_list = [find_the_key_of_my_values]
-
-    final_list = list()
-
-    for items in wanted_value:
-        temp_dict = dict()
-
-        if len(items) != len(my_key_value_list):
-            raise ValueError("Key's value len != from value len")
-
-        for my_index, my_value in enumerate(items):
-            temp_dict.update({my_key_value_list[my_index]: my_value})
-        final_list.append(temp_dict)
 
-    return final_list
 

netcompare/utils/data/__init__.py renamed to netcompare/utils/__init__.py

@@ -1,7 +1,7 @@
 from typing import Mapping, List
-from ..jmspath.parsers import jmspath_value_parser
-from ..list.flatten import flatten_list
-from ...runner import associate_key_of_my_value
+from .jmspath_parsers import jmspath_value_parser
+from .flatten_utils import flatten_list
+from .refkey_utils import associate_key_of_my_value
 
 
 def exclude_filter(data: Mapping, exclude: List):