MAP: Use explicit address/prefix types in API

[vpp.git] / test / vpp_ip.py

diff --git a/test/vpp_ip.py b/test/vpp_ip.py
index 6d22c16..2083e03 100644 (file)
--- a/test/vpp_ip.py
+++ b/test/vpp_ip.py
@@ -200,6 +200,20 @@ class VppIpPrefix():
         return False
 
 
+class VppIp6Prefix():
+    def __init__(self, prefix, prefixlen):
+        self.ip_prefix = ip_address(unicode(prefix))
+        self.prefixlen = prefixlen
+
+    def encode(self):
+        return {'prefix': {'address': self.ip_prefix.packed},
+                'len': self.prefixlen}
+
+
+class VppIp4Prefix(VppIp6Prefix):
+    pass
+
+
 class VppIpMPrefix():
     def __init__(self, saddr, gaddr, len):
         self.saddr = saddr