X-Git-Url: https://gerrit.fd.io/r/gitweb?a=blobdiff_plain;f=test%2Fvpp_l2.py;h=21972bf4148a5638744afdc8e5591a56d36a6752;hb=56e17cf7a23831b353bc1836def47dad7f869dad;hp=dc73e4238f38d162050e99ed0bfe14c94bb5f6a5;hpb=36abbf10045ba81733410f50a3f5a9463f7137d1;p=vpp.git diff --git a/test/vpp_l2.py b/test/vpp_l2.py index dc73e4238f3..21972bf4148 100644 --- a/test/vpp_l2.py +++ b/test/vpp_l2.py @@ -4,9 +4,14 @@ """ from vpp_object import VppObject -from vpp_ip import VppIpAddress from vpp_lo_interface import VppLoInterface from vpp_papi import MACAddress +from vpp_sub_interface import L2_VTR_OP + +try: + text_type = unicode +except NameError: + text_type = str class L2_PORT_TYPE: @@ -22,6 +27,7 @@ class BRIDGE_FLAGS: FLOOD = 4 UU_FLOOD = 8 ARP_TERM = 16 + ARP_UFWD = 32 def find_bridge_domain(test, bd_id): @@ -39,19 +45,10 @@ def find_bridge_domain_port(test, bd_id, sw_if_index): def find_bridge_domain_arp_entry(test, bd_id, mac, ip): - vmac = MACAddress(mac) - vip = VppIpAddress(ip) - - if vip.version == 4: - n = 4 - else: - n = 16 - arps = test.vapi.bd_ip_mac_dump(bd_id) for arp in arps: # do IP addr comparison too once .api is fixed... - if vmac.packed == arp.mac_address and \ - vip.bytes == arp.ip_address[:n]: + if mac == str(arp.entry.mac) and ip == str(arp.entry.ip): return True return False @@ -66,10 +63,17 @@ def find_l2_fib_entry(test, bd_id, mac, sw_if_index): class VppBridgeDomain(VppObject): - - def __init__(self, test, bd_id, - flood=1, uu_flood=1, forward=1, - learn=1, arp_term=1): + def __init__( + self, + test, + bd_id, + flood=1, + uu_flood=1, + forward=1, + learn=1, + arp_term=1, + arp_ufwd=0, + ): self._test = test self.bd_id = bd_id self.flood = flood @@ -77,33 +81,34 @@ class VppBridgeDomain(VppObject): self.forward = forward self.learn = learn self.arp_term = arp_term + self.arp_ufwd = arp_ufwd def add_vpp_config(self): - self._test.vapi.bridge_domain_add_del(bd_id=self.bd_id, - flood=self.flood, - uu_flood=self.uu_flood, - forward=self.forward, - learn=self.learn, - arp_term=self.arp_term, is_add=1) + self._test.vapi.bridge_domain_add_del_v2( + bd_id=self.bd_id, + flood=self.flood, + uu_flood=self.uu_flood, + forward=self.forward, + learn=self.learn, + arp_term=self.arp_term, + arp_ufwd=self.arp_ufwd, + is_add=1, + ) self._test.registry.register(self, self._test.logger) + return self def remove_vpp_config(self): - self._test.vapi.bridge_domain_add_del(bd_id=self.bd_id, is_add=0) + self._test.vapi.bridge_domain_add_del_v2(bd_id=self.bd_id, is_add=0) def query_vpp_config(self): return find_bridge_domain(self._test, self.bd_id) - def __str__(self): - return self.object_id() - def object_id(self): return "bridge-domain-%d" % (self.bd_id) class VppBridgeDomainPort(VppObject): - - def __init__(self, test, bd, itf, - port_type=L2_PORT_TYPE.NORMAL): + def __init__(self, test, bd, itf, port_type=L2_PORT_TYPE.NORMAL): self._test = test self.bd = bd self.itf = itf @@ -111,63 +116,59 @@ class VppBridgeDomainPort(VppObject): def add_vpp_config(self): self._test.vapi.sw_interface_set_l2_bridge( - rx_sw_if_index=self.itf.sw_if_index, bd_id=self.bd.bd_id, - port_type=self.port_type, enable=1) + rx_sw_if_index=self.itf.sw_if_index, + bd_id=self.bd.bd_id, + port_type=self.port_type, + enable=1, + ) self._test.registry.register(self, self._test.logger) + return self def remove_vpp_config(self): self._test.vapi.sw_interface_set_l2_bridge( - rx_sw_if_index=self.itf.sw_if_index, bd_id=self.bd.bd_id, - port_type=self.port_type, enable=0) + rx_sw_if_index=self.itf.sw_if_index, + bd_id=self.bd.bd_id, + port_type=self.port_type, + enable=0, + ) def query_vpp_config(self): - return find_bridge_domain_port(self._test, - self.bd.bd_id, - self.itf.sw_if_index) - - def __str__(self): - return self.object_id() + return find_bridge_domain_port(self._test, self.bd.bd_id, self.itf.sw_if_index) def object_id(self): return "BD-Port-%s-%s" % (self.bd, self.itf) class VppBridgeDomainArpEntry(VppObject): - def __init__(self, test, bd, mac, ip): self._test = test self.bd = bd - self.mac = MACAddress(mac) - self.ip = VppIpAddress(ip) + self.mac = mac + self.ip = ip def add_vpp_config(self): - self._test.vapi.bd_ip_mac_add_del(bd_id=self.bd.bd_id, is_add=1, - ip=self.ip.encode(), - mac=self.mac.packed) + self._test.vapi.bd_ip_mac_add_del( + is_add=1, entry={"bd_id": self.bd.bd_id, "ip": self.ip, "mac": self.mac} + ) self._test.registry.register(self, self._test.logger) + return self def remove_vpp_config(self): - self._test.vapi.bd_ip_mac_add_del(bd_id=self.bd.bd_id, is_add=0, - ip=self.ip.encode(), - mac=self.mac.packed) + self._test.vapi.bd_ip_mac_add_del( + is_add=0, entry={"bd_id": self.bd.bd_id, "ip": self.ip, "mac": self.mac} + ) def query_vpp_config(self): - return find_bridge_domain_arp_entry(self._test, - self.bd.bd_id, - self.mac.packed, - self.ip.address) - - def __str__(self): - return self.object_id() + return find_bridge_domain_arp_entry( + self._test, self.bd.bd_id, self.mac, self.ip + ) def object_id(self): - return "BD-Arp-Entry-%s-%s-%s" % (self.bd, self.mac, self.ip.address) + return "BD-Arp-Entry-%s-%s-%s" % (self.bd, self.mac, self.ip) class VppL2FibEntry(VppObject): - - def __init__(self, test, bd, mac, itf, - static_mac=0, filter_mac=0, bvi_mac=-1): + def __init__(self, test, bd, mac, itf, static_mac=0, filter_mac=0, bvi_mac=-1): self._test = test self.bd = bd self.mac = MACAddress(mac) @@ -187,31 +188,26 @@ class VppL2FibEntry(VppObject): is_add=1, static_mac=self.static_mac, filter_mac=self.filter_mac, - bvi_mac=self.bvi_mac) + bvi_mac=self.bvi_mac, + ) self._test.registry.register(self, self._test.logger) + return self def remove_vpp_config(self): self._test.vapi.l2fib_add_del( - self.mac.packed, - self.bd.bd_id, - self.itf.sw_if_index, - is_add=0) + self.mac.packed, self.bd.bd_id, self.itf.sw_if_index, is_add=0 + ) def query_vpp_config(self): - return find_l2_fib_entry(self._test, - self.bd.bd_id, - self.mac.packed, - self.itf.sw_if_index) - - def __str__(self): - return self.object_id() + return find_l2_fib_entry( + self._test, self.bd.bd_id, self.mac.packed, self.itf.sw_if_index + ) def object_id(self): return "L2-Fib-Entry-%s-%s-%s" % (self.bd, self.mac, self.itf) class VppL2Vtr(VppObject): - def __init__(self, test, itf, op): self._test = test self.itf = itf @@ -220,6 +216,7 @@ class VppL2Vtr(VppObject): def add_vpp_config(self): self.itf.set_vtr(self.op) self._test.registry.register(self, self._test.logger) + return self def remove_vpp_config(self): self.itf.set_vtr(L2_VTR_OP.L2_DISABLED) @@ -229,11 +226,8 @@ class VppL2Vtr(VppObject): d = self.itf.get_interface_config_from_dump(ds) if d is not None: - return (d.vtr_op == self.op) + return d.vtr_op == self.op return False - def __str__(self): - return self.object_id() - def object_id(self): return "L2-vtr-%s-%d" % (str(self.itf), self.op)