"""Tests for tools/render-cloudcfg"""

import os

import sys

import pytest

from cloudinit import subp

from cloudinit import util

# TODO(Look to align with tools.render-cloudcfg or cloudinit.distos.OSFAMILIES)

DISTRO_VARIANTS = ["amazon", "arch", "centos", "debian", "fedora", "freebsd",

                   "netbsd", "openbsd", "rhel", "suse", "ubuntu", "unknown"]

@pytest.mark.allow_subp_for(sys.executable)

class TestRenderCloudCfg:

    cmd = [sys.executable, os.path.realpath('tools/render-cloudcfg')]

    tmpl_path = os.path.realpath('config/cloud.cfg.tmpl')

    @pytest.mark.parametrize('variant', (DISTRO_VARIANTS))

    def test_variant_sets_distro_in_cloud_cfg(self, variant, tmpdir):

        outfile = tmpdir.join('outcfg').strpath

        subp.subp(

            self.cmd + ['--variant', variant, self.tmpl_path, outfile])

        with open(outfile) as stream:

            system_cfg = util.load_yaml(stream.read())

        if variant == 'unknown':

            variant = 'ubuntu'  # Unknown is defaulted to ubuntu

        assert system_cfg['system_info']['distro'] == variant

    @pytest.mark.parametrize('variant', (DISTRO_VARIANTS))

    def test_variant_sets_default_user_in_cloud_cfg(self, variant, tmpdir):

        outfile = tmpdir.join('outcfg').strpath

        subp.subp(

            self.cmd + ['--variant', variant, self.tmpl_path, outfile])

        with open(outfile) as stream:

            system_cfg = util.load_yaml(stream.read())

        default_user_exceptions = {

            'amazon': 'ec2-user', 'debian': 'ubuntu', 'unknown': 'ubuntu'}

        default_user = system_cfg['system_info']['default_user']['name']

        assert default_user == default_user_exceptions.get(variant, variant)

    @pytest.mark.parametrize('variant,renderers', (

        ('freebsd', ['freebsd']), ('netbsd', ['netbsd']),

        ('openbsd', ['openbsd']), ('ubuntu', ['netplan', 'eni', 'sysconfig']))

    )

    def test_variant_sets_network_renderer_priority_in_cloud_cfg(

        self, variant, renderers, tmpdir

    ):

        outfile = tmpdir.join('outcfg').strpath

        subp.subp(

            self.cmd + ['--variant', variant, self.tmpl_path, outfile])

        with open(outfile) as stream:

            system_cfg = util.load_yaml(stream.read())

        assert renderers == system_cfg['system_info']['network']['renderers']