Add new TCs remove or delete heketi node

Add new TC remove or delete node from heketi, and verify gluster
peer status, bricks on the new node and heal info.

Change-Id: If4b6006a4a58fd581ac9851b377a82f69ce0f30f

1 files changed, 4 insertions, 3 deletions

diff --git a/openshift-storage-libs/openshiftstoragelibs/gluster_ops.py b/openshift-storage-libs/openshiftstoragelibs/gluster_ops.py
index 785bde58..33ffa18d 100644
--- a/openshift-storage-libs/openshiftstoragelibs/gluster_ops.py
+++ b/openshift-storage-libs/openshiftstoragelibs/gluster_ops.py
@@ -26,16 +26,17 @@ from openshiftstoragelibs import waiter
 
 
 @podcmd.GlustoPod()
-def wait_to_heal_complete(timeout=300, wait_step=5):
+def wait_to_heal_complete(
+        timeout=300, wait_step=5, g_node="auto_get_gluster_endpoint"):
     """Monitors heal for volumes on gluster"""
-    gluster_vol_list = get_volume_list("auto_get_gluster_endpoint")
+    gluster_vol_list = get_volume_list(g_node)
     if not gluster_vol_list:
         raise AssertionError("failed to get gluster volume list")
 
     _waiter = waiter.Waiter(timeout=timeout, interval=wait_step)
     for gluster_vol in gluster_vol_list:
         for w in _waiter:
-            if is_heal_complete("auto_get_gluster_endpoint", gluster_vol):
+            if is_heal_complete(g_node, gluster_vol):
                 # NOTE(vponomar): Reset attempts for waiter to avoid redundant
                 # sleep equal to 'interval' on the next usage.
                 _waiter._attempt = 0