refactor test framework
Change-Id: I31da3b1857b6399f9899276a2d99cdd19436296c
Signed-off-by: Klement Sekera <ksekera@cisco.com>
Signed-off-by: Matej Klotton <mklotton@cisco.com>
Signed-off-by: Jan Gelety <jgelety@cisco.com>
Signed-off-by: Juraj Sloboda <jsloboda@cisco.com>
diff --git a/test/vpp_papi_provider.py b/test/vpp_papi_provider.py
new file mode 100644
index 0000000..b83cd43
--- /dev/null
+++ b/test/vpp_papi_provider.py
@@ -0,0 +1,316 @@
+import vpp_papi
+from logging import error
+from hook import Hook
+
+# from vnet/vnet/mpls/mpls_types.h
+MPLS_IETF_MAX_LABEL = 0xfffff
+MPLS_LABEL_INVALID = MPLS_IETF_MAX_LABEL + 1
+
+
+class VppPapiProvider(object):
+ """VPP-api provider using vpp-papi
+
+ @property hook: hook object providing before and after api/cli hooks
+
+
+ """
+
+ def __init__(self, name, shm_prefix):
+ self.hook = Hook()
+ self.name = name
+ self.shm_prefix = shm_prefix
+
+ def register_hook(self, hook):
+ """Replace hook registration with new hook
+
+ :param hook:
+
+ """
+ self.hook = hook
+
+ def connect(self):
+ """Connect the API to VPP"""
+ vpp_papi.connect(self.name, self.shm_prefix)
+
+ def disconnect(self):
+ """Disconnect the API from VPP"""
+ vpp_papi.disconnect()
+
+ def api(self, api_fn, api_args, expected_retval=0):
+ """Call API function and check it's return value
+ Call the appropriate hooks before and after the API call
+
+ :param api_fn: API function to call
+ :param api_args: tuple of API function arguments
+ :param expected_retval: Expected return value (Default value = 0)
+ :returns: reply from the API
+
+ """
+ self.hook.before_api(api_fn.__name__, api_args)
+ reply = api_fn(*api_args)
+ if hasattr(reply, 'retval') and reply.retval != expected_retval:
+ msg = "API call failed, expected retval == %d, got %s" % (
+ expected_retval, repr(reply))
+ error(msg)
+ raise Exception(msg)
+ self.hook.after_api(api_fn.__name__, api_args)
+ return reply
+
+ def cli(self, cli):
+ """Execute a CLI, calling the before/after hooks appropriately
+
+ :param cli: CLI to execute
+ :returns: CLI output
+
+ """
+ self.hook.before_cli(cli)
+ cli += '\n'
+ r = vpp_papi.cli_inband(len(cli), cli)
+ self.hook.after_cli(cli)
+ if(hasattr(r, 'reply')):
+ return r.reply[0].decode().rstrip('\x00')
+
+ def show_version(self):
+ """ """
+ return vpp_papi.show_version()
+
+ def pg_create_interface(self, pg_index):
+ """
+
+ :param pg_index:
+
+ """
+ return self.api(vpp_papi.pg_create_interface, (pg_index, ))
+
+ def sw_interface_dump(self, filter=None):
+ """
+
+ :param filter: (Default value = None)
+
+ """
+ if filter is not None:
+ args = (1, filter)
+ else:
+ args = (0, b'')
+ return self.api(vpp_papi.sw_interface_dump, args)
+
+ def sw_interface_add_del_address(self, sw_if_index, addr, addr_len,
+ is_ipv6=0, is_add=1, del_all=0):
+ """
+
+ :param addr: param is_ipv6: (Default value = 0)
+ :param sw_if_index:
+ :param addr_len:
+ :param is_ipv6: (Default value = 0)
+ :param is_add: (Default value = 1)
+ :param del_all: (Default value = 0)
+
+ """
+ return self.api(vpp_papi.sw_interface_add_del_address,
+ (sw_if_index, is_add, is_ipv6, del_all, addr_len, addr))
+
+ def sw_interface_ra_suppress(self, sw_if_index):
+ suppress = 1
+ managed = 0
+ other = 0
+ ll_option = 0
+ send_unicast = 0
+ cease = 0
+ is_no = 0
+ default_router = 0
+ max_interval = 0
+ min_interval = 0
+ lifetime = 0
+ initial_count = 0
+ initial_interval = 0
+ async = False
+ return self.api(vpp_papi.sw_interface_ip6nd_ra_config,
+ (sw_if_index, suppress, managed, other,
+ ll_option, send_unicast, cease, is_no,
+ default_router, max_interval, min_interval,
+ lifetime, initial_count, initial_interval, async))
+
+
+ def vxlan_add_del_tunnel(
+ self,
+ src_addr,
+ dst_addr,
+ is_add=1,
+ is_ipv6=0,
+ encap_vrf_id=0,
+ decap_next_index=0xFFFFFFFF,
+ vni=0):
+ """
+
+ :param dst_addr:
+ :param src_addr:
+ :param is_add: (Default value = 1)
+ :param is_ipv6: (Default value = 0)
+ :param encap_vrf_id: (Default value = 0)
+ :param decap_next_index: (Default value = 0xFFFFFFFF)
+ :param vni: (Default value = 0)
+
+ """
+ return self.api(vpp_papi.vxlan_add_del_tunnel,
+ (is_add, is_ipv6, src_addr, dst_addr, encap_vrf_id,
+ decap_next_index, vni))
+
+ def sw_interface_set_l2_bridge(self, sw_if_index, bd_id,
+ shg=0, bvi=0, enable=1):
+ """
+
+ :param bd_id:
+ :param sw_if_index:
+ :param shg: (Default value = 0)
+ :param bvi: (Default value = 0)
+ :param enable: (Default value = 1)
+
+ """
+ return self.api(vpp_papi.sw_interface_set_l2_bridge,
+ (sw_if_index, bd_id, shg, bvi, enable))
+
+ def sw_interface_set_l2_xconnect(self, rx_sw_if_index, tx_sw_if_index,
+ enable):
+ """Create or delete unidirectional cross-connect from Tx interface to
+ Rx interface.
+
+ :param rx_sw_if_index: Software interface index of Rx interface.
+ :param tx_sw_if_index: Software interface index of Tx interface.
+ :param enable: Create cross-connect if equal to 1, delete cross-connect
+ if equal to 0.
+ :type rx_sw_if_index: str or int
+ :type rx_sw_if_index: str or int
+ :type enable: int
+
+ """
+ return self.api(vpp_papi.sw_interface_set_l2_xconnect,
+ (rx_sw_if_index, tx_sw_if_index, enable))
+
+ def sw_interface_set_flags(self, sw_if_index, admin_up_down,
+ link_up_down=0, deleted=0):
+ """
+
+ :param admin_up_down:
+ :param sw_if_index:
+ :param link_up_down: (Default value = 0)
+ :param deleted: (Default value = 0)
+
+ """
+ return self.api(vpp_papi.sw_interface_set_flags,
+ (sw_if_index, admin_up_down, link_up_down, deleted))
+
+ def create_subif(self, sw_if_index, sub_id, outer_vlan, inner_vlan,
+ no_tags=0, one_tag=0, two_tags=0, dot1ad=0, exact_match=0,
+ default_sub=0, outer_vlan_id_any=0, inner_vlan_id_any=0):
+ """Create subinterface
+ from vpe.api: set dot1ad = 0 for dot1q, set dot1ad = 1 for dot1ad
+
+ :param sub_id: param inner_vlan:
+ :param sw_if_index:
+ :param outer_vlan:
+ :param inner_vlan:
+ :param no_tags: (Default value = 0)
+ :param one_tag: (Default value = 0)
+ :param two_tags: (Default value = 0)
+ :param dot1ad: (Default value = 0)
+ :param exact_match: (Default value = 0)
+ :param default_sub: (Default value = 0)
+ :param outer_vlan_id_any: (Default value = 0)
+ :param inner_vlan_id_any: (Default value = 0)
+
+ """
+ return self.api(
+ vpp_papi.create_subif,
+ (sw_if_index,
+ sub_id,
+ no_tags,
+ one_tag,
+ two_tags,
+ dot1ad,
+ exact_match,
+ default_sub,
+ outer_vlan_id_any,
+ inner_vlan_id_any,
+ outer_vlan,
+ inner_vlan))
+
+ def create_vlan_subif(self, sw_if_index, vlan):
+ """
+
+ :param vlan:
+ :param sw_if_index:
+
+ """
+ return self.api(vpp_papi.create_vlan_subif, (sw_if_index, vlan))
+
+ def ip_add_del_route(
+ self,
+ dst_address,
+ dst_address_length,
+ next_hop_address,
+ next_hop_sw_if_index=0xFFFFFFFF,
+ table_id=0,
+ resolve_attempts=0,
+ classify_table_index=0xFFFFFFFF,
+ next_hop_out_label=MPLS_LABEL_INVALID,
+ next_hop_table_id=0,
+ create_vrf_if_needed=0,
+ resolve_if_needed=0,
+ is_add=1,
+ is_drop=0,
+ is_ipv6=0,
+ is_local=0,
+ is_classify=0,
+ is_multipath=0,
+ is_resolve_host=0,
+ is_resolve_attached=0,
+ not_last=0,
+ next_hop_weight=1):
+ """
+
+ :param dst_address_length:
+ :param next_hop_sw_if_index: (Default value = 0xFFFFFFFF)
+ :param dst_address:
+ :param next_hop_address:
+ :param next_hop_sw_if_index: (Default value = 0xFFFFFFFF)
+ :param vrf_id: (Default value = 0)
+ :param lookup_in_vrf: (Default value = 0)
+ :param resolve_attempts: (Default value = 0)
+ :param classify_table_index: (Default value = 0xFFFFFFFF)
+ :param create_vrf_if_needed: (Default value = 0)
+ :param resolve_if_needed: (Default value = 0)
+ :param is_add: (Default value = 1)
+ :param is_drop: (Default value = 0)
+ :param is_ipv6: (Default value = 0)
+ :param is_local: (Default value = 0)
+ :param is_classify: (Default value = 0)
+ :param is_multipath: (Default value = 0)
+ :param is_resolve_host: (Default value = 0)
+ :param is_resolve_attached: (Default value = 0)
+ :param not_last: (Default value = 0)
+ :param next_hop_weight: (Default value = 1)
+
+ """
+ return self.api(
+ vpp_papi.ip_add_del_route,
+ (next_hop_sw_if_index,
+ table_id,
+ resolve_attempts,
+ classify_table_index,
+ next_hop_out_label,
+ next_hop_table_id,
+ create_vrf_if_needed,
+ resolve_if_needed,
+ is_add,
+ is_drop,
+ is_ipv6,
+ is_local,
+ is_classify,
+ is_multipath,
+ is_resolve_host,
+ is_resolve_attached,
+ not_last,
+ next_hop_weight,
+ dst_address_length,
+ dst_address,
+ next_hop_address))