functest/core/feature.py

   1 #!/usr/bin/env python
   2
   3 import time
   4
   5 import functest.core.testcase as base
   6 import functest.utils.functest_utils as ft_utils
   7 import functest.utils.functest_logger as ft_logger
   8 from functest.utils.constants import CONST
   9
  10
  11 class Feature(base.TestCase):
  12
  13     def __init__(self, **kwargs):
  14         super(Feature, self).__init__(**kwargs)
  15         self.cmd = kwargs.get('cmd', '')
  16         self.result_file = "{}/{}.log".format(
  17             CONST.__getattribute__('dir_results'), self.project_name)
  18         self.logger = ft_logger.Logger(self.project_name).getLogger()
  19
  20     def execute(self, **kwargs):
  21         # pylint: disable=unused-argument,no-self-use
  22         return -1
  23
  24     def run(self, **kwargs):
  25         self.start_time = time.time()
  26         exit_code = base.TestCase.EX_RUN_ERROR
  27         self.criteria = "FAIL"
  28         try:
  29             if self.execute(**kwargs) == 0:
  30                 exit_code = base.TestCase.EX_OK
  31                 self.criteria = 'PASS'
  32             ft_utils.logger_test_results(
  33                 self.project_name, self.case_name,
  34                 self.criteria, self.details)
  35             self.logger.info("%s %s", self.project_name, self.criteria)
  36         except Exception:  # pylint: disable=broad-except
  37             self.logger.exception("%s FAILED", self.project_name)
  38         self.logger.info("Test result is stored in '%s'", self.result_file)
  39         self.stop_time = time.time()
  40         return exit_code
  41
  42
  43 class BashFeature(Feature):
  44
  45     def execute(self, **kwargs):
  46         ret = -1
  47         try:
  48             cmd = kwargs["cmd"]
  49             ret = ft_utils.execute_command(cmd, output_file=self.result_file)
  50         except KeyError:
  51             self.logger.error("Please give cmd as arg. kwargs: %s", kwargs)
  52         except Exception:  # pylint: disable=broad-except
  53             self.logger.exception("Execute cmd: %s failed", cmd)
  54         return ret