#!/usr/bin/env python
#
# Copyright (c) 2017 All rights reserved
# This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
#
# http://www.apache.org/licenses/LICENSE-2.0
#

import importlib
import logging
import os
import sys
import traceback
import yaml

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

logger = logging.getLogger(__name__)

COMMON_CONFIG = sdnvpn_config.CommonConfig()


class SdnvpnFunctest(feature.Feature):

    def execute(self):

        nova_client = os_utils.get_nova_client()
        neutron_client = os_utils.get_neutron_client()

        tenant_id = os_utils.get_tenant_id(os_utils.get_keystone_client(),
                                           os.environ['OS_PROJECT_NAME'])

        neutron_quota = test_utils.get_neutron_quota(neutron_client, tenant_id)
        (neutron_nw_quota, neutron_subnet_quota, neutron_port_quota,
         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)

        logger.info("Setting net/subnet/port/router "
                    "quota to unlimited")
        test_utils.update_nw_subnet_port_quota(
            neutron_client,
            tenant_id,
            COMMON_CONFIG.neutron_nw_quota,
            COMMON_CONFIG.neutron_subnet_quota,
            COMMON_CONFIG.neutron_port_quota,
            COMMON_CONFIG.neutron_router_quota)
        test_utils.create_default_flavor()

        # Workaround for
        # https://jira.opnfv.org/projects/SDNVPN/issues/SDNVPN-115
        logger.info("Setting instances quota class to unlimited")
        test_utils.update_instance_quota_class(
            nova_client,
            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'])

        with open(COMMON_CONFIG.config_file) as f:
            config_yaml = yaml.safe_load(f)

        testcases = config_yaml.get("testcases")
        overall_status = "PASS"
        for tc in testcases:
            if testcases[tc]['enabled']:
                test_name = tc
                test_descr = testcases[tc]['description']
                title = ("Running '%s - %s'" %
                         (test_name, test_descr))
                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
                    logger.info("Caught Exception in %s: %s Trace: %s"
                                % (test_name, ex,
                                   traceback.format_exc()))
                if result < 0:
                    status = "FAIL"
                    overall_status = "FAIL"
                    logger.info("Testcase %s failed" % test_name)
                else:
                    status = result.get("status")
                    self.details.update(
                        {test_name: {'status': status,
                                     'details': result.get("details")}})
                    logger.info("Results of test case '%s - %s':\n%s\n"
                                % (test_name, test_descr, result))

                    if status == "FAIL":
                        overall_status = "FAIL"

        logger.info("Resetting subnet/net/port quota")
        test_utils.update_nw_subnet_port_quota(neutron_client,
                                               tenant_id,
                                               neutron_nw_quota,
                                               neutron_subnet_quota,
                                               neutron_port_quota,
                                               neutron_router_quota)

        logger.info("Resetting instances quota class")
        test_utils.update_instance_quota_class(nova_client, instances_quota)

        try:
            installer_type = str(os.environ['INSTALLER_TYPE'].lower())
            if installer_type in ["fuel", "apex"]:
                gather_logs('overall')
            else:
                logger.info("Skipping log gathering because installer"
                            "type %s is neither fuel nor apex" %
                            installer_type)
        except Exception as ex:
            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 feature.Feature.EX_OK

        return feature.Feature.EX_RUN_ERROR


if __name__ == '__main__':
    logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s'
                        '- %(levelname)s - %(message)s')
    SDNVPN = SdnvpnFunctest()
    sys.exit(SDNVPN.execute())