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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
import datetime
import unittest
from glusto.core import Glusto as g
from cnslibs.common.exceptions import ExecutionError, ConfigError
from cnslibs.common.heketi_ops import (hello_heketi,
heketi_volume_delete,
heketi_blockvolume_delete)
from cnslibs.common import openshift_ops
class HeketiBaseClass(unittest.TestCase):
"""
This class initializes heketi config variables, constructs topology info
dictionary and check if heketi server is alive.
"""
@classmethod
def setUpClass(cls):
"""
setUpClass of HeketiBaseClass
"""
super(HeketiBaseClass, cls).setUpClass()
# Initializes config variables
openshift_config = g.config.get("cns", g.config.get("openshift"))
cls.storage_project_name = openshift_config.get(
'storage_project_name',
openshift_config.get('setup', {}).get('cns_project_name'))
cls.ocp_master_nodes = g.config['ocp_servers']['master'].keys()
cls.ocp_master_node = cls.ocp_master_nodes[0]
heketi_config = openshift_config['heketi_config']
cls.heketi_dc_name = heketi_config['heketi_dc_name']
cls.heketi_service_name = heketi_config['heketi_service_name']
cls.heketi_client_node = heketi_config['heketi_client_node']
cls.heketi_server_url = heketi_config['heketi_server_url']
cls.heketi_cli_user = heketi_config['heketi_cli_user']
cls.heketi_cli_key = heketi_config['heketi_cli_key']
cls.gluster_servers = g.config['gluster_servers'].keys()
cls.gluster_servers_info = g.config['gluster_servers']
# Checks if heketi server is alive
if not hello_heketi(cls.heketi_client_node, cls.heketi_server_url):
raise ConfigError("Heketi server %s is not alive"
% cls.heketi_server_url)
# Switch to the storage project
if not openshift_ops.switch_oc_project(
cls.ocp_master_node, cls.storage_project_name):
raise ExecutionError("Failed to switch oc project on node %s"
% cls.ocp_master_node)
# Have a unique string to recognize the test run for logging
if 'glustotest_run_id' not in g.config:
g.config['glustotest_run_id'] = (
datetime.datetime.now().strftime('%H_%M_%d_%m_%Y'))
cls.glustotest_run_id = g.config['glustotest_run_id']
msg = "Setupclass: %s : %s" % (cls.__name__, cls.glustotest_run_id)
g.log.info(msg)
def setUp(self):
super(HeketiBaseClass, self).setUp()
msg = "Starting Test : %s : %s" % (self.id(), self.glustotest_run_id)
g.log.info(msg)
def delete_volumes(self, volume_ids):
"""
Delete volumes by their IDs and raise error with list of failures
Input: (volume_ids) It can be a single volume ID
or a list of volume IDs
"""
errored_ids = []
if not isinstance(volume_ids, (list, set, tuple)):
volume_ids = [volume_ids]
for volume_id in volume_ids:
out = heketi_volume_delete(
self.heketi_client_node, self.heketi_server_url, volume_id)
output_str = 'Volume %s deleted' % volume_id
if output_str not in out:
errored_ids.append(volume_id)
if errored_ids:
raise ExecutionError(
"Failed to delete following heketi volumes: "
"%s" % ',\n'.join(errored_ids))
def delete_block_volumes(self, volume_ids):
"""
Delete block volumes by their volume IDs and raise an error on failures
Args:
volume_ids (str) : Volume ID of the block volume
"""
if not isinstance(volume_ids, (list, set, tuple)):
volume_ids = [volume_ids]
fail = False
for volume_id in volume_ids:
block_out = heketi_blockvolume_delete(
self.heketi_client_node, self.heketi_server_url, volume_id)
if block_out is False:
g.log.error("Block volume delete failed %s " % volume_id)
fail = True
self.assertFalse(fail, "Failed to delete blockvolumes")
def tearDown(self):
super(HeketiBaseClass, self).tearDown()
msg = "Ending Test: %s : %s" % (self.id(), self.glustotest_run_id)
g.log.info(msg)
@classmethod
def tearDownClass(cls):
super(HeketiBaseClass, cls).tearDownClass()
msg = "Teardownclass: %s : %s" % (cls.__name__, cls.glustotest_run_id)
g.log.info(msg)
|
