functest/core/feature.py

   1 import time
   2
   3 import functest.core.testcase 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 Feature(base.TestCase):
  10
  11     def __init__(self, **kwargs):
  12         super(Feature, self).__init__(**kwargs)
  13         self.cmd = kwargs.get('cmd', '')
  14         self.result_file = "{}/{}.log".format(
  15             CONST.__getattribute__('dir_results'), self.project_name)
  16         self.logger = ft_logger.Logger(self.project_name).getLogger()
  17
  18     def execute(self, **kwargs):
  19         return -1
  20
  21     def run(self, **kwargs):
  22         self.start_time = time.time()
  23         exit_code = base.TestCase.EX_RUN_ERROR
  24         self.criteria = "FAIL"
  25         try:
  26             if self.execute() == 0:
  27                 exit_code = base.TestCase.EX_OK
  28                 self.criteria = 'PASS'
  29             ft_utils.logger_test_results(
  30                 self.project_name, self.case_name,
  31                 self.criteria, self.details)
  32             self.logger.info("%s %s", self.project_name, self.criteria)
  33         except Exception:  # pylint: disable=broad-except
  34             self.logger.exception("%s FAILED", self.project_name)
  35         self.logger.info("Test result is stored in '%s'", self.result_file)
  36         self.stop_time = time.time()
  37         return exit_code
  38
  39
  40 class BashFeature(Feature):
  41
  42     def execute(self, **kwargs):
  43         return ft_utils.execute_command(
  44             self.cmd, output_file=self.result_file)