diff options
Diffstat (limited to 'resources/libraries/python/honeycomb')
| -rw-r--r-- | resources/libraries/python/honeycomb/Lisp.py | 206 |
1 files changed, 200 insertions, 6 deletions
diff --git a/resources/libraries/python/honeycomb/Lisp.py b/resources/libraries/python/honeycomb/Lisp.py index 7ec259347c..8c124d194c 100644 --- a/resources/libraries/python/honeycomb/Lisp.py +++ b/resources/libraries/python/honeycomb/Lisp.py @@ -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): @@ -163,6 +227,27 @@ class LispKeywords(object): 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. @@ -217,6 +302,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. @@ -231,6 +361,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 set is specified, disable PITR instead. @@ -257,6 +425,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. |