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