From 35d432c281e7bab73271130bbeb5bce885e8e593 Mon Sep 17 00:00:00 2001
From: Christian Poessinger <christian@poessinger.com>
Date: Thu, 16 Dec 2021 20:31:59 +0100
Subject: Testsuite: add loginVM()/shutdownVM() helpers

(cherry picked from commit a36755da0b10d884eff84f52e24a0f9ec185e461)
---
 scripts/check-qemu-install | 78 ++++++++++++++++++++--------------------------
 1 file changed, 34 insertions(+), 44 deletions(-)

(limited to 'scripts/check-qemu-install')

diff --git a/scripts/check-qemu-install b/scripts/check-qemu-install
index 54fcd6d8..e1a97c85 100755
--- a/scripts/check-qemu-install
+++ b/scripts/check-qemu-install
@@ -150,6 +150,36 @@ def get_qemu_cmd(name, enable_kvm, enable_uefi, disk_img, raid=None, iso_img=Non
 
     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
 log = logging.getLogger()
@@ -236,13 +266,7 @@ try:
     except pexpect.TIMEOUT:
         log.warning('Did not find GRUB countdown window, ignoring')
 
-    log.info('Waiting for login prompt')
-    c.expect('[Ll]ogin:', timeout=300)
-    c.sendline(default_user)
-    c.expect('[Pp]assword:')
-    c.sendline(default_password)
-    c.expect(op_mode_prompt)
-    log.info('Logged in!')
+    loginVM(c, log)
 
     #################################################
     # Installing into VyOS system
@@ -291,18 +315,7 @@ try:
     #################################################
     # Powering down installer
     #################################################
-    log.info('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!')
+    shutdownVM(c, log, 'Shutting down installation system')
     c.close()
 
     #################################################
@@ -322,13 +335,7 @@ try:
     except pexpect.TIMEOUT:
         log.warning('Did not find GRUB countdown window, ignoring')
 
-    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!')
+    loginVM(c, log)
 
     ################################################
     # Always load the WiFi simulation module
@@ -449,24 +456,7 @@ try:
             log.error(tmp)
             raise Exception(tmp)
 
-    #################################################
-    # 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)
+    shutdownVM(c, log, 'Powering off system')
     c.close()
 
 except pexpect.exceptions.TIMEOUT:
-- 
cgit v1.2.3