1 files changed, 43 insertions, 16 deletions

diff --git a/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py b/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py
index ccacfeb5..cf1e342b 100644
--- a/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py
+++ b/openshift-storage-libs/openshiftstoragelibs/openshift_ops.py
@@ -345,25 +345,12 @@ def oc_create_pvc(hostname, sc_name=None, pvc_name_prefix="autotests-pvc",
     return pvc_name
 
 
-def oc_create_app_dc_with_io(
-        hostname, pvc_name, dc_name_prefix="autotests-dc-with-app-io",
-        replicas=1, space_to_use=1048576, label=None):
-    """Create DC with app PODs and attached PVC, constantly running I/O.
-
-    Args:
-        hostname (str): Node on which 'oc create' command will be executed.
-        pvc_name (str): name of the Persistent Volume Claim to attach to
-                        the application PODs where constant I/O will run.
-        dc_name_prefix (str): DC name will consist of this prefix and
-                              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.
-    """
+def _oc_create_app_dc_with_io_image(hostname, pvc_name, dc_name_prefix,
+                                    replicas, space_to_use, label, image):
     dc_name = "%s-%s" % (dc_name_prefix, utils.get_random_str())
     container_data = {
         "name": dc_name,
-        "image": "cirros",
+        "image": image,
         "volumeMounts": [{"mountPath": "/mnt", "name": dc_name}],
         "command": ["sh"],
         "args": [
@@ -419,6 +406,46 @@ def oc_create_app_dc_with_io(
     return dc_name
 
 
+def oc_create_app_dc_with_io(
+        hostname, pvc_name, dc_name_prefix="autotests-dc-with-app-io",
+        replicas=1, space_to_use=1048576, label=None):
+    """Create DC with app PODs and attached PVC, constantly running I/O.
+
+    Args:
+        hostname (str): Node on which 'oc create' command will be executed.
+        pvc_name (str): name of the Persistent Volume Claim to attach to
+                        the application PODs where constant I/O will run.
+        dc_name_prefix (str): DC name will consist of this prefix and
+                              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.
+    """
+    return _oc_create_app_dc_with_io_image(
+        hostname, pvc_name, dc_name_prefix, replicas, space_to_use,
+        label, "cirros")
+
+
+def oc_create_busybox_app_dc_with_io(
+        hostname, pvc_name, dc_name_prefix="autotests-dc-with-app-io",
+        replicas=1, space_to_use=1048576, label=None):
+    """Create DC with app PODs and attached PVC, constantly running I/O.
+
+    Args:
+        hostname (str): Node on which 'oc create' command will be executed.
+        pvc_name (str): name of the Persistent Volume Claim to attach to
+                        the application PODs where constant I/O will run.
+        dc_name_prefix (str): DC name will consist of this prefix and
+                              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.
+    """
+    return _oc_create_app_dc_with_io_image(
+        hostname, pvc_name, dc_name_prefix, replicas, space_to_use,
+        label, "busybox")
+
+
 def oc_create_tiny_pod_with_volume(hostname, pvc_name, pod_name_prefix='',
                                    mount_path='/mnt'):
     """Create tiny POD from image in 10Mb with attached volume at /mnt"""