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
75
76
77
78
79
80
81
82
83
84
|
# -*- coding: utf-8 -*-
# Copyright 2017 Objectif Libre
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.
#
from tempest import config
from tempest.lib import exceptions
import tempest.test
from cloudkitty_tempest_plugin.services import client
CONF = config.CONF
def skipIf(flag, reason):
def decorator(f):
def wrapper(self, *args, **kwargs):
if getattr(self, flag):
self.skipTest(reason)
else:
f(self, *args, **kwargs)
return wrapper
return decorator
class BaseRatingTest(tempest.test.BaseTestCase):
"""Base test class for all Rating API tests."""
client_manager = client.Manager
@classmethod
def setup_clients(cls):
super(BaseRatingTest, cls).setup_clients()
os_var = 'os_{}'.format(cls.credentials[0])
cls.rating_client = getattr(cls, os_var).get_rating_client(
getattr(cls, 'api_version'))
@classmethod
def setup_credentials(cls):
super(BaseRatingTest, cls).setup_credentials()
os_var = 'os_{}'.format(cls.credentials[0])
project_id = getattr(cls, os_var).credentials.project_id
cls.skip_rating_tests = False
try:
cls.custom_identity_client = client.CustomIdentityClient()
cls.custom_identity_client.enable_rating(project_id)
except Exception:
cls.skip_rating_tests = True
@classmethod
def resource_setup(cls):
super(BaseRatingTest, cls).resource_setup()
cls._created_resources = {
'collector_mapping': list(),
'hashmap_service': list(),
'hashmap_field': list(),
'hashmap_mapping': list(),
'hashmap_group': list(),
'hashmap_threshold': list(),
'pyscript': list(),
}
@classmethod
def resource_cleanup(cls):
super(BaseRatingTest, cls).resource_cleanup()
for method, item_ids in cls._created_resources.items():
delete_method = 'delete_' + method
delete_method = getattr(cls.rating_client, delete_method)
for item_id in item_ids:
try:
delete_method(item_id)
except (exceptions.NotFound,
exceptions.UnexpectedResponseCode):
pass
|
