diff options
Diffstat (limited to 'resources/libraries/python')
| -rw-r--r-- | resources/libraries/python/IPv4Util.py | 27 | ||||
| -rw-r--r-- | resources/libraries/python/L2Util.py | 46 |
2 files changed, 68 insertions, 5 deletions
diff --git a/resources/libraries/python/IPv4Util.py b/resources/libraries/python/IPv4Util.py index 5ee73c08fc..4fe711fbff 100644 --- a/resources/libraries/python/IPv4Util.py +++ b/resources/libraries/python/IPv4Util.py @@ -18,6 +18,7 @@ from robot.api.deco import keyword from resources.libraries.python.topology import Topology from resources.libraries.python.IPv4Setup import get_node +from resources.libraries.python.ssh import exec_cmd class IPv4Util(object): @@ -160,3 +161,29 @@ class IPv4Util(object): if net is None: raise ValueError('Link "{0}" not found'.format(link)) return net.get('prefix') + + @staticmethod + def send_ping_from_node_to_dst(node, destination, namespace=None, + ping_count=3): + """Send a ping from node to destination. Optionally, you can define a + namespace from where to send a ping. + + :param node: Node to start ping on. + :param destination: IPv4 address where to send ping. + :param namespace: Namespace to send ping from. + :param ping_count: Number of pings to send. + :type node: dict + :type destination: str + :type namespace: str + :type ping_count: int + :raises RuntimeError: If no response for ping, raise error + """ + cmd = '' + if namespace is not None: + cmd = 'ip netns exec {0} ping -c{1} {2}'.format( + namespace, ping_count, destination) + else: + cmd = 'ping -c{0} {1}'.format(ping_count, destination) + rc, stdout, stderr = exec_cmd(node, cmd, sudo=True) + if rc != 0: + raise RuntimeError("Ping Not Successful") diff --git a/resources/libraries/python/L2Util.py b/resources/libraries/python/L2Util.py index 5903d03549..e44a6b360a 100644 --- a/resources/libraries/python/L2Util.py +++ b/resources/libraries/python/L2Util.py @@ -21,7 +21,7 @@ from resources.libraries.python.ssh import exec_cmd_no_error class L2Util(object): - """Utilities for l2 configuration""" + """Utilities for l2 configuration.""" @staticmethod def vpp_add_l2fib_entry(node, mac, interface, bd_id): @@ -203,6 +203,7 @@ class L2Util(object): :type br_name: str :type if_1: str :type if_2: str + """ cmd = 'brctl addbr {0}'.format(br_name) exec_cmd_no_error(node, cmd, sudo=True) @@ -212,12 +213,41 @@ class L2Util(object): exec_cmd_no_error(node, cmd, sudo=True) @staticmethod + def setup_network_namespace(node, namespace_name, interface_name, + ip_address, prefix): + """Setup namespace on given node and attach interface and IP to + this namespace. Applicable also on TG node. + + :param node: Node to set namespace on. + :param namespace_name: Namespace name. + :param interface_name: Interface name. + :param ip_address: IP address of namespace's interface. + :param prefix: IP address prefix length. + :type node: dict + :type namespace_name: str + :type vhost_if: str + :type ip_address: str + :type prefix: int + + """ + cmd = ('ip netns add {0}'.format(namespace_name)) + exec_cmd_no_error(node, cmd, sudo=True) + + cmd = ('ip link set dev {0} up netns {1}'.format(interface_name, + namespace_name)) + exec_cmd_no_error(node, cmd, sudo=True) + + cmd = ('ip netns exec {0} ip addr add {1}/{2} dev {3}'.format( + namespace_name, ip_address, prefix, interface_name)) + exec_cmd_no_error(node, cmd, sudo=True) + + @staticmethod def linux_del_bridge(node, br_name): """Delete bridge from linux node. :param node: Node to delete bridge from. :param br_name: Bridge name. - .. note:: The network interface corresponding to the bridge must be + ..note:: The network interface corresponding to the bridge must be down before it can be deleted! """ cmd = 'brctl delbr {0}'.format(br_name) @@ -250,17 +280,22 @@ class L2Util(object): return data @staticmethod - def l2_tag_rewrite(node, interface, tag_rewrite_method): + def l2_tag_rewrite(node, interface, tag_rewrite_method, tag1_id=None): """Rewrite tags in frame. :param node: Node to rewrite tags. :param interface: Interface on which rewrite tags. :param tag_rewrite_method: Method of tag rewrite. + :param tag1_id: Optional tag1 ID for VLAN. :type node: dict :type interface: str or int :type tag_rewrite_method : str + :type tag1_id: int """ - + if tag1_id is None: + tag1_id = '' + else: + tag1_id = 'tag1 {0}'.format(tag1_id) if isinstance(interface, basestring): sw_if_index = Topology.get_interface_sw_index(node, interface) else: @@ -270,4 +305,5 @@ class L2Util(object): vat.vat_terminal_exec_cmd_from_template("l2_tag_rewrite.vat", sw_if_index=sw_if_index, tag_rewrite_method= - tag_rewrite_method) + tag_rewrite_method, + tag1_optional=tag1_id) |