aboutsummaryrefslogtreecommitdiffstats
path: root/resources/libraries/python
diff options
context:
space:
mode:
Diffstat (limited to 'resources/libraries/python')
-rw-r--r--resources/libraries/python/LoadBalancerUtil.py33
1 files changed, 32 insertions, 1 deletions
diff --git a/resources/libraries/python/LoadBalancerUtil.py b/resources/libraries/python/LoadBalancerUtil.py
index 26bf965c39..77f6412973 100644
--- a/resources/libraries/python/LoadBalancerUtil.py
+++ b/resources/libraries/python/LoadBalancerUtil.py
@@ -15,7 +15,7 @@
from socket import htonl
from ipaddress import ip_address
-from resources.libraries.python.topology import NodeType
+from resources.libraries.python.topology import NodeType, Topology
from resources.libraries.python.PapiExecutor import PapiSocketExecutor
class LoadBalancerUtil(object):
@@ -174,3 +174,34 @@ class LoadBalancerUtil(object):
else:
raise ValueError('Node {host} has unknown NodeType: "{type}"'
.format(host=node['host'], type=node['type']))
+
+ @staticmethod
+ def vpp_lb_add_del_intf_nat4(node, **kwargs):
+ """Enable/disable NAT4 feature on the interface.
+
+ :param node: Node where the interface is.
+ :param kwargs: Optional key-value arguments:
+
+ is_add: true if add, false if delete. (bool)
+ interface: software index of the interface. (int)
+
+ :type node: dict
+ :type kwargs: dict
+ :returns: Nothing.
+ :raises ValueError: If the node has an unknown node type.
+ """
+ if node['type'] == NodeType.DUT:
+ cmd = 'lb_add_del_intf_nat4'
+ err_msg = 'Failed to add interface nat4 on host {host}'.format(
+ host=node['host'])
+
+ is_add = kwargs.pop('is_add', True)
+ interface = kwargs.pop('interface', 0)
+ sw_if_index = Topology.get_interface_sw_index(node, interface)
+ args = dict(is_add=is_add, sw_if_index=sw_if_index)
+
+ with PapiSocketExecutor(node) as papi_exec:
+ papi_exec.add(cmd, **args).get_reply(err_msg)
+ else:
+ raise ValueError('Node {host} has unknown NodeType: "{type}"'
+ .format(host=node['host'], type=node['type']))