vstf/vstf/agent/unittest/env/test_origin_driver.py

   1 """
   2 Created on 2015-10-9
   3
   4 @author: y00228926
   5 """
   6 import unittest
   7
   8 from vstf.common.utils import check_output
   9 from vstf.agent.unittest.env import model
  10 from vstf.agent.env.driver_plugins import origin_driver
  11
  12
  13 class Test(model.Test):
  14
  15     def setUp(self):
  16         super(Test, self).setUp()
  17         self.driver_mgr = origin_driver.OriginDriverPlugin()
  18
  19     def tearDown(self):
  20         super(Test, self).tearDown()
  21
  22     def _driver_exists(self, drivers=[]):
  23         all_drivers = check_output("lsmod | awk '{print $1}'",shell = True).split()
  24         for mod in drivers:
  25             if mod not in all_drivers:
  26                 return False
  27         return True
  28
  29     def test_load(self):
  30         self.driver_mgr.clean()
  31         self.assertFalse(self._driver_exists(self.driver_mgr.get_supported_drivers()))
  32
  33         self.driver_mgr.load(['ixgbe','vhost_net'])
  34         self.assertTrue(self._driver_exists(['ixgbe','vhost_net']))
  35
  36         self.driver_mgr.clean()
  37         self.assertFalse(self._driver_exists(self.driver_mgr.get_supported_drivers()))
  38
  39
  40 if __name__ == "__main__":
  41     import logging
  42     logging.basicConfig(level = logging.INFO)
  43     LOG = logging.getLogger(__name__)
  44     unittest.main()