[Test] functional/disperse: Verify remove brick operation

This test verifies remove brick operations on disperse
volume.

Change-Id: If4be3ffc39a8b58e4296d58b288e3843a218c468
Co-authored-by: Sunil Kumar Acharya <sheggodu@redhat.com>
Signed-off-by: Sunil Kumar Acharya <sheggodu@redhat.com>
Signed-off-by: Sri Vignesh <sselvan@redhat.com>

1 files changed, 44 insertions, 1 deletions

diff --git a/glustolibs-gluster/glustolibs/gluster/rebalance_ops.py b/glustolibs-gluster/glustolibs/gluster/rebalance_ops.py
index 1c8c10a4b..d74a04629 100644
--- a/glustolibs-gluster/glustolibs/gluster/rebalance_ops.py
+++ b/glustolibs-gluster/glustolibs/gluster/rebalance_ops.py
@@ -1,5 +1,5 @@
 #!/usr/bin/env python
-#  Copyright (C) 2015-2016  Red Hat, Inc. <http://www.redhat.com>
+#  Copyright (C) 2015-2020 Red Hat, Inc. <http://www.redhat.com>
 #
 #  This program is free software; you can redistribute it and/or modify
 #  it under the terms of the GNU General Public License as published by
@@ -401,3 +401,46 @@ def get_remove_brick_status(mnode, volname, bricks_list):
             else:
                 remove_brick_status[element.tag] = element.text
     return remove_brick_status
+
+
+def wait_for_remove_brick_to_complete(mnode, volname, bricks_list,
+                                      timeout=1200):
+    """Waits for the remove brick to complete
+
+    Args:
+        mnode (str): Node on which command has to be executed.
+        volname (str): volume name
+        bricks_list (str): List of bricks participating in
+        remove-brick operation
+
+    Kwargs:
+        timeout (int): timeout value in seconds to wait for remove brick
+            to complete
+
+    Returns:
+        True on success, False otherwise
+
+    Examples:
+        >>> wait_for_remove_brick_to_complete("abc.com", "testvol")
+    """
+
+    count = 0
+    while count < timeout:
+        status_info = get_remove_brick_status(mnode, volname, bricks_list)
+        if status_info is None:
+            return False
+        status = status_info['aggregate']['statusStr']
+        if status == 'completed':
+            g.log.info("Remove brick is successfully completed in %s sec",
+                       count)
+            return True
+        elif status == 'failed':
+            g.log.error(" Remove brick failed on one or more nodes. "
+                        "Check remove brick status for more details")
+            return False
+        else:
+            time.sleep(10)
+            count += 10
+            g.log.error("Remove brick operation has not completed. "
+                        "Wait timeout is %s" % count)
+    return False