vstf/vstf/agent/unittest/env/test_builder.py

   1 """
   2 Created on 2015-9-24
   3
   4 @author: y00228926
   5 """
   6 import unittest
   7 import os
   8
   9 from vstf.controller.env_build.cfg_intent_parse import IntentParser
  10 from vstf.agent.env import builder
  11
  12
  13 class Test(unittest.TestCase):
  14     def setUp(self):
  15         self.mgr = builder.PluginManager()
  16         self.dir = os.path.dirname(__file__)
  17
  18     def tearDown(self):
  19         self.mgr.clean()
  20
  21     def __build(self, filepath, drivers=None):
  22         parser = IntentParser(filepath)
  23         cfg_intent = parser.parse_cfg_file()
  24         host_cfg = cfg_intent['env-build'][0]
  25         print filepath
  26         print host_cfg
  27         host_cfg["src-install"] = {}
  28         if drivers:
  29             host_cfg['drivers'] = drivers
  30         return self.mgr.build(host_cfg)
  31
  32     def test_build_tn_using_origin_driver(self):
  33         ret = self.__build(os.path.join(self.dir, 'configuration/Tn.json'), drivers=['ixgbe'])
  34         self.assertTrue(ret, "test_build_tn_using_origin_driver failed, ret = %s" % ret)
  35
  36     def test_build_tn1v_using_origin_driver(self):
  37         ret = self.__build(os.path.join(self.dir, 'configuration/Tn1v.json'), drivers=['ixgbe', 'vhost_net'])
  38         self.assertTrue(ret, "test_build_tn1v_using_origin_driver failed, ret = %s" % ret)
  39
  40     def test_build_ti_using_origin_driver(self):
  41         ret = self.__build(os.path.join(self.dir, 'configuration/Ti.json'), drivers=['ixgbe', 'vhost_net'])
  42         self.assertTrue(ret, "test_build_ti_using_origin_driver failed, ret = %s" % ret)
  43
  44     def test_build_tu_using_origin_driver(self):
  45         ret = self.__build(os.path.join(self.dir, 'configuration/Tu.json'), drivers=['vhost_net'])
  46         self.assertTrue(ret, "test_build_ti_using_origin_driver failed, ret = %s" % ret)
  47
  48     @unittest.skip('can be tested by tn1v')
  49     def test_build_tn(self):
  50         ret = self.__build(os.path.join(self.dir, 'configuration/Tn.json'))
  51         self.assertTrue(ret, "test_build_tn failed, ret = %s" % ret)
  52
  53     @unittest.skip('can be tested by tn1v')
  54     def test_build_tn1v(self):
  55         ret = self.__build(os.path.join(self.dir, 'configuration/Tn1v.json'))
  56         self.assertTrue(ret, "test_build_tn failed,ret = %s" % ret)
  57
  58     @unittest.skip('can be tested by tn1v')
  59     def test_build_ti(self):
  60         ret = self.__build(os.path.join(self.dir, 'configuration/Ti.json'))
  61         self.assertTrue(ret, "test_build_tn failed, ret = %s" % ret)
  62
  63     @unittest.skip('can be tested by tn1v')
  64     def test_build_tu(self):
  65         ret = self.__build(os.path.join(self.dir, 'configuration/Tu.json'))
  66         self.assertTrue(ret, "test_build_tn failed, ret = %s" % ret)
  67
  68
  69 if __name__ == "__main__":
  70     import logging
  71
  72     LOG = logging.getLogger(__name__)
  73     logging.basicConfig(level=logging.DEBUG)
  74     unittest.main()