Add validation abilities to test cases

Change-Id: I76b28e6170d6e91836b195d58c0b882168c11a67
Issue-ID: VNFSDK-275
Signed-off-by: Moshe <moshehoa@amdocs.com>

Add unit tests

Issue-ID: VNFSDK-275
Change-Id: I34bc9a11e16e4092fdad3b4a1733c7219e624f5f
Signed-off-by: Moshe <moshehoa@amdocs.com>

add unit tests

Issue-ID: VNFSDK-275
Change-Id: Ib99c3521438b002e0d8aaff9870224673e34899f
Signed-off-by: Moshe <moshehoa@amdocs.com>

add unit tests

Issue-ID: VNFSDK-275
Change-Id: I1ac560dfb40df5f346b0db8f40b8c52a2fb6b350
Signed-off-by: Moshe <moshehoa@amdocs.com>

1 files changed, 51 insertions, 0 deletions

diff --git a/vnftest/common/openstack_utils.py b/vnftest/common/openstack_utils.py
index c97c1c2..829b916 100644
--- a/vnftest/common/openstack_utils.py
+++ b/vnftest/common/openstack_utils.py
@@ -27,6 +27,7 @@ from cinderclient import client as cinderclient
 from novaclient import client as novaclient
 from glanceclient import client as glanceclient
 from neutronclient.neutron import client as neutronclient
+from heatclient.client import Client as heatclient
 
 log = logging.getLogger(__name__)
 
@@ -159,6 +160,11 @@ def get_neutron_client():   # pragma: no cover
     return neutronclient.Client(get_neutron_client_version(), session=sess)
 
 
+def get_heat_client():   # pragma: no cover
+    sess = get_session()
+    return heatclient(get_heat_api_version(), session=sess)
+
+
 def get_glance_client_version():    # pragma: no cover
     try:
         api_version = os.environ['OS_IMAGE_API_VERSION']
@@ -199,6 +205,14 @@ def get_instance_by_name(nova_client, instance_name):   # pragma: no cover
                       instance_name)
 
 
+def get_instance_by_id(instance_id):   # pragma: no cover
+    try:
+        return get_nova_client().servers.find(id=instance_id)
+    except Exception:
+        log.exception("Error [get_instance_by_id(nova_client, '%s')]",
+                      instance_id)
+
+
 def get_aggregates(nova_client):    # pragma: no cover
     try:
         return nova_client.aggregates.list()
@@ -440,6 +454,15 @@ def delete_keypair(nova_client, key):     # pragma: no cover
 # *********************************************
 #   NEUTRON
 # *********************************************
+def get_network_by_name(network_name):       # pragma: no cover
+    try:
+        networks = get_neutron_client().list_networks()['networks']
+        return next((n for n in networks if n['name'] == network_name), None)
+    except Exception:
+        log.exception("Error [get_instance_by_id(nova_client, '%s')]",
+                      network_name)
+
+
 def get_network_id(neutron_client, network_name):       # pragma: no cover
     networks = neutron_client.list_networks()['networks']
     return next((n['id'] for n in networks if n['name'] == network_name), None)
@@ -760,3 +783,31 @@ def detach_volume(server_id, volume_id):      # pragma: no cover
         log.exception("Error [detach_server_volume(nova_client, '%s', '%s')]",
                       server_id, volume_id)
         return False
+# *********************************************
+#   HEAT
+# *********************************************
+
+
+def get_stack(heat_stack_id):    # pragma: no cover
+    try:
+        client = get_heat_client()
+        return client.stacks.get(heat_stack_id)
+    except Exception as e:
+        log.exception("Error [get_stack(heat_stack_id)]", e)
+
+
+def get_stack_resources(heat_stack_id):    # pragma: no cover
+    try:
+        client = get_heat_client()
+        return client.resources.list(heat_stack_id)
+    except Exception as e:
+        log.exception("Error [get_stack_resources(heat_stack_id)]", e)
+
+
+def get_stack_vms(heat_stack_id):    # pragma: no cover
+    resources = get_stack_resources(heat_stack_id)
+    ret_vms = []
+    for resource in resources:
+        if resource.resource_type == "OS::Nova::Server":
+            ret_vms.append(resource)
+    return ret_vms