From 237e89c33fcea281b4190c11edc25896a578ab31 Mon Sep 17 00:00:00 2001
From: Nitin Goyal <nigoyal@redhat.com>
Date: Thu, 13 Feb 2020 15:30:51 +0530
Subject: [Lib] Add arg label for labeling the DC

Change-Id: I28af35fdcc27e30957e02d83e19f0a7d4211f18b
---
 openshift-storage-libs/openshiftstoragelibs/baseclass.py    | 13 +++++++++----
 .../openshiftstoragelibs/openshift_ops.py                   | 11 +++++++++--
 2 files changed, 18 insertions(+), 6 deletions(-)

(limited to 'openshift-storage-libs')

diff --git a/openshift-storage-libs/openshiftstoragelibs/baseclass.py b/openshift-storage-libs/openshiftstoragelibs/baseclass.py
index fd2c3937..b173c3b5 100644
--- a/openshift-storage-libs/openshiftstoragelibs/baseclass.py
+++ b/openshift-storage-libs/openshiftstoragelibs/baseclass.py
@@ -388,7 +388,8 @@ class BaseClass(unittest.TestCase):
 
     def create_dcs_with_pvc(
             self, pvc_names, timeout=600, wait_step=5,
-            dc_name_prefix='autotests-dc', skip_cleanup=False):
+            dc_name_prefix='autotests-dc', label=None,
+            skip_cleanup=False):
         """Create bunch of DCs with app PODs which use unique PVCs.
 
         Args:
@@ -397,6 +398,7 @@ class BaseClass(unittest.TestCase):
             timeout (int): timeout value, default value is 600 seconds.
             wait_step( int): wait step, default value is 5 seconds.
             dc_name_prefix(str): name prefix for deployement config.
+            lable (dict): keys and value for adding label into DC.
         Returns: dictionary with following structure:
             {
                 "pvc_name_1": ("dc_name_1", "pod_name_1"),
@@ -411,7 +413,8 @@ class BaseClass(unittest.TestCase):
         dc_and_pod_names, dc_names = {}, {}
         for pvc_name in pvc_names:
             dc_name = oc_create_app_dc_with_io(
-                self.ocp_client[0], pvc_name, dc_name_prefix=dc_name_prefix)
+                self.ocp_client[0], pvc_name, dc_name_prefix=dc_name_prefix,
+                label=label)
             dc_names[pvc_name] = dc_name
             if not skip_cleanup:
                 self.addCleanup(oc_delete, self.ocp_client[0], 'dc', dc_name)
@@ -431,10 +434,12 @@ class BaseClass(unittest.TestCase):
 
     def create_dc_with_pvc(
             self, pvc_name, timeout=300, wait_step=10,
-            dc_name_prefix='autotests-dc', skip_cleanup=False):
+            dc_name_prefix='autotests-dc', label=None,
+            skip_cleanup=False):
         return self.create_dcs_with_pvc(
             pvc_name, timeout, wait_step,
-            dc_name_prefix=dc_name_prefix, skip_cleanup=skip_cleanup)[pvc_name]
+            dc_name_prefix=dc_name_prefix, label=label,
+            skip_cleanup=skip_cleanup)[pvc_name]
 
     def create_heketi_volume_with_name_and_wait(
             self, name, size, raise_on_cleanup_error=True,
diff --git a/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py b/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py
index 07a527da..aaf384d1 100644
--- a/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py
+++ b/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py
@@ -347,7 +347,7 @@ def oc_create_pvc(hostname, sc_name=None, pvc_name_prefix="autotests-pvc",
 
 def oc_create_app_dc_with_io(
         hostname, pvc_name, dc_name_prefix="autotests-dc-with-app-io",
-        replicas=1, space_to_use=1048576):
+        replicas=1, space_to_use=1048576, label=None):
     """Create DC with app PODs and attached PVC, constantly running I/O.
 
     Args:
@@ -358,6 +358,7 @@ def oc_create_app_dc_with_io(
                               random str.
         replicas (int): amount of application POD replicas.
         space_to_use (int): value in bytes which will be used for I/O.
+        label (dict): dict of keys and values to add labels in DC.
     """
     dc_name = "%s-%s" % (dc_name_prefix, utils.get_random_str())
     container_data = {
@@ -385,6 +386,12 @@ def oc_create_app_dc_with_io(
             ]},
         },
     }
+
+    labels = {"name": dc_name}
+    if label:
+        label.pop("name", None)
+        labels.update(label)
+
     dc_data = json.dumps({
         "kind": "DeploymentConfig",
         "apiVersion": "v1",
@@ -395,7 +402,7 @@ def oc_create_app_dc_with_io(
             "paused": False,
             "revisionHistoryLimit": 2,
             "template": {
-                "metadata": {"labels": {"name": dc_name}},
+                "metadata": {"labels": labels},
                 "spec": {
                     "restartPolicy": "Always",
                     "volumes": [{
-- 
cgit