CSIT-563: HC Test: improved Lisp test coverage
[csit.git] / resources / libraries / python / honeycomb / Lisp.py
index 7ec2593..8c124d1 100644 (file)
@@ -94,24 +94,88 @@ class LispKeywords(object):
             return resp
 
     @staticmethod
-    def set_lisp_state(node, state):
+    def verify_map_server_data_from_honeycomb(data, ip_addresses):
+        """Verify whether MAP server data from Honeycomb is correct.
+
+        :param data: Lisp operational data containing map server IP addresses.
+        :param ip_addresses: IP addresses to verify map server data against.
+        :type data: dict
+        :type ip_addresses: list
+        :returns: Boolean Value indicating equality of IP Lists.
+        :rtype: bool
+        """
+
+        data =\
+            data['lisp-state']['lisp-feature-data']['map-servers']['map-server']
+
+        data = sorted([entry['ip-address'] for entry in data])
+        ip_addresses.sort()
+
+        return data == ip_addresses
+
+    @staticmethod
+    def verify_map_server_data_from_vat(data, ip_addresses):
+        """Verify whether MAP server data from VAT is correct.
+
+        :param data: Lisp operational data containing map server IP addresses.
+        :param ip_addresses: IP addresses to verify map server data against.
+        :type data: dict
+        :type ip_addresses: list
+        :returns: Boolean Value indicating equality of IP Lists.
+        :rtype: bool
+        """
+
+        data = sorted([entry['map-server'] for entry in data])
+        ip_addresses.sort()
+
+        return data == ip_addresses
+
+    @staticmethod
+    def set_lisp_state(node, state=True):
         """Enable or disable the Lisp feature.
 
         :param node: Honeycomb node.
-        :param state: Desired Lisp state, enable or disable.
+        :param state: Enable or disable Lisp.
         :type node: dict
-        :type state: str
+        :type state: bool
         :returns: Content of response.
         :rtype: bytearray
+        :raises HoneycombError: If the return code is not 200:OK
+        or 404:NOT FOUND.
         """
 
+        ret_code, data = HcUtil.get_honeycomb_data(node, "config_lisp")
+        if ret_code == HTTPCodes.OK:
+            data["lisp"]["enable"] = bool(state)
+        elif ret_code == HTTPCodes.NOT_FOUND:
+                data = {"lisp": {"enable": bool(state)}}
+        else:
+            raise HoneycombError("Unexpected return code when getting existing"
+                                 " Lisp configuration.")
+
+        return LispKeywords._set_lisp_properties(node, '', data)
+
+    @staticmethod
+    def set_rloc_probe_state(node, state=False):
+        """Enable or disable the Routing Locator probe.
+
+        :param node: Honeycomb node.
+        :param state: Enable or Disable the Rloc probe.
+        :type node: dict
+        :type state: bool
+        :returns: Content of response.
+        :rtype: bytearray
+        """
+
+        path = "/lisp-feature-data/rloc-probe"
+
         data = {
-            "lisp": {
-                "enable": True if state.lower() == "enable" else False
+            "rloc-probe": {
+                "enabled": bool(state)
             }
         }
 
-        return LispKeywords._set_lisp_properties(node, '', data)
+        return LispKeywords._set_lisp_properties(node, path, data)
 
     @staticmethod
     def add_locator(node, interface, locator_set, priority=1, weight=1):
@@ -162,6 +226,27 @@ class LispKeywords(object):
         path = "/lisp-feature-data/eid-table"
         return LispKeywords._set_lisp_properties(node, path, data)
 
+    @staticmethod
+    def configure_lisp_map_request_mode(node, option):
+        """Modify Lisp Map Request Mode configuration to the data provided.
+
+        :param node: Honeycomb node.
+        :param option: Settings for the Lisp map request mode.
+        :type node: dict
+        :type option: str
+        :returns: Content of response.
+        :rtype: bytearray
+        """
+
+        data = {
+            "map-request-mode": {
+                "mode": option
+            }
+        }
+
+        path = "/lisp-feature-data/map-request-mode"
+        return LispKeywords._set_lisp_properties(node, path, data)
+
     @staticmethod
     def add_lisp_adjacency(node, vni_id, map_name, adjacency_name, data):
         """Add an adjacency to an existing Lisp mapping.
@@ -216,6 +301,51 @@ class LispKeywords(object):
 
         return LispKeywords._set_lisp_properties(node, path, data)
 
+    @staticmethod
+    def set_map_register(node, map_register=False):
+        """Configure Map Register.
+
+        :param node: Honeycomb node.
+        :param map_register: Enable or disable Map Register.
+        :type node: dict
+        :type map_register: bool
+        :returns: Content of response.
+        :rtype: bytearray
+        """
+
+        path = "/lisp-feature-data/map-register"
+
+        data = {
+            "map-register": {
+                "enabled": bool(map_register)
+            }
+        }
+
+        return LispKeywords._set_lisp_properties(node, path, data)
+
+    @staticmethod
+    def set_map_request_mode(node, src_dst=False):
+        """Configure Map Request Mode.
+
+        :param node: Honeycomb node.
+        :param src_dst: Configure Map Request Mode with source destination.
+        :type node: dict
+        :type src_dst: bool
+        :returns: Content of response.
+        :rtype: bytearray
+        """
+
+        path = "/lisp-feature-data/map-request-mode"
+
+        data = {
+            "map-request-mode": {
+                "mode": "source-destination" if src_dst
+                else "target-destination"
+            }
+        }
+
+        return LispKeywords._set_lisp_properties(node, path, data)
+
     @staticmethod
     def delete_map_resolver(node):
         """Delete an existing map resolver.
@@ -230,6 +360,44 @@ class LispKeywords(object):
 
         return LispKeywords._set_lisp_properties(node, path)
 
+    @staticmethod
+    def add_map_server(node, *ip_addresses):
+        """Configure map server with the specified IP addresses.
+
+        :param node: Honeycomb node.
+        :param ip_addresses: IP addresses to configure map server with.
+        :type node: dict
+        :type ip_addresses: list
+        :returns: Content of response.
+        :rtype: bytearray
+        """
+
+        path = "/lisp-feature-data/map-servers"
+
+        data = {
+            "map-servers": {
+                "map-server": [
+                    {"ip-address": ip_address} for ip_address in ip_addresses
+                ]
+            }
+        }
+
+        return LispKeywords._set_lisp_properties(node, path, data)
+
+    @staticmethod
+    def delete_map_server(node):
+        """Delete all map servers.
+
+        :param node: Honeycomb node
+        :type node: dict
+        :returns: Content of response
+        :rtype: bytearray
+        """
+
+        path = "/lisp-feature-data/map-servers"
+
+        return LispKeywords._set_lisp_properties(node, path)
+
     @staticmethod
     def configure_pitr(node, locator_set=None):
         """Configure PITR feature with the specified locator set. If not locator
@@ -256,6 +424,32 @@ class LispKeywords(object):
 
         return LispKeywords._set_lisp_properties(node, path, data)
 
+    @staticmethod
+    def configure_petr(node, ip_address):
+        """Configure PETR feature with the specified IP. If no IP
+        specified, disable PETR instead.
+
+        :param node: Honeycomb node.
+        :param ip_address: IPv6 address.
+        :type node: dict
+        :type ip_address: str
+        :returns: Content of response.
+        :rtype: bytearray
+        """
+
+        path = "/lisp-feature-data/petr-cfg"
+
+        if ip_address:
+            data = {
+                "petr-cfg": {
+                    "petr-address": ip_address
+                }
+            }
+        else:
+            data = None
+
+        return LispKeywords._set_lisp_properties(node, path, data)
+
     @staticmethod
     def disable_lisp(node):
         """Remove all Lisp settings on the node.