functest/core/feature_base.py

   1 import time
   2
   3 import testcase_base as base
   4 import functest.utils.functest_utils as ft_utils
   5 import functest.utils.functest_logger as ft_logger
   6 from functest.utils.constants import CONST
   7
   8
   9 class FeatureBase(base.TestcaseBase):
  10     def __init__(self, project='functest', case='', repo='', cmd=''):
  11         super(FeatureBase, self).__init__()
  12         self.project_name = project
  13         self.case_name = case
  14         self.cmd = cmd
  15         self.repo = CONST.__getattribute__(repo)
  16         self.result_file = self.get_result_file()
  17         self.logger = ft_logger.Logger(project).getLogger()
  18
  19     def run(self, **kwargs):
  20         self.prepare()
  21         self.start_time = time.time()
  22         ret = ft_utils.execute_command(self.cmd, output_file=self.result_file)
  23         self.stop_time = time.time()
  24         self.post()
  25         self.parse_results(ret)
  26         self.log_results()
  27         self.logger.info("Test result is stored in '%s'" % self.result_file)
  28         return base.TestcaseBase.EX_OK
  29
  30     def prepare(self, **kwargs):
  31         pass
  32
  33     def post(self, **kwargs):
  34         pass
  35
  36     def parse_results(self, ret):
  37         exit_code = base.TestcaseBase.EX_OK
  38         if ret == 0:
  39             self.logger.info("{} OK".format(self.project_name))
  40             self.criteria = 'PASS'
  41         else:
  42             self.logger.info("{} FAILED".format(self.project_name))
  43             exit_code = base.TestcaseBase.EX_RUN_ERROR
  44             self.criteria = "FAIL"
  45
  46         return exit_code
  47
  48     def get_result_file(self):
  49         return "{}/{}.log".format(CONST.dir_results, self.project_name)
  50
  51     def log_results(self):
  52         ft_utils.logger_test_results(self.project_name,
  53                                      self.case_name,
  54                                      self.criteria,
  55                                      self.details)