Replace nova client calls with openstack sdk
[sdnvpn.git] / sdnvpn / test / functest / run_sdnvpn_tests.py
index 7da6571..3f3b519 100644 (file)
@@ -1,4 +1,4 @@
-#!/bin/python
+#!/usr/bin/env python
 #
 # Copyright (c) 2017 All rights reserved
 # This program and the accompanying materials
@@ -15,22 +15,23 @@ import sys
 import traceback
 import yaml
 
-from functest.core import feature as base
+from collections import OrderedDict
+from xtesting.core import feature
 from sdnvpn.lib import config as sdnvpn_config
 from sdnvpn.lib import openstack_utils as os_utils
 from sdnvpn.lib.gather_logs import gather_logs
 from sdnvpn.lib import utils as test_utils
 
-COMMON_CONFIG = sdnvpn_config.CommonConfig()
+logger = logging.getLogger(__name__)
 
+COMMON_CONFIG = sdnvpn_config.CommonConfig()
 
-class SdnvpnFunctest(base.Feature):
 
-    __logger = logging.getLogger(__name__)
+class SdnvpnFunctest(feature.Feature):
 
     def execute(self):
 
-        nova_client = os_utils.get_nova_client()
+        cloud = os_utils.get_os_cloud()
         neutron_client = os_utils.get_neutron_client()
 
         tenant_id = os_utils.get_tenant_id(os_utils.get_keystone_client(),
@@ -41,10 +42,10 @@ class SdnvpnFunctest(base.Feature):
          neutron_router_quota) = (
             neutron_quota['network'], neutron_quota['subnet'],
             neutron_quota['port'], neutron_quota['router'])
-        instances_quota = test_utils.get_nova_instances_quota(nova_client)
+        instances_quota = test_utils.get_nova_instances_quota(cloud)
 
-        self.__logger.info("Setting net/subnet/port/router "
-                           "quota to unlimited")
+        logger.info("Setting net/subnet/port/router "
+                    "quota to unlimited")
         test_utils.update_nw_subnet_port_quota(
             neutron_client,
             tenant_id,
@@ -56,48 +57,81 @@ class SdnvpnFunctest(base.Feature):
 
         # Workaround for
         # https://jira.opnfv.org/projects/SDNVPN/issues/SDNVPN-115
-        self.__logger.info("Setting instances quota class to unlimited")
+        logger.info("Setting instances quota class to unlimited")
         test_utils.update_instance_quota_class(
-            nova_client,
-            COMMON_CONFIG.nova_instances_quota_class)
+            cloud, COMMON_CONFIG.nova_instances_quota_class)
+
+        # Clean up the stale floating ip's so that required
+        # ip addresses are available for sdnvpn testcases
+        logger.info("Cleaning up the Floating IP Addresses")
+        floating_ips = os_utils.get_floating_ips(neutron_client)
+        if floating_ips is not None:
+            for floating_ip in floating_ips:
+                os_utils.delete_floating_ip(
+                    neutron_client, floating_ip['id'])
+
+        # Workaround for
+        # https://jira.opnfv.org/browse/SNAPS-318
+        # Clean up the stale routers
+        logger.info("Cleaning up the stale routers")
+        ports = os_utils.get_port_list(neutron_client)
+        if ports is not None:
+            for port in ports:
+                if port['device_owner'] == 'network:router_interface':
+                    os_utils.delete_neutron_port(
+                            neutron_client, port['id'])
+        routers = os_utils.get_router_list(neutron_client)
+        if routers is not None:
+            for router in routers:
+                os_utils.remove_gateway_router(
+                    neutron_client, router['id'])
+                os_utils.delete_neutron_router(
+                    neutron_client, router['id'])
 
         with open(COMMON_CONFIG.config_file) as f:
             config_yaml = yaml.safe_load(f)
 
         testcases = config_yaml.get("testcases")
+        testcases_ordered = OrderedDict(sorted(testcases.items(),
+                                               key=lambda x: x[1]['order']))
         overall_status = "PASS"
-        for tc in testcases:
-            if testcases[tc]['enabled']:
+        for tc, test_sdnvpn in testcases_ordered.items():
+            if test_sdnvpn['enabled']:
                 test_name = tc
                 test_descr = testcases[tc]['description']
                 title = ("Running '%s - %s'" %
                          (test_name, test_descr))
-                self.__logger.info(title)
-                self.__logger.info("%s\n" % ("=" * len(title)))
-                t = importlib.import_module(test_name, package=None)
+                logger.info(title)
+                logger.info("%s\n" % ("=" * len(title)))
                 try:
+                    logger.info("Importing the testcase %s" % test_name)
+                    t = importlib.import_module(test_name, package=None)
+                    logger.info("Calling the testcase %s main method"
+                                % test_name)
                     result = t.main()
+                    logger.info("Execution is complete for the"
+                                " testcase %s" % test_name)
                 except Exception as ex:
                     result = -1
-                    self.__logger.info("Caught Exception in %s: %s Trace: %s"
-                                       % (test_name, ex,
-                                          traceback.format_exc()))
+                    logger.info("Caught Exception in %s: %s Trace: %s"
+                                % (test_name, ex,
+                                   traceback.format_exc()))
                 if result < 0:
                     status = "FAIL"
                     overall_status = "FAIL"
-                    self.__logger.info("Testcase %s failed" % test_name)
+                    logger.info("Testcase %s failed" % test_name)
                 else:
                     status = result.get("status")
                     self.details.update(
                         {test_name: {'status': status,
                                      'details': result.get("details")}})
-                    self.__logger.info("Results of test case '%s - %s':\n%s\n"
-                                       % (test_name, test_descr, result))
+                    logger.info("Results of test case '%s - %s':\n%s\n"
+                                % (test_name, test_descr, result))
 
                     if status == "FAIL":
                         overall_status = "FAIL"
 
-        self.__logger.info("Resetting subnet/net/port quota")
+        logger.info("Resetting subnet/net/port quota")
         test_utils.update_nw_subnet_port_quota(neutron_client,
                                                tenant_id,
                                                neutron_nw_quota,
@@ -105,27 +139,27 @@ class SdnvpnFunctest(base.Feature):
                                                neutron_port_quota,
                                                neutron_router_quota)
 
-        self.__logger.info("Resetting instances quota class")
-        test_utils.update_instance_quota_class(nova_client, instances_quota)
+        logger.info("Resetting instances quota class")
+        test_utils.update_instance_quota_class(cloud, instances_quota)
 
         try:
             installer_type = str(os.environ['INSTALLER_TYPE'].lower())
             if installer_type in ["fuel", "apex"]:
                 gather_logs('overall')
             else:
-                self.__logger.info("Skipping log gathering because installer"
-                                   "type %s is neither fuel nor apex" %
-                                   installer_type)
+                logger.info("Skipping log gathering because installer"
+                            "type %s is neither fuel nor apex" %
+                            installer_type)
         except Exception as ex:
-            self.__logger.error(('Something went wrong in the Log gathering.'
-                                 'Ex: %s, Trace: %s')
-                                % (ex, traceback.format_exc()))
+            logger.error(('Something went wrong in the Log gathering.'
+                          'Ex: %s, Trace: %s')
+                         % (ex, traceback.format_exc()))
 
         if overall_status == "PASS":
             self.result = 100
-            return base.Feature.EX_OK
+            return feature.Feature.EX_OK
 
-        return base.Feature.EX_RUN_ERROR
+        return feature.Feature.EX_RUN_ERROR
 
 
 if __name__ == '__main__':