vhost-user: Add disable feature support in api

[vpp.git] / test / vpp_vhost_interface.py

diff --git a/test/vpp_vhost_interface.py b/test/vpp_vhost_interface.py
index 2249b06..0a5f6b7 100644 (file)
--- a/test/vpp_vhost_interface.py
+++ b/test/vpp_vhost_interface.py
@@ -5,14 +5,17 @@ class VppVhostInterface(VppInterface):
     """VPP vhost interface."""
 
     def __init__(self, test, sock_filename, is_server=0, renumber=0,
+                 disable_mrg_rxbuf=0, disable_indirect_desc=0,
                  custom_dev_instance=0, use_custom_mac=0, mac_address='',
                  tag=''):
 
         """ Create VPP Vhost interface """
-        self._test = test
+        super(VppVhostInterface, self).__init__(test)
         self.is_server = is_server
         self.sock_filename = sock_filename
         self.renumber = renumber
+        self.disable_mrg_rxbuf = disable_mrg_rxbuf
+        self.disable_indirect_desc = disable_indirect_desc
         self.custom_dev_instance = custom_dev_instance
         self.use_custom_mac = use_custom_mac
         self.mac_address = mac_address
@@ -22,12 +25,13 @@ class VppVhostInterface(VppInterface):
         r = self.test.vapi.create_vhost_user_if(self.is_server,
                                                 self.sock_filename,
                                                 self.renumber,
+                                                self.disable_mrg_rxbuf,
+                                                self.disable_indirect_desc,
                                                 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)
+        self.set_sw_if_index(r.sw_if_index)
 
     def remove_vpp_config(self):
         self.test.vapi.delete_vhost_user_if(self.sw_if_index)