diff options
Diffstat (limited to 'cns-libs/cnslibs/common')
-rw-r--r-- | cns-libs/cnslibs/common/heketi_libs.py | 34 | ||||
-rw-r--r-- | cns-libs/cnslibs/common/heketi_ops.py | 12 | ||||
-rw-r--r-- | cns-libs/cnslibs/common/openshift_ops.py | 24 |
3 files changed, 16 insertions, 54 deletions
diff --git a/cns-libs/cnslibs/common/heketi_libs.py b/cns-libs/cnslibs/common/heketi_libs.py index 7ac5c6a6..745ab229 100644 --- a/cns-libs/cnslibs/common/heketi_libs.py +++ b/cns-libs/cnslibs/common/heketi_libs.py @@ -1,4 +1,3 @@ -from collections import OrderedDict import datetime import unittest @@ -25,14 +24,16 @@ class HeketiBaseClass(unittest.TestCase): super(HeketiBaseClass, cls).setUpClass() - # Initializes heketi config variables - cls.cns_username = g.config['cns']['setup']['cns_username'] - cls.cns_password = g.config['cns']['setup']['cns_password'] - cls.cns_project_name = g.config['cns']['setup']['cns_project_name'] + # 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 = g.config['cns']['heketi_config'] + 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'] @@ -41,21 +42,6 @@ class HeketiBaseClass(unittest.TestCase): 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'] - cls.topo_info = g.config['cns']['trusted_storage_pool_list'] - - # Constructs topology info dictionary - cls.topology_info = OrderedDict() - for i in range(len(cls.topo_info)): - cluster = 'cluster' + str(i + 1) - cls.topology_info[cluster] = OrderedDict() - for index, node in enumerate(cls.topo_info[i]): - node_name = 'gluster_node' + str(index + 1) - cls.topology_info[cluster][node_name] = { - 'manage': cls.gluster_servers_info[node]['manage'], - 'storage': cls.gluster_servers_info[node]['storage'], - 'zone': cls.gluster_servers_info[node]['zone'], - 'devices': cls.gluster_servers_info[node]['devices'], - } # Checks if heketi server is alive if not hello_heketi(cls.heketi_client_node, cls.heketi_server_url): @@ -63,12 +49,8 @@ class HeketiBaseClass(unittest.TestCase): % cls.heketi_server_url) # Switch to the storage project - if not openshift_ops.oc_login( - cls.ocp_master_node, cls.cns_username, cls.cns_password): - raise ExecutionError("Failed to do oc login on node %s" - % cls.ocp_master_node) if not openshift_ops.switch_oc_project( - cls.ocp_master_node, cls.cns_project_name): + cls.ocp_master_node, cls.storage_project_name): raise ExecutionError("Failed to switch oc project on node %s" % cls.ocp_master_node) diff --git a/cns-libs/cnslibs/common/heketi_ops.py b/cns-libs/cnslibs/common/heketi_ops.py index 19729485..c13c8582 100644 --- a/cns-libs/cnslibs/common/heketi_ops.py +++ b/cns-libs/cnslibs/common/heketi_ops.py @@ -18,10 +18,12 @@ def _set_heketi_global_flags(heketi_server_url, **kwargs): secret_arg = "--secret %s" % secret if secret else "" user_arg = "--user %s" % user if user else "" if not user_arg: - heketi_cli_user = g.config['cns']['heketi_config']['heketi_cli_user'] + openshift_config = g.config.get("cns", g.config.get("openshift")) + heketi_cli_user = openshift_config['heketi_config']['heketi_cli_user'] if heketi_cli_user: user_arg = "--user %s" % heketi_cli_user - heketi_cli_key = g.config['cns']['heketi_config']['heketi_cli_key'] + heketi_cli_key = openshift_config[ + 'heketi_config']['heketi_cli_key'] if heketi_cli_key is not None: secret_arg = "--secret '%s'" % heketi_cli_key @@ -69,10 +71,12 @@ def heketi_volume_create(heketi_client_node, heketi_server_url, size, """ if not kwargs.get('user'): - heketi_cli_user = g.config['cns']['heketi_config']['heketi_cli_user'] + openshift_config = g.config.get("cns", g.config.get("openshift")) + heketi_cli_user = openshift_config['heketi_config']['heketi_cli_user'] if heketi_cli_user: kwargs['user'] = heketi_cli_user - heketi_cli_key = g.config['cns']['heketi_config']['heketi_cli_key'] + heketi_cli_key = openshift_config[ + 'heketi_config']['heketi_cli_key'] if heketi_cli_key is not None: kwargs['secret'] = heketi_cli_key diff --git a/cns-libs/cnslibs/common/openshift_ops.py b/cns-libs/cnslibs/common/openshift_ops.py index d9511980..172a3d6a 100644 --- a/cns-libs/cnslibs/common/openshift_ops.py +++ b/cns-libs/cnslibs/common/openshift_ops.py @@ -133,30 +133,6 @@ def get_amount_of_gluster_nodes(ocp_node): "Haven't found neither Gluster PODs nor Gluster nodes.") -def oc_login(ocp_node, username, password): - """Login to ocp master node. - - Args: - ocp_node (str): Node in which ocp command will be executed. - username (str): username of ocp master node to login. - password (str): password of ocp master node to login. - - Returns: - bool : True on successful login to ocp master node. - False otherwise - - Example: - oc_login(ocp_node, "test","test") - """ - - cmd = "oc login --username=%s --password=%s" % (username, password) - ret, _, _ = g.run(ocp_node, cmd) - if ret != 0: - g.log.error("Failed to login to ocp master node %s" % ocp_node) - return False - return True - - def switch_oc_project(ocp_node, project_name): """Switch to the given project. |