Merge "Modify how to disable logging in unit test."
[functest.git] / functest / tests / unit / openstack / refstack_client / test_refstack_client.py
1 #!/usr/bin/env python
2
3 # All rights reserved. This program and the accompanying materials
4 # are made available under the terms of the Apache License, Version 2.0
5 # which accompanies this distribution, and is available at
6 # http://www.apache.org/licenses/LICENSE-2.0
7
8 import logging
9 import mock
10 import os
11 import unittest
12
13 from functest.core import testcase
14 from functest.opnfv_tests.openstack.refstack_client import refstack_client
15 from functest.utils.constants import CONST
16
17
18 class OSRefstackClientTesting(unittest.TestCase):
19
20     _config = os.path.join(CONST.dir_functest_test,
21                            CONST.refstack_tempest_conf_path)
22     _testlist = os.path.join(CONST.dir_functest_test,
23                              CONST.refstack_defcore_list)
24
25     def setUp(self):
26         self.defaultargs = {'config': self._config,
27                             'testlist': self._testlist}
28         self.refstackclient = refstack_client.RefstackClient()
29
30     def test_source_venv(self):
31         CONST.dir_refstack_client = 'test_repo_dir'
32         with mock.patch('functest.opnfv_tests.openstack.refstack_client.'
33                         'refstack_client.ft_utils.execute_command') as m:
34             cmd = ("cd {0};"
35                    ". .venv/bin/activate;"
36                    "cd -;".format(CONST.dir_refstack_client))
37             self.refstackclient.source_venv()
38             m.assert_any_call(cmd)
39
40     def test_run_defcore(self):
41         config = 'tempest.conf'
42         testlist = 'testlist'
43         with mock.patch('functest.opnfv_tests.openstack.refstack_client.'
44                         'refstack_client.ft_utils.execute_command') as m:
45             cmd = ("cd {0};"
46                    "./refstack-client test -c {1} -v --test-list {2};"
47                    "cd -;".format(CONST.dir_refstack_client,
48                                   config,
49                                   testlist))
50             self.refstackclient.run_defcore(config, testlist)
51             m.assert_any_call(cmd)
52
53     def _get_main_kwargs(self, key=None):
54         kwargs = {'config': self._config,
55                   'testlist': self._testlist}
56         if key:
57             del kwargs[key]
58         return kwargs
59
60     def _test_main(self, status, *args):
61         kwargs = self._get_main_kwargs()
62         self.assertEqual(self.refstackclient.main(**kwargs), status)
63         if len(args) > 0:
64             args[0].assert_called_once_with(
65                  refstack_client.RefstackClient.result_dir)
66         if len(args) > 1:
67             args
68
69     def _test_main_missing_keyword(self, key):
70         kwargs = self._get_main_kwargs(key)
71         self.assertEqual(self.refstackclient.main(**kwargs),
72                          testcase.TestCase.EX_RUN_ERROR)
73
74     def test_main_missing_conf(self):
75         self._test_main_missing_keyword('config')
76
77     def test_main_missing_testlist(self):
78         self._test_main_missing_keyword('testlist')
79
80     def _test_argparser(self, arg, value):
81         self.defaultargs[arg] = value
82         parser = refstack_client.RefstackClientParser()
83         self.assertEqual(parser.parse_args(["--{}={}".format(arg, value)]),
84                          self.defaultargs)
85
86     def test_argparser_conf(self):
87         self._test_argparser('config', self._config)
88
89     def test_argparser_testlist(self):
90         self._test_argparser('testlist', self._testlist)
91
92     def test_argparser_multiple_args(self):
93         self.defaultargs['config'] = self._config
94         self.defaultargs['testlist'] = self._testlist
95         parser = refstack_client.RefstackClientParser()
96         self.assertEqual(parser.parse_args(
97             ["--config={}".format(self._config),
98              "--testlist={}".format(self._testlist)
99              ]), self.defaultargs)
100
101
102 if __name__ == "__main__":
103     logging.disable(logging.CRITICAL)
104     unittest.main(verbosity=2)