Skip to content

Commit b03c5d8

Browse files
sammjsammj
committed
testcases/PetitbootConfig: Test Petitboot Configuration UI
A few tests to navigate around the System Configuration screen and check that settings are properly saved to NVRAM. Signed-off-by: Samuel Mendoza-Jonas <[email protected]>
1 parent dee0ad5 commit b03c5d8

File tree

1 file changed

+351
-0
lines changed

1 file changed

+351
-0
lines changed

‎testcases/PetitbootConfig.py

+351
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,351 @@
1+
#!/usr/bin/env python2
2+
3+
import time
4+
import unittest
5+
6+
import OpTestConfiguration
7+
from common.OpTestUtil import OpTestUtil
8+
from common.OpTestSystem import OpSystemState
9+
from common.OpTestError import OpTestError
10+
from common.OpTestKeys import OpTestKeys as keys
11+
12+
class ConfigEditorTestCase(unittest.TestCase):
13+
def setUp(self):
14+
conf = OpTestConfiguration.conf
15+
self.system = conf.system()
16+
self.bmc = conf.bmc()
17+
self.console = self.system.console
18+
19+
self.system.goto_state(OpSystemState.PETITBOOT)
20+
21+
# Let Petitboot discovery settle
22+
time.sleep(10)
23+
24+
def testConfigAvailable(self):
25+
c = self.console.get_console()
26+
c.send('c')
27+
c.expect('Petitboot System Configuration')
28+
29+
c.send('x')
30+
31+
c.expect("e=edit, n=new")
32+
33+
class TimeoutConfigTestCase(unittest.TestCase):
34+
default_timeout = 10
35+
new_timeout = 314
36+
37+
def setUp(self):
38+
conf = OpTestConfiguration.conf
39+
self.system = conf.system()
40+
self.system.goto_state(OpSystemState.PETITBOOT)
41+
self.console = self.system.console
42+
43+
# Let Petitboot discovery settle
44+
time.sleep(10)
45+
46+
def setConfigTimeout(self, val):
47+
c = self.console.get_console()
48+
c.send('c')
49+
c.expect('Petitboot System Configuration')
50+
51+
# navigate to timeout field
52+
c.send(keys.TAB)
53+
c.send(keys.TAB)
54+
c.send(keys.TAB)
55+
c.send(keys.TAB)
56+
c.send(keys.TAB)
57+
58+
# clear field & insert new value
59+
c.send('\x7f\x7f\x7f\x7f\x7f')
60+
c.send(str(val))
61+
62+
# save
63+
c.send(keys.PGDOWN)
64+
c.send(keys.BTAB)
65+
c.send(keys.BTAB)
66+
c.send(keys.ENTER)
67+
68+
c.expect("e=edit, n=new")
69+
70+
def testConfigTimeoutDisplay(self):
71+
c = self.console.get_console()
72+
self.setConfigTimeout(self.new_timeout)
73+
74+
# wait for UI to settle after update
75+
time.sleep(10)
76+
77+
c.send('c')
78+
# expect our new timeout value
79+
c.expect(str(self.new_timeout))
80+
81+
c.send('x')
82+
c.expect("e=edit, n=new")
83+
84+
def testConfigTimeoutSave(self):
85+
c = self.console.get_console()
86+
self.setConfigTimeout(self.new_timeout)
87+
88+
# exit to shell, check saved timeout
89+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
90+
timeout = self.console.run_command_ignore_fail('nvram --print-config=petitboot,timeout')
91+
self.assertTrue(str(self.new_timeout) in timeout, "New timeout value not seen")
92+
93+
self.system.goto_state(OpSystemState.PETITBOOT)
94+
95+
def testConfigTimeoutSaveDefault(self):
96+
c = self.console.get_console()
97+
98+
# set a new timeout
99+
self.setConfigTimeout(self.new_timeout)
100+
101+
# exit to shell and check setting
102+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
103+
self.console.run_command('ls')
104+
timeout = self.console.run_command('nvram --print-config=petitboot,timeout')
105+
self.assertTrue("314" in timeout, "New timeout value not seen")
106+
107+
# back to UI
108+
self.system.goto_state(OpSystemState.PETITBOOT)
109+
110+
# wait for UI to settle after update
111+
time.sleep(10)
112+
113+
# reset to default
114+
self.setConfigTimeout(self.default_timeout)
115+
116+
# exit to shell and expect no setting
117+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
118+
timeout = self.console.run_command_ignore_fail('nvram --print-config | grep -c petitboot,timeout')
119+
self.assertTrue("0" in timeout, "Timeout doesn't appear to be reset")
120+
121+
self.system.goto_state(OpSystemState.PETITBOOT)
122+
123+
class StaticNetworkConfigTestCase(unittest.TestCase):
124+
def setUp(self):
125+
conf = OpTestConfiguration.conf
126+
self.system = conf.system()
127+
self.bmc = conf.bmc()
128+
self.console = self.system.console
129+
130+
if conf.args.bmc_type is not 'qemu':
131+
self.skipTest("This test is intended for qemu")
132+
133+
# TODO extend to use real network device in qemu / real machine
134+
self.network_config_str = (self.bmc.console.mac_str +
135+
',static,192.168.0.1/24,192.168.0.2 dns,192.168.0.3')
136+
137+
self.system.goto_state(OpSystemState.PETITBOOT)
138+
139+
# Let Petitboot discovery settle
140+
time.sleep(10)
141+
142+
def testConfigStaticNetworkVar(self):
143+
c = self.console.get_console()
144+
c.send('c')
145+
c.expect('Petitboot System Configuration')
146+
c.expect('Static IP')
147+
time.sleep(1)
148+
149+
# navigate to network-type widget
150+
c.send(self.KEY_TAB)
151+
c.send(self.KEY_TAB)
152+
c.send(self.KEY_TAB)
153+
c.send(self.KEY_TAB)
154+
c.send(self.KEY_TAB)
155+
c.send(self.KEY_TAB)
156+
157+
# select static
158+
c.send(self.KEY_DOWN)
159+
c.send(self.KEY_DOWN)
160+
c.send(' ')
161+
162+
# navigate to ip params
163+
c.send(self.KEY_TAB)
164+
c.send(self.KEY_TAB)
165+
166+
# set up, mask, gateway, DNS
167+
c.send('192.168.0.1')
168+
c.send(self.KEY_DOWN)
169+
c.send('24')
170+
c.send(self.KEY_DOWN)
171+
c.send('192.168.0.2')
172+
c.send(self.KEY_DOWN)
173+
# skip URL field
174+
c.send(self.KEY_DOWN)
175+
c.send('192.168.0.3')
176+
177+
# OK!
178+
c.send(self.KEY_PGDOWN)
179+
c.send(self.KEY_UP)
180+
c.send(self.KEY_UP)
181+
c.send(' ')
182+
c.expect("e=edit, n=new")
183+
184+
# drop to shell
185+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
186+
config = self.console.run_command('nvram --print-config')
187+
self.assertTrue('petitboot,network=%s' % self.network_config_str in tconfig, "Network config not correct")
188+
189+
self.system.goto_state(OpSystemState.PETITBOOT)
190+
191+
def testReconfig(self):
192+
# Test saving an un-changed config.
193+
# We should see the same results as the first save
194+
195+
self.testConfigStaticNetworkVar()
196+
197+
# return to petitboot UI
198+
c.send('exit')
199+
c.expect('Petitboot')
200+
201+
# enter system config
202+
c.send('c')
203+
c.expect('Petitboot System Configuration')
204+
c.expect('Static IP')
205+
206+
# select 'OK' button to save config
207+
time.sleep(0.1)
208+
c.send(self.KEY_PGDOWN)
209+
c.send(self.KEY_BTAB)
210+
c.send(self.KEY_BTAB)
211+
c.send(' ')
212+
time.sleep(0.1)
213+
214+
# back to shell, check for the same config string
215+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
216+
config = self.console.run_command('nvram --print-config')
217+
self.assertTrue('petitboot,network=%s' % self.network_config_str in tconfig, "Network config not correct")
218+
219+
self.system.goto_state(OpSystemState.PETITBOOT)
220+
221+
class RestoreConfigDefaultTestCase(unittest.TestCase):
222+
def setUp(self):
223+
conf = OpTestConfiguration.conf
224+
self.system = conf.system()
225+
self.bmc = conf.bmc()
226+
self.console = self.system.console
227+
228+
if conf.args.bmc_type is not 'qemu':
229+
self.skipTest("This test is intended for qemu")
230+
231+
# TODO extend to use real network device in qemu / real machine
232+
self.network_config_str = (self.bmc.console.mac_str +
233+
',static,192.168.0.1/24,192.168.0.2 dns,192.168.0.3')
234+
235+
self.system.goto_state(OpSystemState.PETITBOOT)
236+
237+
# Let Petitboot discovery settle
238+
time.sleep(10)
239+
240+
def testRestoreDefaultAutoboot(self):
241+
c = self.console.get_console()
242+
243+
# enter config
244+
c.send('c')
245+
c.expect('Petitboot System Configuration')
246+
c.expect('Boot Order')
247+
248+
# clear boot order list
249+
c.send(self.KEY_TAB)
250+
c.send(self.KEY_TAB)
251+
c.send(self.KEY_DOWN)
252+
c.send(self.KEY_DOWN)
253+
c.send(' ')
254+
255+
# save config, exit from UI
256+
c.send(self.KEY_PGDOWN)
257+
c.send(self.KEY_BTAB)
258+
c.send(self.KEY_BTAB)
259+
c.send(' ')
260+
261+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
262+
config = self.console.run_command('nvram --print-config | grep auto-boot')
263+
self.assertTrue("auto-boot?=false" in config, "Autoboot not disabled")
264+
265+
self.system.goto_state(OpSystemState.PETITBOOT)
266+
c.send('c')
267+
c.expect('Petitboot System Configuration')
268+
# Autoboot will have been disabled by an empty boot order
269+
c.expect('Autoboot:')
270+
271+
# re-enable autoboot
272+
c.send(self.KEY_DOWN)
273+
c.send(' ')
274+
# set autoboot option
275+
c.send(self.KEY_TAB)
276+
c.send(self.KEY_TAB)
277+
c.send(' ')
278+
c.send(self.KEY_UP)
279+
c.send(self.KEY_UP)
280+
c.send(' ')
281+
282+
# save config again, exit from UI
283+
c.send(self.KEY_PGDOWN)
284+
c.send(self.KEY_BTAB)
285+
c.send(self.KEY_BTAB)
286+
c.send(' ')
287+
288+
# we should see that the auto-boot nvram setting has been removed
289+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
290+
config = self.console.run_command_ignore_fail('nvram --print-config | grep -q auto-boot; echo $?')
291+
self.assertTrue("auto-boot?=false" not in config, "Autoboot still disabled")
292+
293+
self.system.goto_state(OpSystemState.PETITBOOT)
294+
295+
def testRestoreDefaultTimeout(self):
296+
c = self.console.get_console()
297+
298+
# enter config
299+
c.send('c')
300+
c.expect('Petitboot System Configuration')
301+
c.expect('Boot Order')
302+
303+
# set timeout to non-default
304+
c.send(self.KEY_TAB)
305+
c.send(self.KEY_TAB)
306+
c.send(self.KEY_TAB)
307+
c.send(self.KEY_TAB)
308+
c.send(self.KEY_TAB)
309+
c.sendcontrol('h')
310+
c.sendcontrol('h')
311+
c.send('42')
312+
313+
# save config, exit from UI
314+
c.send(self.KEY_PGDOWN)
315+
c.send(self.KEY_BTAB)
316+
c.send(self.KEY_BTAB)
317+
c.send(' ')
318+
319+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
320+
timeout = self.console.run_command('nvram --print-config | grep petitboot,timeout')
321+
self.assertTrue("petitboot,timeout=42" in timeout, "New timeout value not seen")
322+
323+
# exit shell
324+
self.system.goto_state(OpSystemState.PETITBOOT)
325+
326+
# set timeout back to default of 10
327+
c.send('c')
328+
c.expect('Petitboot System Configuration')
329+
c.expect('Boot Order')
330+
c.send(self.KEY_TAB)
331+
c.send(self.KEY_TAB)
332+
c.send(self.KEY_TAB)
333+
c.send(self.KEY_TAB)
334+
c.send(self.KEY_TAB)
335+
c.sendcontrol('h')
336+
c.sendcontrol('h')
337+
c.send('10')
338+
339+
# save config again, exit from UI
340+
c.send(self.KEY_PGDOWN)
341+
c.send(self.KEY_BTAB)
342+
c.send(self.KEY_BTAB)
343+
c.send(' ')
344+
345+
self.system.goto_state(OpSystemState.PETITBOOT_SHELL)
346+
347+
# we should see that the auto-boot nvram setting has been removed
348+
timeout = self.console.run_command_ignore_fail('nvram --print-config | grep -q petitboot,timeout')
349+
self.assertTrue("0" in timeout, "Timeout doesn't appear to be reset")
350+
351+
self.system.goto_state(OpSystemState.PETITBOOT)

0 commit comments

Comments
 (0)