Modify how to disable logging in unit test.
[functest.git] / functest / tests / unit / cli / commands / test_cli_tier.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
9 import logging
10 import unittest
11
12 import mock
13
14 from functest.cli.commands import cli_tier
15 from functest.utils.constants import CONST
16
17
18 class CliTierTesting(unittest.TestCase):
19
20     def setUp(self):
21         self.tiername = 'tiername'
22         self.testnames = 'testnames'
23         with mock.patch('functest.cli.commands.cli_tier.tb'):
24             self.cli_tier = cli_tier.CliTier()
25
26     @mock.patch('functest.cli.commands.cli_tier.click.echo')
27     def test_list(self, mock_click_echo):
28         with mock.patch.object(self.cli_tier.tiers, 'get_tiers',
29                                return_value=[]):
30             self.cli_tier.list()
31             mock_click_echo.assert_called_with("")
32
33     @mock.patch('functest.cli.commands.cli_tier.click.echo')
34     def test_show_default(self, mock_click_echo):
35         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
36                                return_value=self.tiername):
37             self.cli_tier.show(self.tiername)
38             mock_click_echo.assert_called_with(self.tiername)
39
40     @mock.patch('functest.cli.commands.cli_tier.click.echo')
41     def test_show_missing_tier(self, mock_click_echo):
42         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
43                                return_value=None), \
44             mock.patch.object(self.cli_tier.tiers, 'get_tier_names',
45                               return_value='tiernames'):
46             self.cli_tier.show(self.tiername)
47             mock_click_echo.assert_called_with("The tier with name '%s' does "
48                                                "not exist. Available tiers are"
49                                                ":\n  %s\n" % (self.tiername,
50                                                               'tiernames'))
51
52     @mock.patch('functest.cli.commands.cli_tier.click.echo')
53     def test_gettests_default(self, mock_click_echo):
54         mock_obj = mock.Mock()
55         attrs = {'get_test_names.return_value': self.testnames}
56         mock_obj.configure_mock(**attrs)
57
58         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
59                                return_value=mock_obj):
60             self.cli_tier.gettests(self.tiername)
61             mock_click_echo.assert_called_with("Test cases in tier "
62                                                "'%s':\n %s\n" % (self.tiername,
63                                                                  self.testnames
64                                                                  ))
65
66     @mock.patch('functest.cli.commands.cli_tier.click.echo')
67     def test_gettests_missing_tier(self, mock_click_echo):
68         with mock.patch.object(self.cli_tier.tiers, 'get_tier',
69                                return_value=None), \
70             mock.patch.object(self.cli_tier.tiers, 'get_tier_names',
71                               return_value='tiernames'):
72             self.cli_tier.gettests(self.tiername)
73             mock_click_echo.assert_called_with("The tier with name '%s' does "
74                                                "not exist. Available tiers are"
75                                                ":\n  %s\n" % (self.tiername,
76                                                               'tiernames'))
77
78     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
79                 return_value=False)
80     @mock.patch('functest.cli.commands.cli_tier.click.echo')
81     def test_run_missing_env_file(self, mock_click_echo, mock_os):
82         self.cli_tier.run(self.tiername)
83         mock_click_echo.assert_called_with("Functest environment is not ready."
84                                            " Run first 'functest env prepare'")
85
86     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
87                 return_value=True)
88     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
89     def test_run_default(self, mock_ft_utils, mock_os):
90         cmd = ("python %s/functest/ci/run_tests.py "
91                "%s -t %s" % (CONST.dir_repo_functest, "-n -r ",
92                              self.tiername))
93         self.cli_tier.run(self.tiername, noclean=True, report=True)
94         mock_ft_utils.assert_called_with(cmd)
95
96     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
97                 return_value=True)
98     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
99     def test_run_report_missing_noclean(self, mock_ft_utils, mock_os):
100         cmd = ("python %s/functest/ci/run_tests.py "
101                "%s -t %s" % (CONST.dir_repo_functest, "-r ",
102                              self.tiername))
103         self.cli_tier.run(self.tiername, noclean=False, report=True)
104         mock_ft_utils.assert_called_with(cmd)
105
106     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
107                 return_value=True)
108     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
109     def test_run_noclean_missing_report(self, mock_ft_utils, mock_os):
110         cmd = ("python %s/functest/ci/run_tests.py "
111                "%s -t %s" % (CONST.dir_repo_functest, "-n ",
112                              self.tiername))
113         self.cli_tier.run(self.tiername, noclean=True, report=False)
114         mock_ft_utils.assert_called_with(cmd)
115
116     @mock.patch('functest.cli.commands.cli_tier.os.path.isfile',
117                 return_value=True)
118     @mock.patch('functest.cli.commands.cli_tier.ft_utils.execute_command')
119     def test_run_missing_noclean_report(self, mock_ft_utils, mock_os):
120         cmd = ("python %s/functest/ci/run_tests.py "
121                "%s -t %s" % (CONST.dir_repo_functest, "",
122                              self.tiername))
123         self.cli_tier.run(self.tiername, noclean=False, report=False)
124         mock_ft_utils.assert_called_with(cmd)
125
126
127 if __name__ == "__main__":
128     logging.disable(logging.CRITICAL)
129     unittest.main(verbosity=2)