#!/bin/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 argparse
import importlib
import logging
import os
import sys
import time
import traceback
import yaml

import functest.utils.functest_utils as ft_utils
from sdnvpn.lib import config as sdnvpn_config
from sdnvpn.lib.gather_logs import gather_logs

logger = logging.getLogger('sdnvpn-run-tests')

COMMON_CONFIG = sdnvpn_config.CommonConfig()
TEST_DB_URL = COMMON_CONFIG.test_db


def push_results(testname, start_time, end_time, criteria, details):
    logger.info("Push testcase '%s' results into the DB...\n" % testname)
    ft_utils.push_results_to_db("sdnvpn",
                                testname,
                                start_time,
                                end_time,
                                criteria,
                                details)


def main(report=False):
    # Workaround for https://jira.opnfv.org/projects/SDNVPN/issues/SDNVPN-100
    # and SDNVPN-126
    cmd_line = "neutron quota-update --subnet -1 --network -1 --port -1"
    logger.info("Setting subnet/net quota to unlimited : %s" % cmd_line)
    cmd = os.popen(cmd_line)
    output = cmd.read()
    logger.debug(output)

    # Workaround for https://jira.opnfv.org/projects/SDNVPN/issues/SDNVPN-115
    cmd_line = "nova quota-class-update --instances -1 default"
    logger.info("Setting instances quota to unlimited : %s" % cmd_line)
    cmd = os.popen(cmd_line)
    output = cmd.read()
    logger.debug(output)

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

    testcases = config_yaml.get("testcases")
    overall_status = "PASS"
    for testcase in testcases:
        if testcases[testcase]['enabled']:
            test_name = testcase
            test_descr = testcases[testcase]['description']
            test_name_db = testcases[testcase]['testname_db']
            title = ("Running '%s - %s'" %
                     (test_name, test_descr))
            logger.info(title)
            logger.info("%s\n" % ("=" * len(title)))
            t = importlib.import_module(testcase, package=None)
            start_time = time.time()
            try:
                result = t.main()
            except Exception as ex:
                result = -1
                logger.info("Caught Exception in %s: %s Trace: %s" %
                            (test_name, ex, traceback.format_exc()))
            end_time = time.time()
            if result < 0:
                status = "FAIL"
                overall_status = "FAIL"
                logger.info("Testcase %s failed" % test_name)
            else:
                status = result.get("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"

            if report:
                push_results(
                    test_name_db, start_time, end_time, status, details)

    try:
        gather_logs('overall')
    except Exception as ex:
        logger.error(('Something went wrong in the Log gathering.'
                      'Ex: %s, Trace: %s')
                     % ex, traceback.format_exc())
    if overall_status == "FAIL":
        sys.exit(-1)

    sys.exit(0)


if __name__ == '__main__':
    logging.basicConfig(level=logging.INFO)
    parser = argparse.ArgumentParser()
    parser.add_argument("-r", "--report",
                        help="Create json result file",
                        action="store_true")
    args = parser.parse_args()
    main(report=args.report)