vyos/vyos-build
1
0
Fork 0
mirror of https://github.com/vyos/vyos-build.git synced 2025-10-01 20:28:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

Testsuite: add loginVM()/shutdownVM() helpers

Browse Source
This commit is contained in:
Christian Poessinger 2021-12-16 20:31:59 +01:00
parent e836f72458
commit a36755da0b
1 changed files with 34 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
scripts/check-qemu-install
View File

@ -156,6 +156,36 @@ def get_qemu_cmd(name, enable_kvm, enable_uefi, disk_img, raid=None, iso_img=Non
return cmd return cmd
def shutdownVM(c, log, message=''):
#################################################
# Powering off system
#################################################
if message:
log.info(message)
c.sendline('poweroff')
c.expect(r'\nAre you sure you want to poweroff this system.*\]')
c.sendline('Y')
log.info('Shutting down virtual machine')
for i in range(30):
log.info('Waiting for shutdown...')
if not c.isalive():
log.info('VM is shut down!')
break
time.sleep(10)
else:
tmp = 'VM Did not shut down after 300sec'
log.error(tmp)
raise Exception(tmp)
def loginVM(c, log):
log.info('Waiting for login prompt')
c.expect('[Ll]ogin:', timeout=600)
c.sendline(default_user)
c.expect('[Pp]assword:')
c.sendline(default_password)
c.expect(op_mode_prompt)
log.info('Logged in!')
# Setting up logger # Setting up logger
log = logging.getLogger() log = logging.getLogger()
@ -243,13 +273,7 @@ try:
except pexpect.TIMEOUT: except pexpect.TIMEOUT:
log.warning('Did not find GRUB countdown window, ignoring') log.warning('Did not find GRUB countdown window, ignoring')
log.info('Waiting for login prompt') loginVM(c, log)
c.expect('[Ll]ogin:', timeout=600)
c.sendline('vyos')
c.expect('[Pp]assword:')
c.sendline('vyos')
c.expect(op_mode_prompt)
log.info('Logged in!')
################################################# #################################################
# Installing into VyOS system # Installing into VyOS system
@ -298,18 +322,7 @@ try:
################################################# #################################################
# Powering down installer # Powering down installer
################################################# #################################################
log.info('Shutting down installation system') shutdownVM(c, log, 'Shutting down installation system')
c.sendline('poweroff')
c.expect(r'\nAre you sure you want to poweroff this system.*\]')
c.sendline('Y')
for i in range(30):
log.info('Waiting for shutdown...')
if not c.isalive():
log.info('VM shutdown!')
break
time.sleep(10)
else:
log.error('VM Did not shutdown after 300sec, killing it!')
c.close() c.close()
################################################# #################################################
@ -329,13 +342,7 @@ try:
except pexpect.TIMEOUT: except pexpect.TIMEOUT:
log.warning('Did not find GRUB countdown window, ignoring') log.warning('Did not find GRUB countdown window, ignoring')
log.info('Waiting for login prompt') loginVM(c, log)
c.expect('[Ll]ogin:', timeout=600)
c.sendline(default_user)
c.expect('[Pp]assword:')
c.sendline(default_password)
c.expect(op_mode_prompt)
log.info('Logged in!')
################################################ ################################################
# Always load the WiFi simulation module # Always load the WiFi simulation module
@ -470,24 +477,7 @@ try:
log.error(tmp) log.error(tmp)
raise Exception(tmp) raise Exception(tmp)
################################################# shutdownVM(c, log, 'Powering off system')
# Powering off system
#################################################
log.info("Powering off system ")
c.sendline('poweroff')
c.expect(r'\nAre you sure you want to poweroff this system.*\]')
c.sendline('Y')
log.info('Shutting down virtual machine')
for i in range(30):
log.info('Waiting for shutdown...')
if not c.isalive():
log.info('VM is shut down!')
break
time.sleep(10)
else:
tmp = 'VM Did not shut down after 300sec'
log.error(tmp)
raise Exception(tmp)
c.close() c.close()
except pexpect.exceptions.TIMEOUT: except pexpect.exceptions.TIMEOUT: