summaryrefslogtreecommitdiff
path: root/tests/unittests/test_handler/test_handler_spacewalk.py
blob: 44f95e4c945571a0c018a789ca47647673f1e753 (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
from cloudinit.config import cc_spacewalk
from cloudinit import util

from .. import helpers

import logging

try:
    from unittest import mock
except ImportError:
    import mock

LOG = logging.getLogger(__name__)


class TestSpacewalk(helpers.TestCase):
    space_cfg = {
        'spacewalk': {
            'server': 'localhost',
            'profile_name': 'test',
        }
    }

    @mock.patch("cloudinit.config.cc_spacewalk.util.subp")
    def test_not_is_registered(self, mock_util_subp):
        mock_util_subp.side_effect = util.ProcessExecutionError(exit_code=1)
        self.assertFalse(cc_spacewalk.is_registered())

    @mock.patch("cloudinit.config.cc_spacewalk.util.subp")
    def test_is_registered(self, mock_util_subp):
        mock_util_subp.side_effect = None
        self.assertTrue(cc_spacewalk.is_registered())

    @mock.patch("cloudinit.config.cc_spacewalk.util.subp")
    def test_do_register(self, mock_util_subp):
        cc_spacewalk.do_register(**self.space_cfg['spacewalk'])
        mock_util_subp.assert_called_with([
            'rhnreg_ks',
            '--serverUrl', 'https://localhost/XMLRPC',
            '--profilename', 'test',
            '--sslCACert', cc_spacewalk.def_ca_cert_path,
        ], capture=False)