# This file is part of cloud-init. See LICENSE file for license information.

from cloudinit.config import cc_spacewalk

from cloudinit import util

from cloudinit.tests 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)

# vi: ts=4 expandtab