1 files changed, 40 insertions, 0 deletions
diff --git a/test/vpp_vhost_interface.py b/test/vpp_vhost_interface.py
new file mode 100644
index 00000000000..2249b062a64
--- /dev/null
+++ b/test/vpp_vhost_interface.py
@@ -0,0 +1,40 @@
+from vpp_interface import VppInterface
+
+
+class VppVhostInterface(VppInterface):
+    """VPP vhost interface."""
+
+    def __init__(self, test, sock_filename, is_server=0, renumber=0,
+                 custom_dev_instance=0, use_custom_mac=0, mac_address='',
+                 tag=''):
+
+        """ Create VPP Vhost interface """
+        self._test = test
+        self.is_server = is_server
+        self.sock_filename = sock_filename
+        self.renumber = renumber
+        self.custom_dev_instance = custom_dev_instance
+        self.use_custom_mac = use_custom_mac
+        self.mac_address = mac_address
+        self.tag = tag
+
+    def add_vpp_config(self):
+        r = self.test.vapi.create_vhost_user_if(self.is_server,
+                                                self.sock_filename,
+                                                self.renumber,
+                                                self.custom_dev_instance,
+                                                self.use_custom_mac,
+                                                self.mac_address,
+                                                self.tag)
+        self._sw_if_index = r.sw_if_index
+        super(VppVhostInterface, self).__init__(self._test)
+
+    def remove_vpp_config(self):
+        self.test.vapi.delete_vhost_user_if(self.sw_if_index)
+
+    def is_interface_config_in_dump(self, dump):
+        for i in dump:
+            if i.sw_if_index == self.sw_if_index:
+                return True
+        else:
+            return False