aboutsummaryrefslogtreecommitdiffstats
path: root/functest/tests/unit/cli/commands/test_cli_env.py
blob: 14e926eb9f2682a75cfd3fe4f6ab393a3a43b74c (plain)
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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#!/usr/bin/env python

# 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

import logging
import unittest

from git.exc import NoSuchPathError
import mock

from functest.cli.commands import cli_env
from functest.utils.constants import CONST
from functest.tests.unit import test_utils


class CliEnvTesting(unittest.TestCase):

    def setUp(self):
        self.cli_environ = cli_env.CliEnv()

    @mock.patch('functest.cli.commands.cli_testcase.os.path.isfile',
                return_value=False)
    @mock.patch('functest.cli.commands.cli_testcase.ft_utils.execute_command')
    def test_prepare_default(self, mock_ft_utils, mock_os):
        cmd = ("python %s/functest/ci/prepare_env.py start" %
               CONST.__getattribute__('dir_repo_functest'))
        self.cli_environ.prepare()
        mock_ft_utils.assert_called_with(cmd)

    @mock.patch('functest.cli.commands.cli_testcase.os.path.isfile',
                return_value=True)
    @mock.patch('functest.cli.commands.cli_testcase.ft_utils.execute_command')
    def test_prepare_missing_status(self, mock_ft_utils, mock_os):
        with mock.patch('__builtin__.raw_input', return_value="y"), \
                mock.patch('functest.cli.commands.cli_testcase.os.remove') \
                as mock_os_remove:
            cmd = ("python %s/functest/ci/prepare_env.py start" %
                   CONST.__getattribute__('dir_repo_functest'))
            self.cli_environ.prepare()
            mock_os_remove.assert_called_once_with(
                CONST.__getattribute__('env_active'))
            mock_ft_utils.assert_called_with(cmd)

    def _test_show_missing_env_var(self, var, *args):
        if var == 'INSTALLER_TYPE':
            CONST.__setattr__('INSTALLER_TYPE', None)
            reg_string = "|  INSTALLER: Unknown, \S+\s*|"
        elif var == 'INSTALLER_IP':
            CONST.__setattr__('INSTALLER_IP', None)
            reg_string = "|  INSTALLER: \S+, Unknown\s*|"
        elif var == 'SCENARIO':
            CONST.__setattr__('DEPLOY_SCENARIO', None)
            reg_string = "|   SCENARIO: Unknown\s*|"
        elif var == 'NODE':
            CONST.__setattr__('NODE_NAME', None)
            reg_string = "|        POD: Unknown\s*|"
        elif var == 'BUILD_TAG':
            CONST.__setattr__('BUILD_TAG', None)
            reg_string = "|  BUILD TAG: None|"
        elif var == 'DEBUG':
            CONST.__setattr__('CI_DEBUG', None)
            reg_string = "| DEBUG FLAG: false\s*|"
        elif var == 'STATUS':
            reg_string = "|     STATUS: not ready\s*|"

        with mock.patch('functest.cli.commands.cli_env.click.echo') \
                as mock_click_echo:
            self.cli_environ.show()
            mock_click_echo.assert_called_with(test_utils.
                                               RegexMatch(reg_string))

    @mock.patch('functest.cli.commands.cli_env.git.Repo')
    def test_show_missing_ci_installer_type(self, *args):
        self._test_show_missing_env_var('INSTALLER_TYPE', *args)

    @mock.patch('functest.cli.commands.cli_env.git.Repo')
    def test_show_missing_ci_installer_ip(self, *args):
        self._test_show_missing_env_var('INSTALLER_IP', *args)

    @mock.patch('functest.cli.commands.cli_env.git.Repo')
    def test_show_missing_ci_scenario(self, *args):
        self._test_show_missing_env_var('SCENARIO', *args)

    @mock.patch('functest.cli.commands.cli_env.git.Repo')
    def test_show_missing_ci_node(self, *args):
        self._test_show_missing_env_var('NODE', *args)

    @mock.patch('functest.cli.commands.cli_env.git.Repo')
    def test_show_missing_ci_build_tag(self, *args):
        self._test_show_missing_env_var('BUILD_TAG', *args)

    @mock.patch('functest.cli.commands.cli_env.git.Repo')
    def test_show_missing_ci_debug(self, *args):
        self._test_show_missing_env_var('DEBUG', *args)

    @mock.patch('functest.cli.commands.cli_env.git.Repo')
    @mock.patch('functest.cli.commands.cli_env.os.path.isfile',
                return_value=False)
    def test_show_missing_environment(self, *args):
        self._test_show_missing_env_var('STATUS', *args)

    @mock.patch('functest.cli.commands.cli_env.os.path.exists',
                return_value=False)
    def test_show_missing_git_repo_dir(self, *args):
        CONST.__setattr__('dir_repo_functest', None)
        self.assertRaises(NoSuchPathError, lambda: self.cli_environ.show())

    @mock.patch('functest.cli.commands.cli_env.click.echo')
    @mock.patch('functest.cli.commands.cli_env.os.path.isfile',
                return_value=True)
    def test_status_environment_present(self, mock_path, mock_click_echo):
        self.assertEqual(self.cli_environ.status(), 0)
        mock_click_echo.assert_called_with("Functest environment"
                                           " ready to run tests.\n")

    @mock.patch('functest.cli.commands.cli_env.click.echo')
    @mock.patch('functest.cli.commands.cli_env.os.path.isfile',
                return_value=False)
    def test_status_environment_absent(self, mock_path, mock_click_echo):
        self.assertEqual(self.cli_environ.status(), 1)
        mock_click_echo.assert_called_with("Functest environment"
                                           " is not installed.\n")


if __name__ == "__main__":
    logging.disable(logging.CRITICAL)
    unittest.main(verbosity=2)