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.cmd = kwargs.get('cmd', '')
  23         self.result_file = "{}/{}.log".format(
  24             CONST.__getattribute__('dir_results'), self.project_name)
  25         self.logger = ft_logger.Logger(self.project_name).getLogger()
  26
  27     def execute(self, **kwargs):
  28         # pylint: disable=unused-argument,no-self-use
  29         return -1
  30
  31     def run(self, **kwargs):
  32         self.start_time = time.time()
  33         exit_code = base.TestCase.EX_RUN_ERROR
  34         self.criteria = "FAIL"
  35         try:
  36             if self.execute(**kwargs) == 0:
  37                 exit_code = base.TestCase.EX_OK
  38                 self.criteria = 'PASS'
  39             ft_utils.logger_test_results(
  40                 self.project_name, self.case_name,
  41                 self.criteria, self.details)
  42             self.logger.info("%s %s", self.project_name, self.criteria)
  43         except Exception:  # pylint: disable=broad-except
  44             self.logger.exception("%s FAILED", self.project_name)
  45         self.logger.info("Test result is stored in '%s'", self.result_file)
  46         self.stop_time = time.time()
  47         return exit_code
  48
  49
  50 class BashFeature(Feature):
  51
  52     def execute(self, **kwargs):
  53         ret = -1
  54         try:
  55             cmd = kwargs["cmd"]
  56             ret = ft_utils.execute_command(cmd, output_file=self.result_file)
  57         except KeyError:
  58             self.logger.error("Please give cmd as arg. kwargs: %s", kwargs)
  59         except Exception:  # pylint: disable=broad-except
  60             self.logger.exception("Execute cmd: %s failed", cmd)
  61         return ret