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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
"""Tests for tools/render-cloudcfg"""
import sys
import pytest
from cloudinit import subp, templater, util
from tests.unittests.helpers import cloud_init_project_dir
# TODO(Look to align with tools.render-cloudcfg or cloudinit.distos.OSFAMILIES)
DISTRO_VARIANTS = [
"amazon",
"arch",
"centos",
"debian",
"eurolinux",
"fedora",
"freebsd",
"netbsd",
"openbsd",
"photon",
"rhel",
"suse",
"ubuntu",
"unknown",
]
@pytest.mark.allow_subp_for(sys.executable)
class TestRenderCloudCfg:
cmd = [sys.executable, cloud_init_project_dir("tools/render-cloudcfg")]
tmpl_path = cloud_init_project_dir("config/cloud.cfg.tmpl")
def test_variant_sets_distro_in_cloud_cfg_subp(self, tmpdir):
outfile = tmpdir.join("outcfg").strpath
subp.subp(self.cmd + ["--variant", "ubuntu", self.tmpl_path, outfile])
with open(outfile) as stream:
system_cfg = util.load_yaml(stream.read())
assert system_cfg["system_info"]["distro"] == "ubuntu"
@pytest.mark.parametrize("variant", (DISTRO_VARIANTS))
def test_variant_sets_distro_in_cloud_cfg(self, variant, tmpdir):
"""Testing parametrized inputs with imported function saves ~0.5s per
call versus calling as subp
"""
outfile = tmpdir.join("outcfg").strpath
templater.render_cloudcfg(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):
"""Testing parametrized inputs with imported function saves ~0.5s per
call versus calling as subp
"""
outfile = tmpdir.join("outcfg").strpath
templater.render_cloudcfg(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
):
"""Testing parametrized inputs with imported function saves ~0.5s per
call versus calling as subp
"""
outfile = tmpdir.join("outcfg").strpath
templater.render_cloudcfg(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"]
|
