vpp/test/vpp_neighbor.py
Paul Vinciguerra 3593a9174a tests: refactor vpp_object.py
Move __str__ to super for all subclasses of VppObject
Implement __repr__ in VppObject
Implement __hash__ and __eq__

Change-Id: Ibd4ea37b84b17f499ab86630fb5b9ed9c8b4b1c2
Signed-off-by: Paul Vinciguerra <pvinci@vinciconsulting.com>
2019-03-29 16:02:27 +00:00

86 lines
2.4 KiB
Python

"""
Neighbour Entries
object abstractions for ARP and ND
"""
from ipaddress import ip_address
from vpp_object import VppObject
from vpp_papi import mac_pton, VppEnum
try:
text_type = unicode
except NameError:
text_type = str
def find_nbr(test, sw_if_index, nbr_addr, is_static=0, mac=None):
ip_addr = ip_address(text_type(nbr_addr))
e = VppEnum.vl_api_ip_neighbor_flags_t
nbrs = test.vapi.ip_neighbor_dump(sw_if_index,
is_ipv6=(6 == ip_addr.version))
for n in nbrs:
if ip_addr == n.neighbor.ip_address and \
is_static == (n.neighbor.flags & e.IP_API_NEIGHBOR_FLAG_STATIC):
if mac:
if mac == str(n.neighbor.mac_address):
return True
else:
return True
return False
class VppNeighbor(VppObject):
"""
ARP Entry
"""
def __init__(self, test, sw_if_index, mac_addr, nbr_addr,
is_static=False, is_no_fib_entry=False):
self._test = test
self.sw_if_index = sw_if_index
self.mac_addr = mac_addr
self.nbr_addr = nbr_addr
e = VppEnum.vl_api_ip_neighbor_flags_t
self.flags = e.IP_API_NEIGHBOR_FLAG_NONE
if is_static:
self.flags |= e.IP_API_NEIGHBOR_FLAG_STATIC
if is_no_fib_entry:
self.flags |= e.IP_API_NEIGHBOR_FLAG_NO_FIB_ENTRY
def add_vpp_config(self):
r = self._test.vapi.ip_neighbor_add_del(
self.sw_if_index,
self.mac_addr,
self.nbr_addr,
is_add=1,
flags=self.flags)
self.stats_index = r.stats_index
self._test.registry.register(self, self._test.logger)
def remove_vpp_config(self):
self._test.vapi.ip_neighbor_add_del(
self.sw_if_index,
self.mac_addr,
self.nbr_addr,
is_add=0,
flags=self.flags)
def is_static(self):
e = VppEnum.vl_api_ip_neighbor_flags_t
return (self.flags & e.IP_API_NEIGHBOR_FLAG_STATIC)
def query_vpp_config(self):
return find_nbr(self._test,
self.sw_if_index,
self.nbr_addr,
self.is_static())
def object_id(self):
return ("%d:%s" % (self.sw_if_index, self.nbr_addr))
def get_stats(self):
c = self._test.statistics.get_counter("/net/adjacency")
return c[0][self.stats_index]