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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
|
# pylint: disable=superfluous-parens,ungrouped-imports
from lava_scheduler_app.tests.test_base_templates import (
BaseTemplate,
prepare_jinja_template,
)
# pylint: disable=too-many-branches,too-many-public-methods
# pylint: disable=too-many-nested-blocks
class TestGrubTemplates(BaseTemplate.BaseTemplateCases):
"""
Test rendering of jinja2 templates
When adding or modifying a jinja2 template, add or update the test here.
Use realistic data - complete exports of the device dictionary preferably.
Set debug to True to see the content of the rendered templates
Set system to True to use the system templates - note that this requires
that the templates in question are in sync with the branch upon which the
test is run. Therefore, if the templates should be the same, this can be
used to check that the templates are correct. If there are problems, check
for a template with a .dpkg-dist extension. Check the diff between the
checkout and the system file matches the difference between the system file
and the dpkg-dist version. If the diffs match, copy the dpkg-dist onto the
system file.
"""
def test_mustang_pxe_grub_efi_template(self): # pylint: disable=invalid-name
data = """{% extends 'mustang-grub-efi.jinja2' %}
{% set hard_reset_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command reboot --port 05' %}
{% set power_off_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command off --port 05' %}
{% set power_on_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command on --port 05' %}
{% set connection_command = 'telnet localhost 7012' %}"""
self.assertTrue(self.validate_data('staging-mustang-01', data))
template_dict = prepare_jinja_template('staging-mustang-01', data, raw=False)
self.assertIn('uefi-menu', template_dict['actions']['boot']['methods'])
self.assertIn('pxe-grub', template_dict['actions']['boot']['methods']['uefi-menu'])
self.assertNotIn('grub', template_dict['actions']['boot']['methods']['uefi-menu'])
# label class regex is mangled by jinja/yaml processing
self.assertNotIn('label_class', template_dict['actions']['boot']['methods']['uefi-menu']['parameters'])
self.assertIn('grub-efi', template_dict['actions']['boot']['methods'])
self.assertIn('menu_options', template_dict['actions']['boot']['methods']['grub-efi'])
self.assertEqual(template_dict['actions']['boot']['methods']['grub-efi']['menu_options'], 'pxe-grub')
self.assertIn('ramdisk', template_dict['actions']['boot']['methods']['grub-efi'])
self.assertIn('commands', template_dict['actions']['boot']['methods']['grub-efi']['ramdisk'])
self.assertIn('nfs', template_dict['actions']['boot']['methods']['grub-efi'])
self.assertIn('commands', template_dict['actions']['boot']['methods']['grub-efi']['nfs'])
nfs_commands = template_dict['actions']['boot']['methods']['grub-efi']['nfs']['commands']
self.assertNotIn('insmod efinet', nfs_commands)
self.assertNotIn('net_bootp', nfs_commands)
def test_mustang_grub_efi_template(self):
data = """{% extends 'mustang-grub-efi.jinja2' %}
{% set grub_efi_method = 'grub' %}
{% set hard_reset_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command reboot --port 05' %}
{% set power_off_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command off --port 05' %}
{% set power_on_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command on --port 05' %}
{% set connection_command = 'telnet localhost 7012' %}"""
self.assertTrue(self.validate_data('staging-mustang-01', data))
template_dict = prepare_jinja_template('staging-mustang-01', data, raw=False)
self.assertIn('uefi-menu', template_dict['actions']['boot']['methods'])
self.assertNotIn('pxe-grub', template_dict['actions']['boot']['methods']['uefi-menu'])
self.assertIn('grub', template_dict['actions']['boot']['methods']['uefi-menu'])
self.assertEqual(template_dict['actions']['boot']['methods']['grub-efi']['menu_options'], 'grub')
self.assertIn('ramdisk', template_dict['actions']['boot']['methods']['grub-efi'])
self.assertIn('commands', template_dict['actions']['boot']['methods']['grub-efi']['ramdisk'])
self.assertIn('nfs', template_dict['actions']['boot']['methods']['grub-efi'])
self.assertIn('commands', template_dict['actions']['boot']['methods']['grub-efi']['nfs'])
nfs_commands = template_dict['actions']['boot']['methods']['grub-efi']['nfs']['commands']
self.assertIn('insmod efinet', nfs_commands)
self.assertIn('net_bootp', nfs_commands)
def test_mustang_secondary_media(self):
data = """{% extends 'mustang-grub-efi.jinja2' %}
{% set sata_label = 'ST500DM002' %}
{% set sata_uuid = 'ata-ST500DM002-1BD142_S2AKYFSN' %}
{% set grub_efi_method = 'pxe-grub' %}
{% set hard_reset_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command reboot --port 05' %}
{% set power_off_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command off --port 05' %}
{% set power_on_command = '/usr/bin/pduclient --daemon services --hostname pdu09 --command on --port 05' %}
{% set connection_command = 'telnet localhost 7012' %}"""
self.assertTrue(self.validate_data('staging-mustang-01', data))
template_dict = prepare_jinja_template('staging-mustang-01', data, raw=False)
parameters = {
'parameters': {
'media': {
'sata': {
'ST500DM002': {
'boot_part': 1,
'device_id': 0,
'grub_interface': 'hd0',
'uboot_interface': 'scsi',
'uuid': 'ata-ST500DM002-1BD142_S2AKYFSN'
},
'UUID-required': True
}
}
}
}
self.assertTrue(template_dict['parameters'] == parameters['parameters'])
self.assertIn('sata', template_dict['actions']['boot']['methods']['grub-efi'])
commands = {
'commands': [
'insmod gzio',
'linux (hd0,gpt1)/{KERNEL} console=ttyS0,115200n8 debug root=/dev/sda2 rw ip=:::::eth0:dhcp',
'initrd (hd0,gpt1/{RAMDISK}',
'boot']}
self.assertEqual(
commands,
template_dict['actions']['boot']['methods']['grub-efi']['sata']
)
def test_overdrive_template(self):
data = """{% extends 'overdrive.jinja2' %}
{% set connection_command = 'telnet serial4 7001' %}
{% set hard_reset_command = '/usr/local/lab-scripts/snmp_pdu_control --hostname pdu18 --command reboot --port 10 --delay 10' %}
{% set power_off_command = '/usr/local/lab-scripts/snmp_pdu_control --hostname pdu18 --command off --port 10 --delay 10' %}
{% set power_on_command = '/usr/local/lab-scripts/snmp_pdu_control --hostname pdu18 --command on --port 10 --delay 10' %}
{% set map = {'iface0': {'lngswitch03': 13}, 'iface1': {'lngswitch03': 1}, 'iface2': {'lngswitch02': 9}, 'iface3': {'lngswitch02': 10}} %}
{% set tags = {'iface0': [], 'iface1': ['RJ45', '1G', '10G'], 'iface2': ['SFP+', '1G', '10G'], 'iface3': ['SFP+', '1G', '10G']} %}
{% set mac_addr = {'iface0': '00:00:1a:1b:8b:f6', 'iface1': '00:00:1a:1b:8b:f7', 'iface2': '00:11:0a:68:94:30', 'iface3': '00:11:0a:68:94:31'} %}
{% set interfaces = ['iface0', 'iface1', 'iface2', 'iface3'] %}
{% set sysfs = {'iface0': '/sys/devices/platform/AMDI8001:00/net/',
'iface1': '/sys/devices/platform/AMDI8001:01/net/',
'iface2': '/sys/devices/pci0000:00/0000:00:02.1/0000:01:00.0/net/',
'iface3': '/sys/devices/pci0000:00/0000:00:02.1/0000:01:00.1/net/'} %}
{% set boot_character_delay = 100 %}"""
self.assertTrue(self.validate_data('staging-overdrive-01', data))
template_dict = prepare_jinja_template('staging-overdrive-01', data, raw=False)
self.assertIsNotNone(template_dict)
self.assertIn('parameters', template_dict)
self.assertIn('interfaces', template_dict['parameters'])
self.assertIn('actions', template_dict)
self.assertIn('character_delays', template_dict)
self.assertIn('boot', template_dict['character_delays'])
self.assertEqual(100, template_dict['character_delays']['boot'])
self.assertIn('iface2', template_dict['parameters']['interfaces'])
self.assertIn('iface1', template_dict['parameters']['interfaces'])
self.assertIn('iface0', template_dict['parameters']['interfaces'])
self.assertIn('sysfs', template_dict['parameters']['interfaces']['iface2'])
self.assertEqual(
[check for check in template_dict['actions']['boot']['methods']['grub']['nfs']['commands'] if 'nfsroot' in check][0].count('nfsroot'),
1
)
self.assertIn(
' rw',
[check for check in template_dict['actions']['boot']['methods']['grub']['nfs']['commands'] if 'nfsroot' in check][0]
)
def test_synquacer_acpi_template(self):
template_dict = self.render_device_dictionary_file('synquacer-acpi-01.jinja2', raw=False)
self.assertIsNotNone(template_dict['actions']['boot']['methods']['grub'])
self.assertIn('ramdisk', template_dict['actions']['boot']['methods']['grub'])
self.assertIn('commands', template_dict['actions']['boot']['methods']['grub']['ramdisk'])
self.assertIn(
'net_add_addr lava efinet0 192.168.25.43',
template_dict['actions']['boot']['methods']['grub']['ramdisk']['commands']
)
self.assertIn(
'linux (tftp,{SERVER_IP})/{KERNEL} ip=192.168.25.43::192.168.25.1:255.255.255.0:::off:192.168.25.1: ',
template_dict['actions']['boot']['methods']['grub']['ramdisk']['commands']
)
def test_synquacer_dtb_template(self):
template_dict = self.render_device_dictionary_file('synquacer-dtb-01.jinja2', raw=False)
self.assertIsNotNone(template_dict['actions']['boot']['methods']['grub'])
self.assertIn('ramdisk', template_dict['actions']['boot']['methods']['grub'])
self.assertIn('commands', template_dict['actions']['boot']['methods']['grub']['ramdisk'])
self.assertIn(
'net_add_addr lava efinet0 192.168.25.42',
template_dict['actions']['boot']['methods']['grub']['ramdisk']['commands']
)
self.assertIn(
'linux (tftp,{SERVER_IP})/{KERNEL} ip=192.168.25.42::192.168.25.1:255.255.255.0:::off:192.168.25.1: ',
template_dict['actions']['boot']['methods']['grub']['ramdisk']['commands']
)
def test_moonshot_m400_template(self):
template_dict = self.render_device_dictionary_file('moonshot-m400-17.jinja2', raw=False)
self.assertIsNotNone(template_dict['actions']['boot']['methods']['grub'])
self.assertIn('ramdisk', template_dict['actions']['boot']['methods']['grub'])
self.assertIn('commands', template_dict['actions']['boot']['methods']['grub']['ramdisk'])
self.assertIn(
'linux (tftp,{SERVER_IP})/{KERNEL} console=ttyS0,9600 ip=dhcp ',
template_dict['actions']['boot']['methods']['grub']['ramdisk']['commands']
)
def test_minnowboard_turbot_template(self):
template_dict = self.render_device_dictionary_file('minnowboard-turbot-E3826-01.jinja2', raw=False)
grub = template_dict['actions']['boot']['methods']['grub']
self.assertIsNotNone(grub)
self.assertIn('ramdisk', grub)
self.assertIn('commands', grub['ramdisk'])
ramdisk_commands = grub['ramdisk']['commands']
ramdisk_ref_commands = [
'set net_default_server={SERVER_IP}',
'linux (tftp)/{KERNEL} console=tty0 console=ttyS0,115200 root=/dev/ram0 ip=:::::eth0:dhcp',
'initrd (tftp)/{RAMDISK}',
'boot',
]
self.assertEqual(ramdisk_commands, ramdisk_ref_commands)
self.assertIn('nfs', grub)
self.assertIn('commands', grub['nfs'])
nfs_commands = grub['nfs']['commands']
nfs_ref_commands = [
'set net_default_server={SERVER_IP}',
'linux (tftp)/{KERNEL} console=tty0 console=ttyS0,115200 root=/dev/nfs rw nfsroot={NFS_SERVER_IP}:{NFSROOTFS},tcp,hard,intr ip=dhcp',
'boot',
]
self.assertEqual(nfs_commands, nfs_ref_commands)
|
