8 from vstf.common.utils import check_output
9 from vstf.agent.unittest.env import model
10 from vstf.agent.env.driver_plugins import manager
13 class Test(model.Test):
15 super(Test, self).setUp()
16 self.driver_mgr = manager.DriverPluginManager()
19 super(Test, self).tearDown()
21 def _driver_exists(self, drivers=[]):
22 all_drivers = check_output("lsmod | awk '{print $1}'", shell=True).split()
24 if mod not in all_drivers:
29 self.driver_mgr.clean()
30 for _, drivers in self.driver_mgr.get_all_supported_drivers().items():
31 self.assertFalse(self._driver_exists(drivers))
34 self.driver_mgr.load(['ixgbe', 'vhost_net'])
35 self.assertTrue(self._driver_exists(['ixgbe', 'vhost_net']))
37 self.driver_mgr.clean()
38 self.assertFalse(self._driver_exists(['ixgbe', 'vhost_net']))
40 def test_load_unsuported_pair(self):
41 with self.assertRaises(Exception):
42 self.driver_mgr.load(['ixgbe', 'tap_vhost'])
45 if __name__ == "__main__":
48 logging.basicConfig(level=logging.INFO)
49 LOG = logging.getLogger(__name__)