functest/tests/unit/core/test_feature.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2017 Orange 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 # pylint: disable=missing-docstring
  11
  12 import logging
  13 import unittest
  14
  15 import mock
  16
  17 from functest.core import feature
  18 from functest.core import testcase
  19
  20
  21 class FeatureTestingBase(unittest.TestCase):
  22
  23     _case_name = "foo"
  24     _project_name = "bar"
  25     _repo = "dir_repo_bar"
  26     _cmd = "run_bar_tests.py"
  27     _output_file = '/home/opnfv/functest/results/foo.log'
  28     feature = None
  29
  30     @mock.patch('time.time', side_effect=[1, 2])
  31     def _test_run(self, status, mock_method=None):
  32         self.assertEqual(self.feature.run(cmd=self._cmd), status)
  33         if status == testcase.TestCase.EX_OK:
  34             self.assertEqual(self.feature.result, 100)
  35         else:
  36             self.assertEqual(self.feature.result, 0)
  37         mock_method.assert_has_calls([mock.call(), mock.call()])
  38         self.assertEqual(self.feature.start_time, 1)
  39         self.assertEqual(self.feature.stop_time, 2)
  40
  41     def test_logger_module_ko(self):
  42         with mock.patch('six.moves.builtins.open'):
  43             self.feature = feature.Feature(
  44                 project_name=self._project_name, case_name=self._case_name)
  45             self.assertEqual(self.feature.logger.name, self._case_name)
  46
  47     def test_logger_module(self):
  48         with mock.patch('six.moves.builtins.open'):
  49             self.feature = feature.Feature(
  50                 project_name=self._project_name, case_name=self._case_name,
  51                 run={'module': 'bar'})
  52             self.assertEqual(self.feature.logger.name, 'bar')
  53
  54
  55 class FeatureTesting(FeatureTestingBase):
  56
  57     def setUp(self):
  58         # logging must be disabled else it calls time.time()
  59         # what will break these unit tests.
  60         logging.disable(logging.CRITICAL)
  61         with mock.patch('six.moves.builtins.open'):
  62             self.feature = feature.Feature(
  63                 project_name=self._project_name, case_name=self._case_name)
  64
  65     def test_run_exc(self):
  66         # pylint: disable=bad-continuation
  67         with mock.patch.object(
  68                 self.feature, 'execute',
  69                 side_effect=Exception) as mock_method:
  70             self._test_run(testcase.TestCase.EX_RUN_ERROR)
  71             mock_method.assert_called_once_with(cmd=self._cmd)
  72
  73     def test_run(self):
  74         self._test_run(testcase.TestCase.EX_RUN_ERROR)
  75
  76
  77 class BashFeatureTesting(FeatureTestingBase):
  78
  79     def setUp(self):
  80         # logging must be disabled else it calls time.time()
  81         # what will break these unit tests.
  82         logging.disable(logging.CRITICAL)
  83         with mock.patch('six.moves.builtins.open'):
  84             self.feature = feature.BashFeature(
  85                 project_name=self._project_name, case_name=self._case_name)
  86
  87     @mock.patch('subprocess.Popen')
  88     def test_run_no_cmd(self, mock_subproc):
  89         self.assertEqual(
  90             self.feature.run(), testcase.TestCase.EX_RUN_ERROR)
  91         mock_subproc.assert_not_called()
  92
  93     @mock.patch('subprocess.Popen')
  94     def test_run_ko(self, mock_subproc):
  95         with mock.patch('six.moves.builtins.open', mock.mock_open()) as mopen:
  96             mock_obj = mock.Mock()
  97             attrs = {'wait.return_value': 1}
  98             mock_obj.configure_mock(**attrs)
  99
 100             mock_subproc.return_value = mock_obj
 101             self._test_run(testcase.TestCase.EX_RUN_ERROR)
 102             mopen.assert_called_once_with(self._output_file, "w+")
 103
 104     @mock.patch('subprocess.Popen')
 105     def test_run(self, mock_subproc):
 106         with mock.patch('six.moves.builtins.open', mock.mock_open()) as mopen:
 107             mock_obj = mock.Mock()
 108             attrs = {'wait.return_value': 0}
 109             mock_obj.configure_mock(**attrs)
 110
 111             mock_subproc.return_value = mock_obj
 112             self._test_run(testcase.TestCase.EX_OK)
 113             mopen.assert_called_once_with(self._output_file, "w+")
 114
 115
 116 if __name__ == "__main__":
 117     unittest.main(verbosity=2)