simplify unit tests implementation of update scenario
[releng.git] / utils / test / testapi / opnfv_testapi / tests / unit / resources / test_scenario.py
1 import functools
2 import httplib
3 import json
4 import os
5
6 from datetime import datetime
7
8 from opnfv_testapi.common import message
9 import opnfv_testapi.resources.scenario_models as models
10 from opnfv_testapi.tests.unit.resources import test_base as base
11
12
13 def _none_default(check, default):
14     return check if check else default
15
16
17 class TestScenarioBase(base.TestBase):
18     def setUp(self):
19         super(TestScenarioBase, self).setUp()
20         self.get_res = models.Scenario
21         self.list_res = models.Scenarios
22         self.basePath = '/api/v1/scenarios'
23         self.req_d = self._load_request('scenario-c1.json')
24         self.req_2 = self._load_request('scenario-c2.json')
25
26     def tearDown(self):
27         pass
28
29     def assert_body(self, project, req=None):
30         pass
31
32     @staticmethod
33     def _load_request(f_req):
34         abs_file = os.path.join(os.path.dirname(__file__), f_req)
35         with open(abs_file, 'r') as f:
36             loader = json.load(f)
37             f.close()
38         return loader
39
40     def create_return_name(self, req):
41         _, res = self.create(req)
42         return res.href.split('/')[-1]
43
44     def assert_res(self, code, scenario, req=None):
45         self.assertEqual(code, httplib.OK)
46         if req is None:
47             req = self.req_d
48         self.assertIsNotNone(scenario._id)
49         self.assertIsNotNone(scenario.creation_date)
50         self.assertEqual(scenario, models.Scenario.from_dict(req))
51
52     @staticmethod
53     def set_query(*args):
54         uri = ''
55         for arg in args:
56             uri += arg + '&'
57         return uri[0: -1]
58
59     def get_and_assert(self, name):
60         code, body = self.get(name)
61         self.assert_res(code, body, self.req_d)
62
63
64 class TestScenarioCreate(TestScenarioBase):
65     def test_withoutBody(self):
66         (code, body) = self.create()
67         self.assertEqual(code, httplib.BAD_REQUEST)
68
69     def test_emptyName(self):
70         req_empty = models.ScenarioCreateRequest('')
71         (code, body) = self.create(req_empty)
72         self.assertEqual(code, httplib.BAD_REQUEST)
73         self.assertIn(message.missing('name'), body)
74
75     def test_noneName(self):
76         req_none = models.ScenarioCreateRequest(None)
77         (code, body) = self.create(req_none)
78         self.assertEqual(code, httplib.BAD_REQUEST)
79         self.assertIn(message.missing('name'), body)
80
81     def test_success(self):
82         (code, body) = self.create_d()
83         self.assertEqual(code, httplib.OK)
84         self.assert_create_body(body)
85
86     def test_alreadyExist(self):
87         self.create_d()
88         (code, body) = self.create_d()
89         self.assertEqual(code, httplib.FORBIDDEN)
90         self.assertIn(message.exist_base, body)
91
92
93 class TestScenarioGet(TestScenarioBase):
94     def setUp(self):
95         super(TestScenarioGet, self).setUp()
96         self.scenario_1 = self.create_return_name(self.req_d)
97         self.scenario_2 = self.create_return_name(self.req_2)
98
99     def test_getByName(self):
100         self.get_and_assert(self.scenario_1)
101
102     def test_getAll(self):
103         self._query_and_assert(query=None, reqs=[self.req_d, self.req_2])
104
105     def test_queryName(self):
106         query = self.set_query('name=nosdn-nofeature-ha')
107         self._query_and_assert(query, reqs=[self.req_d])
108
109     def test_queryInstaller(self):
110         query = self.set_query('installer=apex')
111         self._query_and_assert(query, reqs=[self.req_d])
112
113     def test_queryVersion(self):
114         query = self.set_query('version=master')
115         self._query_and_assert(query, reqs=[self.req_d])
116
117     def test_queryProject(self):
118         query = self.set_query('project=functest')
119         self._query_and_assert(query, reqs=[self.req_d, self.req_2])
120
121     # close due to random fail, open again after solve it in another patch
122     # def test_queryCombination(self):
123     #     query = self._set_query('name=nosdn-nofeature-ha',
124     #                             'installer=apex',
125     #                             'version=master',
126     #                             'project=functest')
127     #
128     #     self._query_and_assert(query, reqs=[self.req_d])
129
130     def _query_and_assert(self, query, found=True, reqs=None):
131         code, body = self.query(query)
132         if not found:
133             self.assertEqual(code, httplib.OK)
134             self.assertEqual(0, len(body.scenarios))
135         else:
136             self.assertEqual(len(reqs), len(body.scenarios))
137             for req in reqs:
138                 for scenario in body.scenarios:
139                     if req['name'] == scenario.name:
140                         self.assert_res(code, scenario, req)
141
142
143 class TestScenarioDelete(TestScenarioBase):
144     def test_notFound(self):
145         code, body = self.delete('notFound')
146         self.assertEqual(code, httplib.NOT_FOUND)
147
148     def test_success(self):
149         scenario = self.create_return_name(self.req_d)
150         code, _ = self.delete(scenario)
151         self.assertEqual(code, httplib.OK)
152         code, _ = self.get(scenario)
153         self.assertEqual(code, httplib.NOT_FOUND)
154
155
156 class TestScenarioUpdate(TestScenarioBase):
157     def setUp(self):
158         super(TestScenarioUpdate, self).setUp()
159         self.scenario = self.create_return_name(self.req_d)
160         self.scenario_2 = self.create_return_name(self.req_2)
161         self.update_url = ''
162         self.scenario_url = '/api/v1/scenarios/{}'.format(self.scenario)
163         self.installer = self.req_d['installers'][0]['installer']
164         self.version = self.req_d['installers'][0]['versions'][0]['version']
165         self.locate_project = 'installer={}&version={}&project={}'.format(
166             self.installer,
167             self.version,
168             'functest')
169
170     def update_url_fixture(item):
171         def _update_url_fixture(xstep):
172             def wrapper(self, *args, **kwargs):
173                 locator = None
174                 if item in ['projects', 'owner']:
175                     locator = 'installer={}&version={}'.format(
176                         self.installer,
177                         self.version)
178                 elif item in ['versions']:
179                     locator = 'installer={}'.format(
180                         self.installer)
181
182                 self.update_url = '{}/{}?{}'.format(self.scenario_url,
183                                                     item,
184                                                     locator)
185                 xstep(self, *args, **kwargs)
186             return wrapper
187         return _update_url_fixture
188
189     def update_partial(operate, expected):
190         def _update_partial(set_update):
191             @functools.wraps(set_update)
192             def wrapper(self):
193                 update = set_update(self)
194                 code, body = getattr(self, operate)(update)
195                 getattr(self, expected)(code)
196             return wrapper
197         return _update_partial
198
199     @update_partial('_add', '_success')
200     def test_addScore(self):
201         add = models.ScenarioScore(date=str(datetime.now()), score='11/12')
202         projects = self.req_d['installers'][0]['versions'][0]['projects']
203         functest = filter(lambda f: f['project'] == 'functest', projects)[0]
204         functest['scores'].append(add.format())
205         self.update_url = '{}/scores?{}'.format(self.scenario_url,
206                                                 self.locate_project)
207
208         return add
209
210     @update_partial('_add', '_success')
211     def test_addTrustIndicator(self):
212         add = models.ScenarioTI(date=str(datetime.now()), status='gold')
213         projects = self.req_d['installers'][0]['versions'][0]['projects']
214         functest = filter(lambda f: f['project'] == 'functest', projects)[0]
215         functest['trust_indicators'].append(add.format())
216         self.update_url = '{}/trust_indicators?{}'.format(self.scenario_url,
217                                                           self.locate_project)
218
219         return add
220
221     @update_partial('_add', '_success')
222     def test_addCustoms(self):
223         adds = ['odl', 'parser', 'vping_ssh']
224         projects = self.req_d['installers'][0]['versions'][0]['projects']
225         functest = filter(lambda f: f['project'] == 'functest', projects)[0]
226         functest['customs'] = list(set(functest['customs'] + adds))
227         self.update_url = '{}/customs?{}'.format(self.scenario_url,
228                                                  self.locate_project)
229         return adds
230
231     @update_partial('_update', '_success')
232     def test_updateCustoms(self):
233         updates = ['odl', 'parser', 'vping_ssh']
234         projects = self.req_d['installers'][0]['versions'][0]['projects']
235         functest = filter(lambda f: f['project'] == 'functest', projects)[0]
236         functest['customs'] = updates
237         self.update_url = '{}/customs?{}'.format(self.scenario_url,
238                                                  self.locate_project)
239
240         return updates
241
242     @update_partial('_delete', '_success')
243     def test_deleteCustoms(self):
244         deletes = ['vping_ssh']
245         projects = self.req_d['installers'][0]['versions'][0]['projects']
246         functest = filter(lambda f: f['project'] == 'functest', projects)[0]
247         functest['customs'] = ['healthcheck']
248         self.update_url = '{}/customs?{}'.format(self.scenario_url,
249                                                  self.locate_project)
250
251         return deletes
252
253     @update_url_fixture('projects')
254     @update_partial('_add', '_success')
255     def test_addProjects_succ(self):
256         add = models.ScenarioProject(project='qtip').format()
257         self.req_d['installers'][0]['versions'][0]['projects'].append(add)
258         return [add]
259
260     @update_url_fixture('projects')
261     @update_partial('_add', '_conflict')
262     def test_addProjects_already_exist(self):
263         add = models.ScenarioProject(project='functest').format()
264         return [add]
265
266     @update_url_fixture('projects')
267     @update_partial('_add', '_bad_request')
268     def test_addProjects_bad_schema(self):
269         add = models.ScenarioProject(project='functest').format()
270         add['score'] = None
271         return [add]
272
273     @update_url_fixture('projects')
274     @update_partial('_update', '_success')
275     def test_updateProjects_succ(self):
276         update = models.ScenarioProject(project='qtip').format()
277         self.req_d['installers'][0]['versions'][0]['projects'] = [update]
278         return [update]
279
280     @update_url_fixture('projects')
281     @update_partial('_update', '_conflict')
282     def test_updateProjects_duplicated(self):
283         update1 = models.ScenarioProject(project='qtip').format()
284         update2 = models.ScenarioProject(project='qtip').format()
285         return [update1, update2]
286
287     @update_url_fixture('projects')
288     @update_partial('_update', '_bad_request')
289     def test_updateProjects_bad_schema(self):
290         update = models.ScenarioProject(project='functest').format()
291         update['score'] = None
292         return [update]
293
294     @update_url_fixture('projects')
295     @update_partial('_delete', '_success')
296     def test_deleteProjects(self):
297         deletes = ['functest']
298         projects = self.req_d['installers'][0]['versions'][0]['projects']
299         self.req_d['installers'][0]['versions'][0]['projects'] = filter(
300             lambda f: f['project'] != 'functest',
301             projects)
302         return deletes
303
304     @update_url_fixture('owner')
305     @update_partial('_update', '_success')
306     def test_changeOwner(self):
307         new_owner = 'new_owner'
308         update = models.ScenarioChangeOwnerRequest(new_owner).format()
309         self.req_d['installers'][0]['versions'][0]['owner'] = new_owner
310         return update
311
312     @update_url_fixture('versions')
313     @update_partial('_add', '_success')
314     def test_addVersions_succ(self):
315         add = models.ScenarioVersion(version='Euphrates').format()
316         self.req_d['installers'][0]['versions'].append(add)
317         return [add]
318
319     @update_url_fixture('versions')
320     @update_partial('_add', '_conflict')
321     def test_addVersions_already_exist(self):
322         add = models.ScenarioVersion(version='master').format()
323         return [add]
324
325     @update_url_fixture('versions')
326     @update_partial('_add', '_bad_request')
327     def test_addVersions_bad_schema(self):
328         add = models.ScenarioVersion(version='euphrates').format()
329         add['notexist'] = None
330         return [add]
331
332     @update_url_fixture('versions')
333     @update_partial('_update', '_success')
334     def test_updateVersions_succ(self):
335         update = models.ScenarioVersion(version='euphrates').format()
336         self.req_d['installers'][0]['versions'] = [update]
337         return [update]
338
339     @update_url_fixture('versions')
340     @update_partial('_update', '_conflict')
341     def test_updateVersions_duplicated(self):
342         update1 = models.ScenarioVersion(version='euphrates').format()
343         update2 = models.ScenarioVersion(version='euphrates').format()
344         return [update1, update2]
345
346     @update_url_fixture('versions')
347     @update_partial('_update', '_bad_request')
348     def test_updateVersions_bad_schema(self):
349         update = models.ScenarioVersion(version='euphrates').format()
350         update['not_owner'] = 'Iam'
351         return [update]
352
353     @update_url_fixture('versions')
354     @update_partial('_delete', '_success')
355     def test_deleteVersions(self):
356         deletes = ['master']
357         versions = self.req_d['installers'][0]['versions']
358         self.req_d['installers'][0]['versions'] = filter(
359             lambda f: f['version'] != 'master',
360             versions)
361         return deletes
362
363     def _add(self, update_req):
364         return self.post_direct_url(self.update_url, update_req)
365
366     def _update(self, update_req):
367         return self.update_direct_url(self.update_url, update_req)
368
369     def _delete(self, update_req):
370         return self.delete_direct_url(self.update_url, update_req)
371
372     def _success(self, status):
373         self.assertEqual(status, httplib.OK)
374         self.get_and_assert(self.req_d['name'])
375
376     def _forbidden(self, status):
377         self.assertEqual(status, httplib.FORBIDDEN)
378
379     def _bad_request(self, status):
380         self.assertEqual(status, httplib.BAD_REQUEST)
381
382     def _conflict(self, status):
383         self.assertEqual(status, httplib.CONFLICT)