#!/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 argparse
import time

import functest.utils.functest_logger as ft_logger
import functest.utils.functest_utils as functest_utils


parser = argparse.ArgumentParser()
parser.add_argument("-r", "--report",
                    help="Create json result file",
                    action="store_true")
args = parser.parse_args()

functest_yaml = functest_utils.get_functest_yaml()

dirs = functest_yaml.get('general').get('directories')
DOCTOR_REPO = dirs.get('dir_repo_doctor')
RESULTS_DIR = functest_utils.get_parameter_from_yaml(
    'general.directories.dir_results')

logger = ft_logger.Logger("doctor").getLogger()


def main():
    exit_code = -1
    cmd = 'cd %s/tests && ./run.sh' % DOCTOR_REPO
    log_file = RESULTS_DIR + "/doctor.log"

    start_time = time.time()

    ret = functest_utils.execute_command(cmd,
                                         info=True,
                                         exit_on_error=False,
                                         output_file=log_file)

    stop_time = time.time()
    duration = round(stop_time - start_time, 1)
    if ret == 0:
        logger.info("Doctor test case OK")
        test_status = 'OK'
        exit_code = 0
    else:
        logger.info("Doctor test case FAILED")
        test_status = 'NOK'

    details = {
        'timestart': start_time,
        'duration': duration,
        'status': test_status,
    }
    status = "FAIL"
    if details['status'] == "OK":
        status = "PASS"
    functest_utils.logger_test_results("Doctor",
                                       "doctor-notification",
                                       status, details)
    if args.report:
        functest_utils.push_results_to_db("doctor",
                                          "doctor-notification",
                                          start_time,
                                          stop_time,
                                          status,
                                          details)
        logger.info("Doctor results pushed to DB")

    exit(exit_code)

if __name__ == '__main__':
    main()

@@ -546,16 +546,26 @@
 rgparse%0A
+import os%0A
 import t
@@ -568,16 +568,16 @@
 rt time%0A
-
 %0Aimport 
@@ -923,20 +923,27 @@
 yaml()%0A%0A
-dirs
+DOCTOR_REPO
  = funct
@@ -950,71 +950,64 @@
 est_
-yaml.get('general').get('directories')%0ADOCTOR_REPO = dirs.get('
+utils.get_parameter_from_yaml(%0A    'general.directories.
 dir_
@@ -1176,16 +1176,16 @@
 main():%0A
-
     exit
@@ -1195,16 +1195,269 @@
 de = -1%0A
+%0A    # if the image name is explicitly set for the doctor suite, set it as%0A    # enviroment variable%0A    if 'doctor' in functest_yaml and 'image_name' in functest_yaml%5B'doctor'%5D:%0A        os.environ%5B%22IMAGE_NAME%22%5D = functest_yaml%5B'doctor'%5D%5B'image_name'%5D%0A%0A
     cmd 

Add support of arm for doctor