[CNS-433][CNS-436] Fix for failed test cases

Change-Id: I2379043b771cc654eb227d457223b3a24434fd21
Signed-off-by: vamahaja <vamahaja@redhat.com>

2 files changed, 21 insertions, 25 deletions

diff --git a/tests/functional/common/provisioning/test_dynamic_provisioning_block_p0_cases.py b/tests/functional/common/provisioning/test_dynamic_provisioning_block_p0_cases.py
index 11edf4ba..a2d83311 100644
--- a/tests/functional/common/provisioning/test_dynamic_provisioning_block_p0_cases.py
+++ b/tests/functional/common/provisioning/test_dynamic_provisioning_block_p0_cases.py
@@ -1,4 +1,3 @@
-import time
 from unittest import skip
 
 from cnslibs.cns.cns_baseclass import GlusterBlockBaseClass
@@ -7,8 +6,8 @@ from cnslibs.common.openshift_ops import (
     get_gluster_pod_names_by_pvc_name,
     get_pod_name_from_dc,
     get_pv_name_from_pvc,
-    get_pvc_status,
     oc_create_app_dc_with_io,
+    oc_create_pvc,
     oc_delete,
     oc_get_custom_resource,
     oc_rsh,
@@ -94,16 +93,14 @@ class TestDynamicProvisioningBlockP0(GlusterBlockBaseClass):
         wait_for_resource_absence(self.node, 'pod', heketi_pod_name)
 
         # Create second PVC
-        app_2_pvc_name = self.create_and_wait_for_pvc(
-            pvc_name_prefix='autotests-block2')
-
-        # Check status of the second PVC after small pause
-        time.sleep(2)
-        ret, status = get_pvc_status(self.node, app_2_pvc_name)
-        self.assertTrue(ret, "Failed to get pvc status of %s" % app_2_pvc_name)
-        self.assertEqual(
-            status, "Pending",
-            "PVC status of %s is not in Pending state" % app_2_pvc_name)
+        app_2_pvc_name = oc_create_pvc(
+            self.node, pvc_name_prefix='autotest-block2', sc_name=sc_name
+        )
+        self.addCleanup(
+            wait_for_resource_absence, self.node, 'pvc', app_2_pvc_name)
+        self.addCleanup(
+            oc_delete, self.node, 'pvc', app_2_pvc_name
+        )
 
         # Create second app POD
         app_2_dc_name = oc_create_app_dc_with_io(self.node, app_2_pvc_name)
diff --git a/tests/functional/common/provisioning/test_dynamic_provisioning_p0_cases.py b/tests/functional/common/provisioning/test_dynamic_provisioning_p0_cases.py
index 07be5e53..d550e51c 100644
--- a/tests/functional/common/provisioning/test_dynamic_provisioning_p0_cases.py
+++ b/tests/functional/common/provisioning/test_dynamic_provisioning_p0_cases.py
@@ -8,7 +8,6 @@ from cnslibs.common.heketi_ops import (
 from cnslibs.common.openshift_ops import (
     get_gluster_pod_names_by_pvc_name,
     get_pv_name_from_pvc,
-    get_pvc_status,
     get_pod_name_from_dc,
     get_pod_names_from_dc,
     oc_create_secret,
@@ -89,10 +88,12 @@ class TestDynamicProvisioningP0(BaseClass):
         datafile_path = '%s/fake_file_for_%s' % (mount_path, self.id())
 
         # Create secret and storage class
-        self.create_storage_class()
+        sc_name = self.create_storage_class()
 
         # Create PVC
-        app_1_pvc_name = self.create_and_wait_for_pvc()
+        app_1_pvc_name = self.create_and_wait_for_pvc(
+            pvc_name_prefix="autotest-file", sc_name=sc_name
+        )
 
         # Create app POD with attached volume
         app_1_pod_name = oc_create_tiny_pod_with_volume(
@@ -125,16 +126,14 @@ class TestDynamicProvisioningP0(BaseClass):
         self.cmd_run(heketi_down_cmd)
         wait_for_resource_absence(self.node, 'pod', heketi_pod_name)
 
-        # Create second PVC
-        app_2_pvc_name = self.create_and_wait_for_pvc()
-
-        # Check status of the second PVC after small pause
-        time.sleep(2)
-        ret, status = get_pvc_status(self.node, app_2_pvc_name)
-        self.assertTrue(ret, "Failed to get pvc status of %s" % app_2_pvc_name)
-        self.assertEqual(
-            status, "Pending",
-            "PVC status of %s is not in Pending state" % app_2_pvc_name)
+        app_2_pvc_name = oc_create_pvc(
+            self.node, pvc_name_prefix="autotest-file2", sc_name=sc_name
+        )
+        self.addCleanup(
+            wait_for_resource_absence, self.node, 'pvc', app_2_pvc_name)
+        self.addCleanup(
+            oc_delete, self.node, 'pvc', app_2_pvc_name, raise_on_absence=False
+        )
 
         # Create second app POD
         app_2_pod_name = oc_create_tiny_pod_with_volume(