aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries
diff options
context:
space:
mode:
Diffstat (limited to 'resources/libraries')
-rw-r--r--resources/libraries/python/HoneycombAPIKeywords.py543
-rw-r--r--resources/libraries/python/HoneycombUtil.py101
2 files changed, 618 insertions, 26 deletions
diff --git a/resources/libraries/python/HoneycombAPIKeywords.py b/resources/libraries/python/HoneycombAPIKeywords.py
new file mode 100644
index 0000000000..0369a32d5b
--- /dev/null
+++ b/resources/libraries/python/HoneycombAPIKeywords.py
@@ -0,0 +1,543 @@
+# Copyright (c) 2016 Cisco and/or its affiliates.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Keywords used with Honeycomb.
+
+There are implemented keywords which work with:
+- Honeycomb operations
+- VPP Interfaces
+- Bridge domains
+
+The keywords make possible to put and get configuration data and to get
+operational data.
+"""
+
+from json import dumps
+
+from robot.api import logger
+
+from resources.libraries.python.HTTPRequest import HTTPCodes
+from resources.libraries.python.HoneycombSetup import HoneycombError
+from resources.libraries.python.HoneycombUtil import HoneycombUtil as HcUtil
+from resources.libraries.python.HoneycombUtil import DataRepresentation
+
+
+class OperationsKeywords(object):
+ """Keywords which perform "operations" in Honeycomb.
+
+ The keywords in this class are not a part of a specific area in Honeycomb,
+ e.g.: interfaces or bridge domains, but they perform "operations" in any
+ area of Honeycomb.
+ """
+
+ def __init__(self):
+ pass
+
+ @staticmethod
+ def poll_oper_data(node):
+ """Poll operational data.
+
+ You can use this keyword when you configure something in Honeycomb and
+ you want configuration data to make effect immediately, e.g.:
+
+ | | Create Bridge Domain | ....
+ | | Add Bridge Domain To Interface | ....
+ | | Poll Oper Data | ....
+ | | ${br}= | Get Oper Info About Bridge Domain | ....
+
+ ..note:: This is not very reliable way how to poll operational data.
+ This keyword is only temporary workaround and will be removed when this
+ problem is solved in Honeycomb.
+ :param node: Honeycomb node.
+ :type: dict
+ :raises HoneycombError: If it is not possible to poll operational data.
+ """
+
+ status_code, _ = HcUtil.\
+ post_honeycomb_data(node, "poll_oper_data", data='',
+ data_representation=DataRepresentation.NO_DATA,
+ timeout=30)
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError("It was not possible to poll operational data "
+ "on node {0}.".format(node['host']))
+
+
+class InterfaceKeywords(object):
+ """Keywords for Interface manipulation.
+
+ Implements keywords which get configuration and operational data about
+ vpp interfaces and set the interface's parameters using Honeycomb REST API.
+ """
+
+ def __init__(self):
+ pass
+
+ @staticmethod
+ def _configure_interface(node, interface, data,
+ data_representation=DataRepresentation.JSON):
+ """Send interface configuration data and check the response.
+
+ :param node: Honeycomb node.
+ :param interface: The name of interface.
+ :param data: Configuration data to be sent in PUT request.
+ :param data_representation: How the data is represented.
+ :type node: dict
+ :type interface: str
+ :type data: str
+ :type data_representation: DataRepresentation
+ :return: Content of response.
+ :rtype: bytearray
+ :raises HoneycombError: If the status code in response on PUT is not
+ 200 = OK.
+ """
+
+ status_code, resp = HcUtil.\
+ put_honeycomb_data(node, "config_vpp_interfaces", data,
+ data_representation=data_representation)
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError(
+ "The configuration of interface '{0}' was not successful. "
+ "Status code: {1}.".format(interface, status_code))
+ return resp
+
+ @staticmethod
+ def get_all_interfaces_cfg_data(node):
+ """Get configuration data about all interfaces from Honeycomb.
+
+ :param node: Honeycomb node.
+ :type node: dict
+ :return: Configuration data about all interfaces from Honeycomb.
+ :rtype: list
+ :raises HoneycombError: If it is not possible to get configuration data.
+ """
+
+ status_code, resp = HcUtil.get_honeycomb_data(node,
+ "config_vpp_interfaces")
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError(
+ "Not possible to get configuration information about the "
+ "interfaces. Status code: {0}.".format(status_code))
+ try:
+ intf = HcUtil.parse_json_response(resp, ("interfaces", "interface"))
+ return intf
+ except KeyError:
+ return []
+
+ @staticmethod
+ def get_interface_cfg_info(node, interface):
+ """Get configuration data about the given interface from Honeycomb.
+
+ :param node: Honeycomb node.
+ :param interface: The name of interface.
+ :type node: dict
+ :type interface: str
+ :return: Configuration data about the given interface from Honeycomb.
+ :rtype: dict
+ """
+
+ intfs = InterfaceKeywords.get_all_interfaces_cfg_data(node)
+ for intf in intfs:
+ if intf["name"] == interface:
+ return intf
+ return {}
+
+ @staticmethod
+ def get_all_interfaces_oper_info(node):
+ """Get operational data about all interfaces from Honeycomb.
+
+ :param node: Honeycomb node.
+ :type node: dict
+ :return: Operational data about all interfaces from Honeycomb.
+ :rtype: list
+ :raises HoneycombError: If it is not possible to get operational data.
+ """
+
+ status_code, resp = HcUtil.get_honeycomb_data(node,
+ "oper_vpp_interfaces")
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError(
+ "Not possible to get operational information about the "
+ "interfaces. Status code: {0}.".format(status_code))
+ try:
+ intf = HcUtil.parse_json_response(resp, ("interfaces-state",
+ "interface"))
+ return intf
+ except KeyError:
+ return []
+
+ @staticmethod
+ def get_interface_oper_info(node, interface):
+ """Get operational data about the given interface from Honeycomb.
+
+ :param node: Honeycomb node.
+ :param interface: The name of interface.
+ :type node: dict
+ :type interface: str
+ :return: Operational data about the given interface from Honeycomb.
+ :rtype: dict
+ """
+
+ intfs = InterfaceKeywords.get_all_interfaces_oper_info(node)
+ for intf in intfs:
+ if intf["name"] == interface:
+ return intf
+ return {}
+
+ @staticmethod
+ def set_interface_state(node, interface, state="up"):
+ """Set VPP interface state.
+
+ The keyword changes the administration state of interface to up or down
+ depending on the parameter "state".
+
+ :param node: Honeycomb node.
+ :param interface: The name of interface.
+ :param state: The requested state, only "up" and "down" are valid
+ values.
+ :type node: dict
+ :type interface: str
+ :type state: str
+ :return: Content of response.
+ :rtype: bytearray
+ :raises KeyError: If the argument "state" is nor "up" or "down".
+ :raises HoneycombError: If the interface is not present on the node.
+ """
+
+ intf_state = {"up": "true",
+ "down": "false"}
+ intfs = InterfaceKeywords.get_all_interfaces_cfg_data(node)
+ for intf in intfs:
+ if intf["name"] == interface:
+ intf["enabled"] = intf_state[state.lower()]
+ new_intf = {"interfaces": {"interface": intfs}}
+ return InterfaceKeywords._configure_interface(node, interface,
+ dumps(new_intf))
+ raise HoneycombError("The interface '{0}' is not present on node "
+ "'{1}'.".format(interface, node['host']))
+
+ @staticmethod
+ def set_interface_up(node, interface):
+ """Set the administration state of VPP interface to up.
+
+ :param node: Honeycomb node.
+ :param interface: The name of interface.
+ :type node: dict
+ :type interface: str
+ :return: Content of response
+ :rtype: bytearray
+ """
+
+ return InterfaceKeywords.set_interface_state(node, interface, "up")
+
+ @staticmethod
+ def set_interface_down(node, interface):
+ """Set the administration state of VPP interface to down.
+
+ :param node: Honeycomb node.
+ :param interface: The name of interface.
+ :type node: dict
+ :type interface: str
+ :return: Content of response.
+ :rtype: bytearray
+ """
+
+ return InterfaceKeywords.set_interface_state(node, interface, "down")
+
+ @staticmethod
+ def add_bridge_domain_to_interface(node, interface, bd_name,
+ split_horizon_group=None, bvi=None):
+ """Add a new bridge domain to an interface and set its parameters.
+
+ :param node: Honeycomb node.
+ :param interface: The name of interface.
+ :param bd_name: Bridge domain name.
+ :param split_horizon_group: Split-horizon group name.
+ :param bvi: The bridged virtual interface.
+ :type node: dict
+ :type interface: str
+ :type bd_name: str
+ :type split_horizon_group: str
+ :type bvi: str
+ :return: Content of response.
+ :rtype: bytearray
+ :raises HoneycombError: If the interface is not present on the node.
+ """
+
+ intfs = InterfaceKeywords.get_all_interfaces_cfg_data(node)
+ v3po_l2 = {"bridge-domain": str(bd_name)}
+ if split_horizon_group:
+ v3po_l2["split-horizon-group"] = str(split_horizon_group)
+ if bvi:
+ v3po_l2["bridged-virtual-interface"] = str(bvi)
+ for intf in intfs:
+ if intf["name"] == interface:
+ intf["v3po:l2"] = v3po_l2
+ new_intf = {"interfaces": {"interface": intfs}}
+ return InterfaceKeywords._configure_interface(node, interface,
+ dumps(new_intf))
+ raise HoneycombError("The interface '{0}' is not present on node "
+ "'{1}'.".format(interface, node['host']))
+
+
+class BridgeDomainKeywords(object):
+ """Keywords for Bridge domain manipulation.
+
+ Implements keywords which get configuration and operational data about
+ bridge domains and put the bridge domains' parameters using Honeycomb REST
+ API.
+ """
+
+ def __init__(self):
+ pass
+
+ @staticmethod
+ def _create_json_bridge_domain_info(name, **kwargs):
+ """Generate bridge domain information in the structure as it is expected
+ by Honeycomb.
+
+ The generated data structure is as follows:
+ {
+ "bridge-domains": {
+ "bridge-domain": [
+ {
+ "name": "bd_name",
+ "flood": "false",
+ "forward": "false",
+ "learn": "false",
+ "unknown-unicast-flood": "false",
+ "arp-termination": "false"
+ }
+ ]
+ }
+ }
+
+ :param name: The name of new bridge-domain.
+ :param kwargs: named arguments:
+ flood (bool): If True, flooding is enabled.
+ forward (bool): If True, packet forwarding is enabled.
+ learn (bool): If True, learning is enabled.
+ uu_flood (bool): If True, unknown unicast flooding is enabled.
+ arp_termination (bool): If True, ARP termination is enabled.
+ :type name: str
+ :type kwargs: dict
+ :return: Bridge domain information in format suitable for Honeycomb.
+ :rtype: dict
+ :raises KeyError: If at least one of kwargs items is missing.
+ """
+
+ brd_info = {
+ "bridge-domains": {
+ "bridge-domain": [
+ {"name": name,
+ "flood": str(kwargs["flood"]).lower(),
+ "forward": str(kwargs["forward"]).lower(),
+ "learn": str(kwargs["learn"]).lower(),
+ "unknown-unicast-flood": str(kwargs["uu_flood"]).lower(),
+ "arp-termination": str(kwargs["arp_termination"]).lower()},
+ ]
+ }
+ }
+
+ return brd_info
+
+ @staticmethod
+ def create_bridge_domain(node, name, flood=True, forward=True, learn=True,
+ uu_flood=True, arp_termination=False):
+ """Create a bridge domain using Honeycomb.
+
+ This keyword adds a new bridge domain to the list of bridge domains and
+ sets its parameters. The existing bridge domains are untouched.
+ :param node: Node with Honeycomb where the bridge domain should be
+ created.
+ :param name: The name of new bridge-domain.
+ :param flood: If True, flooding is enabled.
+ :param forward: If True, packet forwarding is enabled.
+ :param learn: If True, learning is enabled.
+ :param uu_flood: If True, unknown unicast flooding is enabled.
+ :param arp_termination: If True, ARP termination is enabled.
+ :type node: dict
+ :type name: str
+ :type flood: bool
+ :type forward: bool
+ :type learn: bool
+ :type uu_flood: bool
+ :type arp_termination: bool
+ :raises HoneycombError: If the bridge domain already exists or it has
+ not been created.
+ """
+
+ existing_brds = BridgeDomainKeywords.\
+ get_all_bds_cfg_data(node, ignore_404=True)
+
+ for brd in existing_brds:
+ if brd["name"] == name:
+ raise HoneycombError("Bridge domain {0} already exists.".
+ format(name))
+
+ brd_info = BridgeDomainKeywords._create_json_bridge_domain_info(
+ name, flood=flood, forward=forward, learn=learn, uu_flood=uu_flood,
+ arp_termination=arp_termination)
+ for brd in existing_brds:
+ brd_info["bridge-domains"]["bridge-domain"].append(brd)
+
+ status_code, _ = HcUtil.put_honeycomb_data(node, "config_bridge_domain",
+ dumps(brd_info))
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError(
+ "Bridge domain {0} was not created. "
+ "Status code: {01}.".format(name, status_code))
+
+ @staticmethod
+ def get_all_bds_oper_data(node):
+ """Get operational data about all bridge domains from Honeycomb.
+
+ :param node: Honeycomb node.
+ :type node: dict
+ :return: Operational data about all bridge domains from Honeycomb.
+ :rtype: list
+ :raises HoneycombError: If it is not possible to get information about
+ the bridge domains.
+ """
+
+ status_code, resp = HcUtil.get_honeycomb_data(node,
+ "oper_bridge_domains")
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError(
+ "Not possible to get information about the bridge domains. "
+ "Status code: {0}.".format(status_code))
+ try:
+ br_domains = HcUtil.parse_json_response(resp, ("bridge-domains",
+ "bridge-domain"))
+ except KeyError:
+ return []
+ return br_domains
+
+ @staticmethod
+ def get_bd_oper_data(node, name):
+ """Get operational data about the given bridge domain from Honeycomb.
+
+ :param node: Honeycomb node.
+ :param name: The name of bridge domain.
+ :type node: dict
+ :type name: str
+ :return: Operational data about the given bridge domain from Honeycomb.
+ :rtype: dict
+ """
+
+ br_domains = BridgeDomainKeywords.get_all_bds_oper_data(node)
+ for br_domain in br_domains:
+ if br_domain["name"] == name:
+ br_domain["name"] = br_domain["name"]
+ return br_domain
+ return {}
+
+ @staticmethod
+ def get_all_bds_cfg_data(node, ignore_404=False):
+ """Get configuration data about all bridge domains from Honeycomb.
+
+ :param node: Honeycomb node.
+ :param ignore_404: If True, the error 404 is ignored.
+ :type node: dict
+ :type ignore_404: bool
+ :return: Configuration data about all bridge domains from Honeycomb.
+ :rtype: list
+ :raises HoneycombError: If it is not possible to get information about
+ the bridge domains.
+ """
+
+ status_code, resp = HcUtil.get_honeycomb_data(node,
+ "config_bridge_domain")
+ if status_code != HTTPCodes.OK:
+ if ignore_404 and status_code == HTTPCodes.NOT_FOUND:
+ br_domains = list()
+ logger.debug("Error 404 ignored")
+ else:
+ raise HoneycombError(
+ "Not possible to get information about the bridge domains. "
+ "Status code: {0}.".format(status_code))
+ else:
+ try:
+ br_domains = HcUtil.parse_json_response(resp, ("bridge-domains",
+ "bridge-domain"))
+ except KeyError:
+ return []
+ return br_domains
+
+ @staticmethod
+ def get_bd_cfg_data(node, name):
+ """Get configuration data about the given bridge domain from Honeycomb.
+
+ :param node: Honeycomb node.
+ :param name: The name of bridge domain.
+ :type node: dict
+ :type name: str
+ :return: Configuration data about the given bridge domain from
+ Honeycomb.
+ :rtype: dict
+ """
+
+ br_domains = BridgeDomainKeywords.get_all_bds_cfg_data(node)
+ for br_domain in br_domains:
+ if br_domain["name"] == name:
+ return br_domain
+ return {}
+
+ @staticmethod
+ def delete_all_bridge_domains(node):
+ """Delete all bridge domains on Honeycomb node.
+
+ :param node: Honeycomb node.
+ :type node: dict
+ :return: Response from DELETE request.
+ :rtype: str
+ :raises HoneycombError: If it is not possible to delete all bridge
+ domains.
+ """
+
+ status_code, resp = HcUtil.delete_honeycomb_data(node,
+ "config_bridge_domain")
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError(
+ "Not possible to delete all bridge domains. "
+ "Status code: {0}.".format(status_code))
+ return resp
+
+ @staticmethod
+ def remove_bridge_domain(node, name):
+ """Remove one bridge domain from Honeycomb.
+
+ :param node: Honeycomb node.
+ :param name: Name of the bridge domain to be removed.
+ :type node: dict
+ :type name: str
+ :return: True if the bridge domain was removed.
+ :rtype: bool
+ :raises HoneycombError: If it is not possible to remove the bridge
+ domain.
+ """
+
+ br_domains = BridgeDomainKeywords.get_all_bds_cfg_data(node)
+ for br_domain in br_domains:
+ if br_domain["name"] == name:
+ br_domains.remove(br_domain)
+ brd_info = {"bridge-domains": {"bridge-domain": br_domains}}
+ status_code, _ = HcUtil.put_honeycomb_data(
+ node, "config_bridge_domain", dumps(brd_info))
+ if status_code != HTTPCodes.OK:
+ raise HoneycombError(
+ "Bridge domain '{0}' was not deleted. "
+ "Status code: {1}.".format(name, status_code))
+ return True
+
+ raise HoneycombError("Not possible to delete bridge domain '{0}'. The "
+ "bridge domain was not found".format(name))
diff --git a/resources/libraries/python/HoneycombUtil.py b/resources/libraries/python/HoneycombUtil.py
index 92b545d1fc..7ffa6430ff 100644
--- a/resources/libraries/python/HoneycombUtil.py
+++ b/resources/libraries/python/HoneycombUtil.py
@@ -23,6 +23,7 @@ HoneycombAPIKeywords instead.
"""
from json import loads
+from enum import Enum, unique
from robot.api import logger
@@ -30,6 +31,30 @@ from resources.libraries.python.HTTPRequest import HTTPRequest
from resources.libraries.python.constants import Constants as Const
+@unique
+class DataRepresentation(Enum):
+ """Representation of data sent by PUT and POST requests."""
+ NO_DATA = 0
+ JSON = 1
+ XML = 2
+ TXT = 3
+
+
+# Headers used in requests. Key - content representation, value - header.
+HEADERS = {DataRepresentation.NO_DATA:
+ {}, # Must be empty.
+ DataRepresentation.JSON:
+ {"Content-Type": "application/json",
+ "Accept": "text/plain"},
+ DataRepresentation.XML:
+ {"Content-Type": "application/xml",
+ "Accept": "text/plain"},
+ DataRepresentation.TXT:
+ {"Content-Type": "text/plain",
+ "Accept": "text/plain"}
+ }
+
+
class HoneycombError(Exception):
"""Exception(s) raised by methods working with Honeycomb.
@@ -166,53 +191,77 @@ class HoneycombUtil(object):
without extension, not the full path.
:type node: dict
:type url_file: str
- :return: Requested information.
- :rtype list
+ :return: Status code and content of response.
+ :rtype tuple
"""
path = HoneycombUtil.read_path_from_url_file(url_file)
- status_code, resp = HTTPRequest.get(node, path)
- return status_code, resp
+ return HTTPRequest.get(node, path)
@staticmethod
- def put_honeycomb_data(node, url_file, data, data_representation='json'):
+ def put_honeycomb_data(node, url_file, data,
+ data_representation=DataRepresentation.JSON):
"""Send configuration data using PUT request and return the status code
- and response.
+ and response content.
:param node: Honeycomb node.
:param url_file: URL file. The argument contains only the name of file
without extension, not the full path.
:param data: Configuration data to be sent to Honeycomb.
- :param data_representation: How the data is represented. Supported types
- of representation are: json, xml and txt.
+ :param data_representation: How the data is represented.
:type node: dict
:type url_file: str
:type data: str
- :type data_representation: str
+ :type data_representation: DataRepresentation
:return: Status code and content of response.
:rtype: tuple
+ :raises HoneycombError: If the given data representation is not defined
+ in HEADERS.
"""
- headers = {'json':
- {"Content-Type": "application/json",
- 'Accept': 'text/plain'},
- 'xml':
- {"Content-Type": "application/xml",
- 'Accept': 'text/plain'},
- 'txt':
- {"Content-Type": "text/plain",
- 'Accept': 'text/plain'}
- }
try:
- header = headers[data_representation]
- except KeyError as err:
- raise HoneycombError("Wrong data type: {0}.".
+ header = HEADERS[data_representation]
+ except AttributeError as err:
+ raise HoneycombError("Wrong data representation: {0}.".
format(data_representation), repr(err))
path = HoneycombUtil.read_path_from_url_file(url_file)
- status_code, resp = HTTPRequest.put(node=node, path=path,
- headers=header, payload=data)
- return status_code, resp
+ return HTTPRequest.put(node=node, path=path, headers=header,
+ payload=data)
+
+ @staticmethod
+ def post_honeycomb_data(node, url_file, data=None,
+ data_representation=DataRepresentation.JSON,
+ timeout=10):
+ """Send a POST request and return the status code and response content.
+
+ :param node: Honeycomb node.
+ :param url_file: URL file. The argument contains only the name of file
+ without extension, not the full path.
+ :param data: Configuration data to be sent to Honeycomb.
+ :param data_representation: How the data is represented.
+ :param timeout: How long to wait for the server to send data before
+ giving up.
+ :type node: dict
+ :type url_file: str
+ :type data: str
+ :type data_representation: DataRepresentation
+ :type timeout: int
+ :return: Status code and content of response.
+ :rtype: tuple
+ :raises HoneycombError: If the given data representation is not defined
+ in HEADERS.
+ """
+
+ try:
+ header = HEADERS[data_representation]
+ except AttributeError as err:
+ raise HoneycombError("Wrong data representation: {0}.".
+ format(data_representation), repr(err))
+
+ path = HoneycombUtil.read_path_from_url_file(url_file)
+ return HTTPRequest.post(node=node, path=path, headers=header,
+ payload=data, timeout=timeout)
@staticmethod
def delete_honeycomb_data(node, url_file):
@@ -223,7 +272,7 @@ class HoneycombUtil(object):
without extension, not the full path.
:type node: dict
:type url_file: str
- :return: Status code and response.
+ :return: Status code and content of response.
:rtype tuple
"""