diff options
Diffstat (limited to 'cns-libs/cnslibs/cns')
| -rw-r--r-- | cns-libs/cnslibs/cns/__init__.py | 0 | ||||
| -rw-r--r-- | cns-libs/cnslibs/cns/cns_baseclass.py | 251 |
2 files changed, 0 insertions, 251 deletions
diff --git a/cns-libs/cnslibs/cns/__init__.py b/cns-libs/cnslibs/cns/__init__.py deleted file mode 100644 index e69de29b..00000000 --- a/cns-libs/cnslibs/cns/__init__.py +++ /dev/null diff --git a/cns-libs/cnslibs/cns/cns_baseclass.py b/cns-libs/cnslibs/cns/cns_baseclass.py deleted file mode 100644 index 9e7912b2..00000000 --- a/cns-libs/cnslibs/cns/cns_baseclass.py +++ /dev/null @@ -1,251 +0,0 @@ -import datetime - -from glusto.core import Glusto as g - -from cnslibs.common import baseclass -from cnslibs.common import command -from cnslibs.common.exceptions import ExecutionError -from cnslibs.common.heketi_ops import ( - heketi_blockvolume_delete, - heketi_volume_delete) -from cnslibs.common.openshift_ops import ( - get_pod_name_from_dc, - get_pv_name_from_pvc, - oc_create_app_dc_with_io, - oc_create_pvc, - oc_create_sc, - oc_create_secret, - oc_delete, - oc_get_custom_resource, - scale_dc_pod_amount_and_wait, - verify_pvc_status_is_bound, - wait_for_pod_be_ready, - wait_for_resource_absence, -) - - -class BaseClass(baseclass.BaseClass): - """Base class for test classes.""" - - @classmethod - def setUpClass(cls): - """Initialize all the variables necessary for test cases.""" - super(BaseClass, cls).setUpClass() - - # Initializes OCP config variables - cls.ocp_servers_info = g.config['ocp_servers'] - cls.ocp_master_node = g.config['ocp_servers']['master'].keys() - cls.ocp_master_node_info = g.config['ocp_servers']['master'] - cls.ocp_client = g.config['ocp_servers']['client'].keys() - cls.ocp_client_info = g.config['ocp_servers']['client'] - cls.ocp_nodes = g.config['ocp_servers']['nodes'].keys() - cls.ocp_nodes_info = g.config['ocp_servers']['nodes'] - - # Initializes storage project 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')) - - # Initializes heketi config variables - 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'] - - cls.storage_classes = openshift_config['dynamic_provisioning'][ - 'storage_classes'] - cls.sc = cls.storage_classes.get( - 'storage_class1', cls.storage_classes.get('file_storage_class')) - cmd = "echo -n %s | base64" % cls.heketi_cli_key - ret, out, err = g.run(cls.ocp_master_node[0], cmd, "root") - if ret != 0: - raise ExecutionError("failed to execute cmd %s on %s out: %s " - "err: %s" % ( - cmd, cls.ocp_master_node[0], out, err)) - cls.secret_data_key = out.strip() - - cmd = 'oc project %s' % cls.storage_project_name - ret, out, err = g.run(cls.ocp_client[0], cmd, "root") - if ret != 0: - raise ExecutionError("failed to execute cmd %s on %s out: " - "%s err: %s" % ( - cmd, cls.ocp_client[0], out, err)) - - 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(BaseClass, self).setUp() - msg = "Starting Test : %s : %s" % (self.id(), self.glustotest_run_id) - g.log.info(msg) - - def tearDown(self): - super(BaseClass, self).tearDown() - msg = "Ending Test: %s : %s" % (self.id(), self.glustotest_run_id) - g.log.info(msg) - - @classmethod - def tearDownClass(cls): - super(BaseClass, cls).tearDownClass() - msg = "Teardownclass: %s : %s" % (cls.__name__, cls.glustotest_run_id) - g.log.info(msg) - - def cmd_run(self, cmd, hostname=None, raise_on_error=True): - if not hostname: - hostname = self.ocp_master_node[0] - return command.cmd_run( - cmd=cmd, hostname=hostname, raise_on_error=raise_on_error) - - def create_secret(self, secret_name_prefix="autotests-secret-"): - secret_name = oc_create_secret( - self.ocp_client[0], - secret_name_prefix=secret_name_prefix, - namespace=(self.sc.get( - 'secretnamespace', - self.sc.get('restsecretnamespace', 'default'))), - data_key=self.heketi_cli_key, - secret_type=self.sc.get('provisioner', 'kubernetes.io/glusterfs')) - self.addCleanup( - oc_delete, self.ocp_client[0], 'secret', secret_name) - return secret_name - - def create_storage_class(self, secret_name=None, - sc_name_prefix="autotests-sc", - create_vol_name_prefix=False, - allow_volume_expansion=False, - reclaim_policy="Delete", - set_hacount=None, - is_arbiter_vol=False, arbiter_avg_file_size=None): - - # Create secret if one is not specified - if not secret_name: - secret_name = self.create_secret() - - # Create storage class - secret_name_option = "secretname" - secret_namespace_option = "secretnamespace" - provisioner = self.sc.get("provisioner", "kubernetes.io/glusterfs") - if provisioner != "kubernetes.io/glusterfs": - secret_name_option = "rest%s" % secret_name_option - secret_namespace_option = "rest%s" % secret_namespace_option - parameters = { - "resturl": self.sc["resturl"], - "restuser": self.sc["restuser"], - secret_name_option: secret_name, - secret_namespace_option: self.sc.get( - "secretnamespace", self.sc.get("restsecretnamespace")), - } - if set_hacount: - parameters["hacount"] = self.sc.get("hacount", "3") - if is_arbiter_vol: - parameters["volumeoptions"] = "user.heketi.arbiter true" - if arbiter_avg_file_size: - parameters["volumeoptions"] += ( - ",user.heketi.average-file-size %s" % ( - arbiter_avg_file_size)) - if create_vol_name_prefix: - parameters["volumenameprefix"] = self.sc.get( - "volumenameprefix", "autotest") - self.sc_name = oc_create_sc( - self.ocp_client[0], - sc_name_prefix=sc_name_prefix, - provisioner=provisioner, - allow_volume_expansion=allow_volume_expansion, - reclaim_policy=reclaim_policy, - **parameters) - self.addCleanup(oc_delete, self.ocp_client[0], "sc", self.sc_name) - return self.sc_name - - def create_and_wait_for_pvcs(self, pvc_size=1, - pvc_name_prefix="autotests-pvc", - pvc_amount=1, sc_name=None, - timeout=120, wait_step=3): - node = self.ocp_client[0] - - # Create storage class if not specified - if not sc_name: - if getattr(self, "sc_name", ""): - sc_name = self.sc_name - else: - sc_name = self.create_storage_class() - - # Create PVCs - pvc_names = [] - for i in range(pvc_amount): - pvc_name = oc_create_pvc( - node, sc_name, pvc_name_prefix=pvc_name_prefix, - pvc_size=pvc_size) - pvc_names.append(pvc_name) - self.addCleanup( - wait_for_resource_absence, node, 'pvc', pvc_name) - - # Wait for PVCs to be in bound state - try: - for pvc_name in pvc_names: - verify_pvc_status_is_bound(node, pvc_name, timeout, wait_step) - finally: - reclaim_policy = oc_get_custom_resource( - node, 'sc', ':.reclaimPolicy', sc_name)[0] - - for pvc_name in pvc_names: - if reclaim_policy == 'Retain': - pv_name = get_pv_name_from_pvc(node, pvc_name) - self.addCleanup(oc_delete, node, 'pv', pv_name, - raise_on_absence=False) - custom = (r':.metadata.annotations."gluster\.kubernetes' - r'\.io\/heketi\-volume\-id"') - vol_id = oc_get_custom_resource( - node, 'pv', custom, pv_name)[0] - if self.sc.get('provisioner') == "kubernetes.io/glusterfs": - self.addCleanup(heketi_volume_delete, - self.heketi_client_node, - self.heketi_server_url, vol_id, - raise_on_error=False) - else: - self.addCleanup(heketi_blockvolume_delete, - self.heketi_client_node, - self.heketi_server_url, vol_id, - raise_on_error=False) - self.addCleanup(oc_delete, node, 'pvc', pvc_name, - raise_on_absence=False) - - return pvc_names - - def create_and_wait_for_pvc(self, pvc_size=1, - pvc_name_prefix='autotests-pvc', sc_name=None): - self.pvc_name = self.create_and_wait_for_pvcs( - pvc_size=pvc_size, pvc_name_prefix=pvc_name_prefix, sc_name=sc_name - )[0] - return self.pvc_name - - def create_dc_with_pvc(self, pvc_name, timeout=300, wait_step=10): - dc_name = oc_create_app_dc_with_io(self.ocp_client[0], pvc_name) - self.addCleanup(oc_delete, self.ocp_client[0], 'dc', dc_name) - self.addCleanup( - scale_dc_pod_amount_and_wait, self.ocp_client[0], dc_name, 0) - pod_name = get_pod_name_from_dc(self.ocp_client[0], dc_name) - wait_for_pod_be_ready(self.ocp_client[0], pod_name, - timeout=timeout, wait_step=wait_step) - return dc_name, pod_name - - -class GlusterBlockBaseClass(BaseClass): - """Base class for gluster-block test cases.""" - - @classmethod - def setUpClass(cls): - """Initialize all the variables necessary for test cases.""" - super(GlusterBlockBaseClass, cls).setUpClass() - cls.sc = cls.storage_classes.get( - 'storage_class2', cls.storage_classes.get('block_storage_class')) |