CLI implementation
[functest.git] / ci / tier_handler.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
11 import re
12
13 LINE_LENGTH = 72
14
15
16 def split_text(text, max_len):
17     words = text.split()
18     lines = []
19     line = ""
20     for word in words:
21         if len(line) + len(word) < max_len - 1:
22             line += word + " "
23         else:
24             lines.append(line)
25             line = word + " "
26     if line != "":
27         lines.append(line)
28     return lines
29
30
31 class Tier:
32     def __init__(self, name, order, ci, description=""):
33         self.tests_array = []
34         self.name = name
35         self.order = order
36         self.ci = ci
37         self.description = description
38
39     def add_test(self, testcase):
40         self.tests_array.append(testcase)
41
42     def get_tests(self):
43         array_tests = []
44         for test in self.tests_array:
45             array_tests.append(test)
46         return array_tests
47
48     def get_test_names(self):
49         array_tests = []
50         for test in self.tests_array:
51             array_tests.append(test.get_name())
52         return array_tests
53
54     def get_test(self, test_name):
55         if self.is_test(test_name):
56             for test in self.tests_array:
57                 if test.get_name() == test_name:
58                     return test
59         return None
60
61     def is_test(self, test_name):
62         for test in self.tests_array:
63             if test.get_name() == test_name:
64                 return True
65         return False
66
67     def get_name(self):
68         return self.name
69
70     def get_order(self):
71         return self.order
72
73     def __str__(self):
74         lines = split_text(self.description, LINE_LENGTH-6)
75
76         out = ""
77         out += ("+%s+\n" % ("=" * (LINE_LENGTH - 2)))
78         out += ("| Tier:  " + self.name.ljust(LINE_LENGTH - 10) + "|\n")
79         out += ("+%s+\n" % ("=" * (LINE_LENGTH - 2)))
80         out += ("| Order: " + str(self.order).ljust(LINE_LENGTH - 10) + "|\n")
81         out += ("| Description:".ljust(LINE_LENGTH - 1) + "|\n")
82         for line in lines:
83             out += ("|    " + line.ljust(LINE_LENGTH - 7) + " |\n")
84         out += ("| Test cases:".ljust(LINE_LENGTH - 1) + "|\n")
85         tests = self.get_test_names()
86         if len(tests) > 0:
87             for i in range(len(tests)):
88                 out += ("|    - %s |\n" % tests[i].ljust(LINE_LENGTH - 9))
89         else:
90             out += ("|    (There are no supported test cases "
91                     .ljust(LINE_LENGTH - 1) + "|\n")
92             out += ("|    in this tier for the given scenario) "
93                     .ljust(LINE_LENGTH - 1) + "|\n")
94         out += ("|".ljust(LINE_LENGTH - 1) + "|\n")
95         out += ("+%s+\n" % ("-" * (LINE_LENGTH - 2)))
96         return out
97
98
99 class TestCase:
100     def __init__(self, name, dependency, description=""):
101         self.name = name
102         self.dependency = dependency
103         self.description = description
104
105     def is_compatible(self, ci_installer, ci_scenario):
106         if re.search(self.dependency.get_installer(), ci_installer) is None:
107             return False
108
109         if re.search(self.dependency.get_scenario(), ci_scenario) is None:
110             return False
111
112         return True
113
114     def get_name(self):
115         return self.name
116
117     def __str__(self):
118         lines = split_text(self.description, LINE_LENGTH-6)
119
120         out = ""
121         out += ("+%s+\n" % ("=" * (LINE_LENGTH - 2)))
122         out += ("| Testcase:  " + self.name.ljust(LINE_LENGTH - 14) + "|\n")
123         out += ("+%s+\n" % ("=" * (LINE_LENGTH - 2)))
124         out += ("| Description:".ljust(LINE_LENGTH - 1) + "|\n")
125         for line in lines:
126             out += ("|    " + line.ljust(LINE_LENGTH - 7) + " |\n")
127         out += ("| Dependencies:".ljust(LINE_LENGTH - 1) + "|\n")
128         installer = self.dependency.get_installer()
129         scenario = self.dependency.get_scenario()
130         out += ("|   - Installer:" + installer.ljust(LINE_LENGTH - 17) + "|\n")
131         out += ("|   - Scenario :" + scenario.ljust(LINE_LENGTH - 17) + "|\n")
132         out += ("|".ljust(LINE_LENGTH - 1) + "|\n")
133         out += ("+%s+\n" % ("-" * (LINE_LENGTH - 2)))
134         return out
135
136
137 class Dependency:
138     def __init__(self, installer, scenario):
139         self.installer = installer
140         self.scenario = scenario
141
142     def get_installer(self):
143         return self.installer
144
145     def get_scenario(self):
146         return self.scenario
147
148     def __str__(self):
149         return ("Dependency info:\n"
150                 "        installer: " + self.installer + "\n"
151                 "        scenario:  " + self.scenario + "\n")