functest/core/feature.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2016 ZTE Corp and others.
   4 #
   5 # All rights reserved. This program and the accompanying materials
   6 # are made available under the terms of the Apache License, Version 2.0
   7 # which accompanies this distribution, and is available at
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9
  10 import time
  11
  12 import functest.core.testcase as base
  13 import functest.utils.functest_utils as ft_utils
  14 import functest.utils.functest_logger as ft_logger
  15 from functest.utils.constants import CONST
  16
  17
  18 class Feature(base.TestCase):
  19
  20     def __init__(self, **kwargs):
  21         super(Feature, self).__init__(**kwargs)
  22         self.result_file = "{}/{}.log".format(
  23             CONST.__getattribute__('dir_results'), self.project_name)
  24         self.logger = ft_logger.Logger(self.project_name).getLogger()
  25
  26     def execute(self, **kwargs):
  27         # pylint: disable=unused-argument,no-self-use
  28         return -1
  29
  30     def run(self, **kwargs):
  31         self.start_time = time.time()
  32         exit_code = base.TestCase.EX_RUN_ERROR
  33         self.criteria = "FAIL"
  34         try:
  35             if self.execute(**kwargs) == 0:
  36                 exit_code = base.TestCase.EX_OK
  37                 self.criteria = 'PASS'
  38             ft_utils.logger_test_results(
  39                 self.project_name, self.case_name,
  40                 self.criteria, self.details)
  41             self.logger.info("%s %s", self.project_name, self.criteria)
  42         except Exception:  # pylint: disable=broad-except
  43             self.logger.exception("%s FAILED", self.project_name)
  44         self.logger.info("Test result is stored in '%s'", self.result_file)
  45         self.stop_time = time.time()
  46         return exit_code
  47
  48
  49 class BashFeature(Feature):
  50
  51     def execute(self, **kwargs):
  52         ret = -1
  53         try:
  54             cmd = kwargs["cmd"]
  55             ret = ft_utils.execute_command(cmd, output_file=self.result_file)
  56         except KeyError:
  57             self.logger.error("Please give cmd as arg. kwargs: %s", kwargs)
  58         except Exception:  # pylint: disable=broad-except
  59             self.logger.exception("Execute cmd: %s failed", cmd)
  60         return ret