GBP Endpoint Learning

[vpp.git] / test / vpp_mac.py

diff --git a/test/vpp_mac.py b/test/vpp_mac.py
index c9ee11e..c0e6974 100644 (file)
--- a/test/vpp_mac.py
+++ b/test/vpp_mac.py
@@ -22,3 +22,18 @@ class VppMacAddress():
     @property
     def address(self):
         return self.addr.address
+
+    def __str__(self):
+        return self.address
+
+    def __eq__(self, other):
+        if isinstance(other, self.__class__):
+            return self.address == other.addres
+        elif hasattr(other, "bytes"):
+            # vl_api_mac_addres_t
+            return self.bytes == other.bytes
+        else:
+            raise Exception("Comparing VppMacAddress:%s"
+                            "with unknown type: %s" %
+                            (self, other))
+        return False