Merge "Remove useless odlreport2db args"
[functest.git] / ci / tier_builder.py
1 #!/usr/bin/env python
2 #
3 # jose.lausuch@ericsson.com
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 #
9
10 import tier_handler as th
11 import yaml
12
13
14 class TierBuilder:
15
16     def __init__(self, ci_installer, ci_scenario, testcases_file):
17         self.ci_installer = ci_installer
18         self.ci_scenario = ci_scenario
19         self.testcases_file = testcases_file
20         self.dic_tier_array = None
21         self.tier_objects = []
22         self.testcases_yaml = None
23         self.generate_tiers()
24
25     def read_test_yaml(self):
26         with open(self.testcases_file) as f:
27             self.testcases_yaml = yaml.safe_load(f)
28
29         self.dic_tier_array = []
30         for tier in self.testcases_yaml.get("tiers"):
31             self.dic_tier_array.append(tier)
32
33     def generate_tiers(self):
34         if self.dic_tier_array is None:
35             self.read_test_yaml()
36
37         del self.tier_objects[:]
38         for dic_tier in self.dic_tier_array:
39             tier = th.Tier(name=dic_tier['name'],
40                            order=dic_tier['order'],
41                            ci_loop=dic_tier['ci_loop'],
42                            description=dic_tier['description'])
43
44             for dic_testcase in dic_tier['testcases']:
45                 installer = dic_testcase['dependencies']['installer']
46                 scenario = dic_testcase['dependencies']['scenario']
47                 dep = th.Dependency(installer, scenario)
48
49                 testcase = th.TestCase(name=dic_testcase['name'],
50                                        dependency=dep,
51                                        criteria=dic_testcase['criteria'],
52                                        blocking=dic_testcase['blocking'],
53                                        description=dic_testcase['description'])
54                 if testcase.is_compatible(self.ci_installer, self.ci_scenario):
55                     tier.add_test(testcase)
56
57             self.tier_objects.append(tier)
58
59     def get_tiers(self):
60         return self.tier_objects
61
62     def get_tier_names(self):
63         tier_names = []
64         for tier in self.tier_objects:
65             tier_names.append(tier.get_name())
66         return tier_names
67
68     def get_tier(self, tier_name):
69         for i in range(0, len(self.tier_objects)):
70             if self.tier_objects[i].get_name() == tier_name:
71                 return self.tier_objects[i]
72         return None
73
74     def get_test(self, test_name):
75         for i in range(0, len(self.tier_objects)):
76             if self.tier_objects[i].is_test(test_name):
77                 return self.tier_objects[i].get_test(test_name)
78         return None
79
80     def get_tests(self, tier_name):
81         for i in range(0, len(self.tier_objects)):
82             if self.tier_objects[i].get_name() == tier_name:
83                 return self.tier_objects[i].get_tests()
84         return None
85
86     def __str__(self):
87         output = ""
88         for i in range(0, len(self.tier_objects)):
89             output += str(self.tier_objects[i]) + "\n"
90         return output