[CNS-944] Add test with arbiter volume creation

Change-Id: I257f3e6a7389c9f15a4388c241e787782c186d14

2 files changed, 61 insertions, 0 deletions

diff --git a/cns-libs/cnslibs/common/command.py b/cns-libs/cnslibs/common/command.py
new file mode 100644
index 00000000..e2c41545
--- /dev/null
+++ b/cns-libs/cnslibs/common/command.py
@@ -0,0 +1,20 @@
+from glusto.core import Glusto as g
+
+
+def cmd_run(cmd, hostname, raise_on_error=True):
+    """Glusto's command runner wrapper.
+
+    Args:
+        cmd (str): Shell command to run on the specified hostname.
+        hostname (str): hostname where Glusto should run specified command.
+        raise_on_error (bool): defines whether we should raise exception
+                               in case command execution failed.
+    Returns:
+        str: Stripped shell command's stdout value.
+    """
+    ret, out, err = g.run(hostname, cmd, "root")
+    if raise_on_error:
+        msg = ("Failed to execute command '%s' on '%s' node. Got non-zero "
+               "return code '%s'. Err: %s" % (cmd, hostname, ret, err))
+        assert int(ret) == 0, msg
+    return out.strip()
diff --git a/cns-libs/cnslibs/common/openshift_ops.py b/cns-libs/cnslibs/common/openshift_ops.py
index bbfa4947..ea1cea6a 100644
--- a/cns-libs/cnslibs/common/openshift_ops.py
+++ b/cns-libs/cnslibs/common/openshift_ops.py
@@ -9,8 +9,11 @@ import re
 import types
 
 from glusto.core import Glusto as g
+from glustolibs.gluster import volume_ops
+import mock
 import yaml
 
+from cnslibs.common import command
 from cnslibs.common import exceptions
 from cnslibs.common import utils
 from cnslibs.common import waiter
@@ -530,3 +533,41 @@ def scale_dc_pod_amount_and_wait(hostname, dc_name,
             wait_for_resource_absence(hostname, 'pod', pod)
         else:
             wait_for_pod_be_ready(hostname, pod)
+
+
+def get_gluster_vol_info_by_pvc_name(ocp_node, pvc_name):
+    """Get Gluster volume info based on the PVC name.
+
+    Args:
+        ocp_node (str): Node to execute OCP commands on.
+        pvc_name (str): Name of a PVC to get bound Gluster volume info.
+    Returns:
+        dict: Dictionary containting data about a Gluster volume.
+    """
+
+    # Get PV ID from PVC
+    get_pvc_cmd = "oc get pvc %s -o=custom-columns=:.spec.volumeName" % (
+        pvc_name)
+    pv_name = command.cmd_run(get_pvc_cmd, hostname=ocp_node)
+    assert pv_name, "PV name should not be empty: '%s'" % pv_name
+
+    # Get volume ID from PV
+    get_pv_cmd = "oc get pv %s -o=custom-columns=:.spec.glusterfs.path" % (
+        pv_name)
+    vol_id = command.cmd_run(get_pv_cmd, hostname=ocp_node)
+    assert vol_id, "Gluster volume ID should not be empty: '%s'" % vol_id
+
+    # Get name of one the Gluster PODs
+    gluster_pod = get_ocp_gluster_pod_names(ocp_node)[1]
+
+    # Get Gluster volume info
+    vol_info_cmd = "oc exec %s -- gluster v info %s --xml" % (
+        gluster_pod, vol_id)
+    vol_info = command.cmd_run(vol_info_cmd, hostname=ocp_node)
+
+    # Parse XML output to python dict
+    with mock.patch('glusto.core.Glusto.run', return_value=(0, vol_info, '')):
+        vol_info = volume_ops.get_volume_info(vol_id)
+        vol_info = vol_info[list(vol_info.keys())[0]]
+        vol_info["gluster_vol_id"] = vol_id
+        return vol_info