testcases/features/doctor.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (c) 2015 All rights reserved
   4 # This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 #
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # 0.1: This script boots the VM1 and allocates IP address from Nova
  11 # Later, the VM2 boots then execute cloud-init to ping VM1.
  12 # After successful ping, both the VMs are deleted.
  13 # 0.2: measure test duration and publish results under json format
  14 #
  15 #
  16 import argparse
  17 import os
  18 import time
  19 import yaml
  20
  21 import functest.utils.functest_logger as ft_logger
  22 import functest.utils.functest_utils as functest_utils
  23
  24 parser = argparse.ArgumentParser()
  25 parser.add_argument("-r", "--report",
  26                     help="Create json result file",
  27                     action="store_true")
  28 args = parser.parse_args()
  29
  30 with open(os.environ["CONFIG_FUNCTEST_YAML"]) as f:
  31     functest_yaml = yaml.safe_load(f)
  32
  33 dirs = functest_yaml.get('general').get('directories')
  34 DOCTOR_REPO = dirs.get('dir_repo_doctor')
  35
  36 logger = ft_logger.Logger("doctor").getLogger()
  37
  38
  39 def main():
  40     exit_code = -1
  41     cmd = 'cd %s/tests && ./run.sh' % DOCTOR_REPO
  42     start_time = time.time()
  43
  44     ret = functest_utils.execute_command(cmd, info=True,
  45                                          exit_on_error=False)
  46
  47     stop_time = time.time()
  48     duration = round(stop_time - start_time, 1)
  49     if ret == 0:
  50         logger.info("doctor OK")
  51         test_status = 'OK'
  52         exit_code = 0
  53     else:
  54         logger.info("doctor FAILED")
  55         test_status = 'NOK'
  56
  57     details = {
  58         'timestart': start_time,
  59         'duration': duration,
  60         'status': test_status,
  61     }
  62     status = "FAIL"
  63     if details['status'] == "OK":
  64         status = "PASS"
  65     functest_utils.logger_test_results("Doctor",
  66                                        "doctor-notification",
  67                                        status, details)
  68     if args.report:
  69         functest_utils.push_results_to_db("doctor",
  70                                           "doctor-notification",
  71                                           start_time,
  72                                           stop_time,
  73                                           status,
  74                                           details)
  75         logger.info("Doctor results pushed to DB")
  76
  77     exit(exit_code)
  78
  79 if __name__ == '__main__':
  80     main()