Blame tests/unittests/test_handler/test_handler_spacewalk.py

Packit Service a04d08
# This file is part of cloud-init. See LICENSE file for license information.
Packit Service a04d08
Packit Service a04d08
from cloudinit.config import cc_spacewalk
Packit Service 751c4a
from cloudinit import subp
Packit Service a04d08
Packit Service a04d08
from cloudinit.tests import helpers
Packit Service a04d08
Packit Service a04d08
import logging
Packit Service 751c4a
from unittest import mock
Packit Service a04d08
Packit Service a04d08
LOG = logging.getLogger(__name__)
Packit Service a04d08
Packit Service a04d08
Packit Service a04d08
class TestSpacewalk(helpers.TestCase):
Packit Service a04d08
    space_cfg = {
Packit Service a04d08
        'spacewalk': {
Packit Service a04d08
            'server': 'localhost',
Packit Service a04d08
            'profile_name': 'test',
Packit Service a04d08
        }
Packit Service a04d08
    }
Packit Service a04d08
Packit Service 751c4a
    @mock.patch("cloudinit.config.cc_spacewalk.subp.subp")
Packit Service 751c4a
    def test_not_is_registered(self, mock_subp):
Packit Service 751c4a
        mock_subp.side_effect = subp.ProcessExecutionError(exit_code=1)
Packit Service a04d08
        self.assertFalse(cc_spacewalk.is_registered())
Packit Service a04d08
Packit Service 751c4a
    @mock.patch("cloudinit.config.cc_spacewalk.subp.subp")
Packit Service 751c4a
    def test_is_registered(self, mock_subp):
Packit Service 751c4a
        mock_subp.side_effect = None
Packit Service a04d08
        self.assertTrue(cc_spacewalk.is_registered())
Packit Service a04d08
Packit Service 751c4a
    @mock.patch("cloudinit.config.cc_spacewalk.subp.subp")
Packit Service 751c4a
    def test_do_register(self, mock_subp):
Packit Service a04d08
        cc_spacewalk.do_register(**self.space_cfg['spacewalk'])
Packit Service 751c4a
        mock_subp.assert_called_with([
Packit Service a04d08
            'rhnreg_ks',
Packit Service a04d08
            '--serverUrl', 'https://localhost/XMLRPC',
Packit Service a04d08
            '--profilename', 'test',
Packit Service a04d08
            '--sslCACert', cc_spacewalk.def_ca_cert_path,
Packit Service a04d08
        ], capture=False)
Packit Service a04d08
Packit Service a04d08
# vi: ts=4 expandtab