[dts] [PATCH V1] tests/TestSuite_vf_daemon:Improve the way drivers bind

["zhang,yan" <yanx.a.zhang@intel.com>]

Improved the original single binding driver.

Signed-off-by: zhang,yan <yanx.a.zhang@intel.com>
---
 tests/TestSuite_vf_daemon.py | 22 +++++++++++++++++-----
 1 file changed, 17 insertions(+), 5 deletions(-)

diff --git a/tests/TestSuite_vf_daemon.py b/tests/TestSuite_vf_daemon.py
index cc55f65..fcbf5b8 100644
--- a/tests/TestSuite_vf_daemon.py
+++ b/tests/TestSuite_vf_daemon.py
@@ -17,6 +17,7 @@ MAX_VLAN = 4095
 
 
 class TestVfDaemon(TestCase):
+    supported_vf_driver = ['pci-stub', 'vfio-pci']
 
     def set_up_all(self):
 
@@ -26,6 +27,15 @@ class TestVfDaemon(TestCase):
         self.vm1 = None
         self.env_done = False
 
+        self.vf_driver = self.get_suite_cfg()['vf_driver']
+        if self.vf_driver is None:
+            self.vf_driver = 'pci-stub'
+        self.verify(self.vf_driver in self.supported_vf_driver, "Unspported vf driver")
+        if self.vf_driver == 'pci-stub':
+            self.vf_assign_method = 'pci-assign'
+        else:
+            self.vf_assign_method = 'vfio-pci'
+            self.dut.send_expect('modprobe vfio-pci', '#')
 
     def set_up(self):
         self.setup_vm_env()
@@ -61,7 +71,7 @@ class TestVfDaemon(TestCase):
                     netdev.bind_driver(driver=driver)
 
 
-    def setup_vm_env(self, driver='igb_uio'):
+    def setup_vm_env(self, driver='default'):
         """
         Create testing environment with 2VFs generated from 1PF
         """
@@ -78,12 +88,14 @@ class TestVfDaemon(TestCase):
         self.sriov_vfs_port = self.dut.ports_info[
             self.used_dut_port]['vfs_port']
         for port in self.sriov_vfs_port:
-                port.bind_driver('pci-stub')
+            port.bind_driver(self.vf_driver)
         time.sleep(1)
+        eal_param = '-b %(vf0)s -b %(vf1)s' % {'vf0': self.sriov_vfs_port[0].pci,
+                                               'vf1': self.sriov_vfs_port[1].pci}
 
         self.dut_testpmd = PmdOutput(self.dut)
         self.dut_testpmd.start_testpmd(
-            "Default", "--rxq=4 --txq=4 --port-topology=chained")
+            "Default", "--rxq=4 --txq=4 --port-topology=chained", eal_param=eal_param)
         self.dut_testpmd.execute_cmd("start")
         time.sleep(5)
 
@@ -91,7 +103,7 @@ class TestVfDaemon(TestCase):
         
         # set up VM0 ENV
         self.vm0 = QEMUKvm(self.dut, 'vm0', 'vf_daemon')
-        self.vm0.set_vm_device(driver='pci-assign', **vf0_prop)
+        self.vm0.set_vm_device(driver=self.vf_assign_method, **vf0_prop)
         try:
             self.vm0_dut = self.vm0.start()
             if self.vm0_dut is None:
@@ -105,7 +117,7 @@ class TestVfDaemon(TestCase):
 
         vf1_prop = {'opt_host': self.sriov_vfs_port[1].pci}
         self.vm1 = QEMUKvm(self.dut, 'vm1', 'vf_daemon')
-        self.vm1.set_vm_device(driver='pci-assign', **vf1_prop)
+        self.vm1.set_vm_device(driver=self.vf_assign_method, **vf1_prop)
         try:
             self.vm1_dut = self.vm1.start()
             if self.vm1_dut is None:
-- 
2.17.2