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

from cloudinit.config import cc_puppet

from cloudinit.sources import DataSourceNone

from cloudinit import (distros, helpers, cloud, util)

from cloudinit.tests.helpers import CiTestCase, mock

import logging

import textwrap

LOG = logging.getLogger(__name__)

@mock.patch('cloudinit.config.cc_puppet.util')

@mock.patch('cloudinit.config.cc_puppet.os')

class TestAutostartPuppet(CiTestCase):

    with_logs = True

    def test_wb_autostart_puppet_updates_puppet_default(self, m_os, m_util):

        """Update /etc/default/puppet to autostart if it exists."""

        def _fake_exists(path):

            return path == '/etc/default/puppet'

        m_os.path.exists.side_effect = _fake_exists

        cc_puppet._autostart_puppet(LOG)

        self.assertEqual(

            [mock.call(['sed', '-i', '-e', 's/^START=.*/START=yes/',

                        '/etc/default/puppet'], capture=False)],

            m_util.subp.call_args_list)

    def test_wb_autostart_pupppet_enables_puppet_systemctl(self, m_os, m_util):

        """If systemctl is present, enable puppet via systemctl."""

        def _fake_exists(path):

            return path == '/bin/systemctl'

        m_os.path.exists.side_effect = _fake_exists

        cc_puppet._autostart_puppet(LOG)

        expected_calls = [mock.call(

            ['/bin/systemctl', 'enable', 'puppet.service'], capture=False)]

        self.assertEqual(expected_calls, m_util.subp.call_args_list)

    def test_wb_autostart_pupppet_enables_puppet_chkconfig(self, m_os, m_util):

        """If chkconfig is present, enable puppet via checkcfg."""

        def _fake_exists(path):

            return path == '/sbin/chkconfig'

        m_os.path.exists.side_effect = _fake_exists

        cc_puppet._autostart_puppet(LOG)

        expected_calls = [mock.call(

            ['/sbin/chkconfig', 'puppet', 'on'], capture=False)]

        self.assertEqual(expected_calls, m_util.subp.call_args_list)

@mock.patch('cloudinit.config.cc_puppet._autostart_puppet')

class TestPuppetHandle(CiTestCase):

    with_logs = True

    def setUp(self):

        super(TestPuppetHandle, self).setUp()

        self.new_root = self.tmp_dir()

        self.conf = self.tmp_path('puppet.conf')

        self.csr_attributes_path = self.tmp_path('csr_attributes.yaml')

    def _get_cloud(self, distro):

        paths = helpers.Paths({'templates_dir': self.new_root})

        cls = distros.fetch(distro)

        mydist = cls(distro, {}, paths)

        myds = DataSourceNone.DataSourceNone({}, mydist, paths)

        return cloud.Cloud(myds, paths, {}, mydist, None)

    def test_handler_skips_missing_puppet_key_in_cloudconfig(self, m_auto):

        """Cloud-config containing no 'puppet' key is skipped."""

        mycloud = self._get_cloud('ubuntu')

        cfg = {}

        cc_puppet.handle('notimportant', cfg, mycloud, LOG, None)

        self.assertIn(

            "no 'puppet' configuration found", self.logs.getvalue())

        self.assertEqual(0, m_auto.call_count)

    @mock.patch('cloudinit.config.cc_puppet.util.subp')

    def test_handler_puppet_config_starts_puppet_service(self, m_subp, m_auto):

        """Cloud-config 'puppet' configuration starts puppet."""

        mycloud = self._get_cloud('ubuntu')

        cfg = {'puppet': {'install': False}}

        cc_puppet.handle('notimportant', cfg, mycloud, LOG, None)

        self.assertEqual(1, m_auto.call_count)

        self.assertEqual(

            [mock.call(['service', 'puppet', 'start'], capture=False)],

            m_subp.call_args_list)

    @mock.patch('cloudinit.config.cc_puppet.util.subp')

    def test_handler_empty_puppet_config_installs_puppet(self, m_subp, m_auto):

        """Cloud-config empty 'puppet' configuration installs latest puppet."""

        mycloud = self._get_cloud('ubuntu')

        mycloud.distro = mock.MagicMock()

        cfg = {'puppet': {}}

        cc_puppet.handle('notimportant', cfg, mycloud, LOG, None)

        self.assertEqual(

            [mock.call(('puppet', None))],

            mycloud.distro.install_packages.call_args_list)

    @mock.patch('cloudinit.config.cc_puppet.util.subp')

    def test_handler_puppet_config_installs_puppet_on_true(self, m_subp, _):

        """Cloud-config with 'puppet' key installs when 'install' is True."""

        mycloud = self._get_cloud('ubuntu')

        mycloud.distro = mock.MagicMock()

        cfg = {'puppet': {'install': True}}

        cc_puppet.handle('notimportant', cfg, mycloud, LOG, None)

        self.assertEqual(

            [mock.call(('puppet', None))],

            mycloud.distro.install_packages.call_args_list)

    @mock.patch('cloudinit.config.cc_puppet.util.subp')

    def test_handler_puppet_config_installs_puppet_version(self, m_subp, _):

        """Cloud-config 'puppet' configuration can specify a version."""

        mycloud = self._get_cloud('ubuntu')

        mycloud.distro = mock.MagicMock()

        cfg = {'puppet': {'version': '3.8'}}

        cc_puppet.handle('notimportant', cfg, mycloud, LOG, None)

        self.assertEqual(

            [mock.call(('puppet', '3.8'))],

            mycloud.distro.install_packages.call_args_list)

    @mock.patch('cloudinit.config.cc_puppet.util.subp')

    def test_handler_puppet_config_updates_puppet_conf(self, m_subp, m_auto):

        """When 'conf' is provided update values in PUPPET_CONF_PATH."""

        mycloud = self._get_cloud('ubuntu')

        cfg = {

            'puppet': {

                'conf': {'agent': {'server': 'puppetmaster.example.org'}}}}

        util.write_file(self.conf, '[agent]\nserver = origpuppet\nother = 3')

        puppet_conf_path = 'cloudinit.config.cc_puppet.PUPPET_CONF_PATH'

        mycloud.distro = mock.MagicMock()

        with mock.patch(puppet_conf_path, self.conf):

            cc_puppet.handle('notimportant', cfg, mycloud, LOG, None)

        content = util.load_file(self.conf)

        expected = '[agent]\nserver = puppetmaster.example.org\nother = 3\n\n'

        self.assertEqual(expected, content)

    @mock.patch('cloudinit.config.cc_puppet.util.subp')

    def test_handler_puppet_writes_csr_attributes_file(self, m_subp, m_auto):

        """When csr_attributes is provided

            creates file in PUPPET_CSR_ATTRIBUTES_PATH."""

        mycloud = self._get_cloud('ubuntu')

        mycloud.distro = mock.MagicMock()

        cfg = {

            'puppet': {

              'csr_attributes': {

                'custom_attributes': {

                  '1.2.840.113549.1.9.7': '342thbjkt82094y0ut'

                                          'hhor289jnqthpc2290'},

                'extension_requests': {

                  'pp_uuid': 'ED803750-E3C7-44F5-BB08-41A04433FE2E',

                  'pp_image_name': 'my_ami_image',

                  'pp_preshared_key': '342thbjkt82094y0uthhor289jnqthpc2290'}

                }}}

        csr_attributes = 'cloudinit.config.cc_puppet.' \

                         'PUPPET_CSR_ATTRIBUTES_PATH'

        with mock.patch(csr_attributes, self.csr_attributes_path):

            cc_puppet.handle('notimportant', cfg, mycloud, LOG, None)

        content = util.load_file(self.csr_attributes_path)

        expected = textwrap.dedent("""\

            custom_attributes:

              1.2.840.113549.1.9.7: 342thbjkt82094y0uthhor289jnqthpc2290

            extension_requests:

              pp_image_name: my_ami_image

              pp_preshared_key: 342thbjkt82094y0uthhor289jnqthpc2290

              pp_uuid: ED803750-E3C7-44F5-BB08-41A04433FE2E

            """)

        self.assertEqual(expected, content)