summaryrefslogtreecommitdiffstats
path: root/vstf/vstf/agent/unittest/env/test_builder.py
blob: fd2d4382cc6bf95ed835840c265f45a1ccd7601e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
"""
Created on 2015-9-24

@author: y00228926
"""
import unittest
import os

from vstf.controller.env_build.cfg_intent_parse import IntentParser
from vstf.agent.env import builder


class Test(unittest.TestCase):
    def setUp(self):
        self.mgr = builder.PluginManager()
        self.dir = os.path.dirname(__file__)

    def tearDown(self):
        self.mgr.clean()

    def __build(self, filepath, drivers=None):
        parser = IntentParser(filepath)
        cfg_intent = parser.parse_cfg_file()
        host_cfg = cfg_intent['env-build'][0]
        print filepath
        print host_cfg
        host_cfg["src-install"] = {}
        if drivers:
            host_cfg['drivers'] = drivers
        return self.mgr.build(host_cfg)

    def test_build_tn_using_origin_driver(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Tn.json'), drivers=['ixgbe'])
        self.assertTrue(ret, "test_build_tn_using_origin_driver failed, ret = %s" % ret)

    def test_build_tn1v_using_origin_driver(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Tn1v.json'), drivers=['ixgbe', 'vhost_net'])
        self.assertTrue(ret, "test_build_tn1v_using_origin_driver failed, ret = %s" % ret)

    def test_build_ti_using_origin_driver(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Ti.json'), drivers=['ixgbe', 'vhost_net'])
        self.assertTrue(ret, "test_build_ti_using_origin_driver failed, ret = %s" % ret)

    def test_build_tu_using_origin_driver(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Tu.json'), drivers=['vhost_net'])
        self.assertTrue(ret, "test_build_ti_using_origin_driver failed, ret = %s" % ret)

    @unittest.skip('can be tested by tn1v')
    def test_build_tn(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Tn.json'))
        self.assertTrue(ret, "test_build_tn failed, ret = %s" % ret)

    @unittest.skip('can be tested by tn1v')
    def test_build_tn1v(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Tn1v.json'))
        self.assertTrue(ret, "test_build_tn failed,ret = %s" % ret)

    @unittest.skip('can be tested by tn1v')
    def test_build_ti(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Ti.json'))
        self.assertTrue(ret, "test_build_tn failed, ret = %s" % ret)

    @unittest.skip('can be tested by tn1v')
    def test_build_tu(self):
        ret = self.__build(os.path.join(self.dir, 'configuration/Tu.json'))
        self.assertTrue(ret, "test_build_tn failed, ret = %s" % ret)


if __name__ == "__main__":
    import logging

    LOG = logging.getLogger(__name__)
    logging.basicConfig(level=logging.DEBUG)
    unittest.main()