1import json
2
3import pytest
4from units.compat import mock
5from ansible.modules.remote_management.lxca import lxca_cmms
6
7
8@pytest.fixture(scope='module')
9@mock.patch("ansible.module_utils.remote_management.lxca.common.close_conn", autospec=True)
10def setup_module(close_conn):
11    close_conn.return_value = True
12
13
14class TestMyModule():
15    @pytest.mark.parametrize('patch_ansible_module',
16                             [
17                                 {},
18                                 {
19                                     "auth_url": "https://10.240.14.195",
20                                     "login_user": "USERID",
21                                 },
22                                 {
23                                     "auth_url": "https://10.240.14.195",
24                                     "login_password": "Password",
25                                 },
26                                 {
27                                     "login_user": "USERID",
28                                     "login_password": "Password",
29                                 },
30                             ],
31                             indirect=['patch_ansible_module'])
32    @pytest.mark.usefixtures('patch_ansible_module')
33    @mock.patch("ansible.module_utils.remote_management.lxca.common.setup_conn", autospec=True)
34    @mock.patch("ansible.modules.remote_management.lxca.lxca_cmms.execute_module", autospec=True)
35    def test_without_required_parameters(self, _setup_conn, _execute_module,
36                                         mocker, capfd, setup_module):
37        """Failure must occurs when all parameters are missing"""
38        with pytest.raises(SystemExit):
39            _setup_conn.return_value = "Fake connection"
40            _execute_module.return_value = "Fake execution"
41            lxca_cmms.main()
42        out, err = capfd.readouterr()
43        results = json.loads(out)
44        assert results['failed']
45        assert 'missing required arguments' in results['msg']
46
47    @mock.patch("ansible.module_utils.remote_management.lxca.common.setup_conn", autospec=True)
48    @mock.patch("ansible.modules.remote_management.lxca.lxca_cmms.execute_module", autospec=True)
49    @mock.patch("ansible.modules.remote_management.lxca.lxca_cmms.AnsibleModule", autospec=True)
50    def test__argument_spec(self, ansible_mod_cls, _execute_module, _setup_conn, setup_module):
51        expected_arguments_spec = dict(
52            login_user=dict(required=True),
53            login_password=dict(required=True, no_log=True),
54            command_options=dict(default='cmms', choices=['cmms', 'cmms_by_uuid',
55                                                          'cmms_by_chassis_uuid']),
56            auth_url=dict(required=True),
57            uuid=dict(default=None),
58            chassis=dict(default=None),
59        )
60        _setup_conn.return_value = "Fake connection"
61        _execute_module.return_value = []
62        mod_obj = ansible_mod_cls.return_value
63        args = {
64            "auth_url": "https://10.243.30.195",
65            "login_user": "USERID",
66            "login_password": "password",
67            "command_options": "cmms",
68        }
69        mod_obj.params = args
70        lxca_cmms.main()
71        assert(mock.call(argument_spec=expected_arguments_spec,
72                         supports_check_mode=False) == ansible_mod_cls.call_args)
73
74    @mock.patch("ansible.module_utils.remote_management.lxca.common.setup_conn", autospec=True)
75    @mock.patch("ansible.modules.remote_management.lxca.lxca_cmms._cmms_by_uuid",
76                autospec=True)
77    @mock.patch("ansible.modules.remote_management.lxca.lxca_cmms.AnsibleModule",
78                autospec=True)
79    def test__cmms_empty_list(self, ansible_mod_cls, _get_cmms, _setup_conn, setup_module):
80        mod_obj = ansible_mod_cls.return_value
81        args = {
82            "auth_url": "https://10.243.30.195",
83            "login_user": "USERID",
84            "login_password": "password",
85            "uuid": "3C737AA5E31640CE949B10C129A8B01F",
86            "command_options": "cmms_by_uuid",
87        }
88        mod_obj.params = args
89        _setup_conn.return_value = "Fake connection"
90        empty_nodes_list = []
91        _get_cmms.return_value = empty_nodes_list
92        ret_cmms = _get_cmms(mod_obj, args)
93        assert mock.call(mod_obj, mod_obj.params) == _get_cmms.call_args
94        assert _get_cmms.return_value == ret_cmms
95