#!/usr/bin/python
#
# Copyright (c) 2015 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
#
# 0.1: This script boots the VM1 and allocates IP address from Nova
# Later, the VM2 boots then execute cloud-init to ping VM1.
# After successful ping, both the VMs are deleted.
# 0.2: measure test duration and publish results under json format
#
#

import logging
import os
import sys
import time
import yaml


with open('/home/opnfv/functest/conf/config_functest.yaml') as f:
    functest_yaml = yaml.safe_load(f)

dirs = functest_yaml.get('general').get('directories')
FUNCTEST_REPO = dirs.get('dir_repo_functest')
DOCTOR_REPO = dirs.get('dir_repo_doctor')
TEST_DB_URL = functest_yaml.get('results').get('test_db_url')

sys.path.append('%s/testcases' % FUNCTEST_REPO)
import functest_utils

logger = logging.getLogger('doctor')
logger.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
ch.setFormatter(formatter)
logger.addHandler(ch)


def main():
    cmd = 'cd %s/tests && ./run.sh' % DOCTOR_REPO
    start_time_ts = time.time()

    ret = functest_utils.execute_command(cmd, logger, exit_on_error=False)

    end_time_ts = time.time()
    duration = round(end_time_ts - start_time_ts, 1)
    if ret:
        logger.info("doctor OK")
        test_status = 'OK'
    else:
        logger.info("doctor FAILED")
        test_status = 'NOK'

    details = {
        'timestart': start_time_ts,
        'duration': duration,
        'status': test_status,
    }
    pod_name = functest_utils.get_pod_name(logger)
    scenario = functest_utils.get_scenario(logger)
    logger.info("Pushing result: TEST_DB_URL=%(db)s pod_name=%(pod)s "
                "scenario=%(s)s details=%(d)s" % {
                    'db': TEST_DB_URL,
                    'pod': pod_name,
                    's': scenario,
                    'd': details,
                })
    functest_utils.push_results_to_db(TEST_DB_URL,
                                      'doctor-notification',
                                      logger, pod_name, scenario,
                                      details)


if __name__ == '__main__':
    main()