Add TCs creating block-PVCs and app pods changing node scheduling policy

Add 2 test cases, where one verifies creation of an app pod on the
Gluster node and another on the separate node

Change-Id: I99dfc5db7fa74d0f69115cfed470f72e66b1a256

2 files changed, 149 insertions, 3 deletions

diff --git a/cns-libs/cnslibs/common/cns_libs.py b/cns-libs/cnslibs/common/cns_libs.py
index 16c050ca..03966475 100644
--- a/cns-libs/cnslibs/common/cns_libs.py
+++ b/cns-libs/cnslibs/common/cns_libs.py
@@ -1,6 +1,7 @@
 from glusto.core import Glusto as g
 import yaml
 
+from cnslibs.common.command import cmd_run
 from cnslibs.common.exceptions import (
     ExecutionError,
     NotSupportedException)
@@ -10,7 +11,7 @@ from cnslibs.common.openshift_version import get_openshift_version
 MASTER_CONFIG_FILEPATH = "/etc/origin/master/master-config.yaml"
 
 
-def validate_multipath_pod(hostname, podname, hacount):
+def validate_multipath_pod(hostname, podname, hacount, mpath=""):
     '''
      This function validates multipath for given app-pod
      Args:
@@ -31,7 +32,7 @@ def validate_multipath_pod(hostname, podname, hacount):
     pod_nodename = out.strip()
     active_node_count = 1
     enable_node_count = hacount - 1
-    cmd = "multipath -ll | grep 'status=active' | wc -l"
+    cmd = "multipath -ll %s | grep 'status=active' | wc -l" % mpath
     ret, out, err = g.run(pod_nodename, cmd, "root")
     if ret != 0 or out == "":
         g.log.error("failed to exectute cmd %s on %s, err %s"
@@ -42,7 +43,7 @@ def validate_multipath_pod(hostname, podname, hacount):
         g.log.error("active node count on %s for %s is %s and not 1"
                     % (pod_nodename, podname, active_count))
         return False
-    cmd = "multipath -ll | grep 'status=enabled' | wc -l"
+    cmd = "multipath -ll %s | grep 'status=enabled' | wc -l" % mpath
     ret, out, err = g.run(pod_nodename, cmd, "root")
     if ret != 0 or out == "":
         g.log.error("failed to exectute cmd %s on %s, err %s"
@@ -132,3 +133,95 @@ def enable_pvc_resize(master_node):
         raise ExecutionError(err_msg)
 
     return True
+
+
+def get_iscsi_session(node, iqn=None, raise_on_error=True):
+    """Get the list of ip's of iscsi sessions.
+
+    Args:
+        node (str): where we want to run the command.
+        iqn (str): name of iqn.
+    Returns:
+        list: list of session ip's.
+    raises:
+        ExecutionError: In case of any failure if raise_on_error=True.
+    """
+
+    cmd = "set -o pipefail && ((iscsiadm -m session"
+    if iqn:
+        cmd += " | grep %s" % iqn
+    cmd += ") | awk '{print $3}' | cut -d ':' -f 1)"
+
+    out = cmd_run(cmd, node, raise_on_error=raise_on_error)
+
+    return out.split("\n") if out else out
+
+
+def get_iscsi_block_devices_by_path(node, iqn=None, raise_on_error=True):
+    """Get list of iscsiadm block devices from path.
+
+    Args:
+        node (str): where we want to run the command.
+        iqn (str): name of iqn.
+    returns:
+        dictionary: block devices and there ips.
+    raises:
+        ExecutionError: In case of any failure if raise_on_error=True.
+    """
+    cmd = "set -o pipefail && ((ls --format=context /dev/disk/by-path/ip*"
+    if iqn:
+        cmd += " | grep %s" % iqn
+    cmd += ") | awk -F '/|:|-' '{print $10,$25}')"
+
+    out = cmd_run(cmd, node, raise_on_error=raise_on_error)
+
+    if not out:
+        return out
+
+    out_dic = {}
+    for i in out.split("\n"):
+        ip, device = i.strip().split(" ")
+        out_dic[device] = ip
+
+    return out_dic
+
+
+def get_mpath_name_from_device_name(node, device, raise_on_error=True):
+    """Get name of mpath device form block device
+
+    Args:
+        node (str): where we want to run the command.
+        device (str): for which we have to find mpath.
+    Returns:
+        str: name of device
+    Raises:
+        ExecutionError: In case of any failure if raise_on_error=True.
+    """
+    cmd = ("set -o pipefail && ((lsblk -n --list --output=NAME /dev/%s)"
+           " | tail -1)" % device)
+
+    return cmd_run(cmd, node, raise_on_error=raise_on_error)
+
+
+def get_active_and_enabled_devices_from_mpath(node, mpath):
+    """Get active and enabled devices from mpath name.
+
+    Args:
+        node (str): where we want to run the command.
+        mpath (str): name of mpath for which we have to find devices.
+    Returns:
+        dictionary: devices info
+    Raises:
+        ExecutionError: In case of any failure
+    """
+
+    cmd = ("set -o pipefail && ((multipath -ll %s | grep -A 1 status=%s)"
+           " | cut -d ':' -f 4 | awk '{print $2}')")
+
+    active = cmd_run(cmd % (mpath, 'active'), node).split('\n')[1::2]
+    enabled = cmd_run(cmd % (mpath, 'enabled'), node).split('\n')[1::2]
+
+    out_dic = {
+        'active': active,
+        'enabled': enabled}
+    return out_dic
diff --git a/cns-libs/cnslibs/common/openshift_ops.py b/cns-libs/cnslibs/common/openshift_ops.py
index 2899d531..b7390152 100644
--- a/cns-libs/cnslibs/common/openshift_ops.py
+++ b/cns-libs/cnslibs/common/openshift_ops.py
@@ -1451,3 +1451,56 @@ def restart_service_on_gluster_pod_or_node(ocp_client, service, gluster_node):
     """
     cmd_run_on_gluster_pod_or_node(
         ocp_client, SERVICE_RESTART % service, gluster_node)
+
+
+def oc_adm_manage_node(
+        ocp_client, operation, nodes=None, node_selector=None):
+    """Manage common operations on nodes for administrators.
+
+    Args:
+        ocp_client (str): host on which we want to run 'oc' commands.
+        operations (str):
+            eg. --schedulable=true.
+        nodes (list): list of nodes to manage.
+        node_selector (str): selector to select the nodes.
+            Note: 'nodes' and 'node_selector' are are mutually exclusive.
+            Only either of them should be passed as parameter not both.
+    Returns:
+        str: In case of success.
+    Raises:
+        AssertionError: In case of any failures.
+    """
+
+    if (not nodes) == (not node_selector):
+        raise AssertionError(
+            "'nodes' and 'node_selector' are mutually exclusive. "
+            "Only either of them should be passed as parameter not both.")
+
+    cmd = "oc adm manage-node %s" % operation
+    if node_selector:
+        cmd += " --selector %s" % node_selector
+    else:
+        node = ' '.join(nodes)
+        cmd += " " + node
+
+    return command.cmd_run(cmd, ocp_client)
+
+
+def oc_get_schedulable_nodes(ocp_client):
+    """Get the list of schedulable nodes.
+
+    Args:
+        ocp_client (str): host on which we want to run 'oc' commands.
+
+    Returns:
+        list: list of nodes if present.
+    Raises:
+        AssertionError: In case of any failures.
+    """
+    cmd = ("oc get nodes --field-selector=spec.unschedulable!=true "
+           "-o=custom-columns=:.metadata.name,:.spec.taints[*].effect "
+           "--no-headers | awk '!/NoSchedule/{print $1}'")
+
+    out = command.cmd_run(cmd, ocp_client)
+
+    return out.split('\n') if out else out