Acceptance tests: refactor wait_for_console_pattern
The same utility method is already present in two different test files, so let's consolidate it into a single utility function. Signed-off-by: Cleber Rosa <crosa@redhat.com> Message-Id: <20190916164011.7653-1-crosa@redhat.com> Reviewed-by: Philippe Mathieu-Daudé <philmd@redhat.com> [PMD: failure_message is optional] Reviewed-by: David Gibson <david@gibson.dropbear.id.au> Signed-off-by: Philippe Mathieu-Daudé <philmd@redhat.com> Reviewed-by: Aleksandar Markovic <amarkovic@wavecomp.com> Message-Id: <20191028073441.6448-3-philmd@redhat.com> Signed-off-by: Cleber Rosa <crosa@redhat.com>
This commit is contained in:
parent
0858096702
commit
77bcd2487e
@ -8,6 +8,7 @@
|
||||
# This work is licensed under the terms of the GNU GPL, version 2 or
|
||||
# later. See the COPYING file in the top-level directory.
|
||||
|
||||
import logging
|
||||
import os
|
||||
import sys
|
||||
import uuid
|
||||
@ -54,6 +55,30 @@ def pick_default_qemu_bin(arch=None):
|
||||
return qemu_bin_from_src_dir_path
|
||||
|
||||
|
||||
def wait_for_console_pattern(test, success_message, failure_message=None):
|
||||
"""
|
||||
Waits for messages to appear on the console, while logging the content
|
||||
|
||||
:param test: an Avocado test containing a VM that will have its console
|
||||
read and probed for a success or failure message
|
||||
:type test: :class:`avocado_qemu.Test`
|
||||
:param success_message: if this message appears, test succeeds
|
||||
:param failure_message: if this message appears, test fails
|
||||
"""
|
||||
console = test.vm.console_socket.makefile()
|
||||
console_logger = logging.getLogger('console')
|
||||
while True:
|
||||
msg = console.readline().strip()
|
||||
if not msg:
|
||||
continue
|
||||
console_logger.debug(msg)
|
||||
if success_message in msg:
|
||||
break
|
||||
if failure_message and failure_message in msg:
|
||||
fail = 'Failure message found in console: %s' % failure_message
|
||||
test.fail(fail)
|
||||
|
||||
|
||||
class Test(avocado.Test):
|
||||
def setUp(self):
|
||||
self._vms = {}
|
||||
|
@ -9,12 +9,12 @@
|
||||
# later. See the COPYING file in the top-level directory.
|
||||
|
||||
import os
|
||||
import logging
|
||||
import lzma
|
||||
import gzip
|
||||
import shutil
|
||||
|
||||
from avocado_qemu import Test
|
||||
from avocado_qemu import wait_for_console_pattern
|
||||
from avocado.utils import process
|
||||
from avocado.utils import archive
|
||||
|
||||
@ -29,31 +29,14 @@ class BootLinuxConsole(Test):
|
||||
|
||||
KERNEL_COMMON_COMMAND_LINE = 'printk.time=0 '
|
||||
|
||||
def wait_for_console_pattern(self, success_message,
|
||||
failure_message='Kernel panic - not syncing'):
|
||||
"""
|
||||
Waits for messages to appear on the console, while logging the content
|
||||
|
||||
:param success_message: if this message appears, test succeeds
|
||||
:param failure_message: if this message appears, test fails
|
||||
"""
|
||||
console = self.vm.console_socket.makefile()
|
||||
console_logger = logging.getLogger('console')
|
||||
while True:
|
||||
msg = console.readline().strip()
|
||||
if not msg:
|
||||
continue
|
||||
console_logger.debug(msg)
|
||||
if success_message in msg:
|
||||
break
|
||||
if failure_message in msg:
|
||||
fail = 'Failure message found in console: %s' % failure_message
|
||||
self.fail(fail)
|
||||
def wait_for_console_pattern(self, success_message):
|
||||
wait_for_console_pattern(self, success_message,
|
||||
failure_message='Kernel panic - not syncing')
|
||||
|
||||
def exec_command_and_wait_for_pattern(self, command, success_message):
|
||||
command += '\n'
|
||||
self.vm.console_socket.sendall(command.encode())
|
||||
self.wait_for_console_pattern(success_message)
|
||||
wait_for_console_pattern(self, success_message)
|
||||
|
||||
def extract_from_deb(self, deb, path):
|
||||
"""
|
||||
|
@ -13,6 +13,7 @@ import time
|
||||
|
||||
from avocado import skipUnless
|
||||
from avocado_qemu import Test
|
||||
from avocado_qemu import wait_for_console_pattern
|
||||
from avocado.utils import process
|
||||
from avocado.utils import archive
|
||||
from avocado.utils import ssh
|
||||
@ -69,19 +70,6 @@ class LinuxSSH(Test):
|
||||
def setUp(self):
|
||||
super(LinuxSSH, self).setUp()
|
||||
|
||||
def wait_for_console_pattern(self, success_message,
|
||||
failure_message='Oops'):
|
||||
console = self.vm.console_socket.makefile()
|
||||
console_logger = logging.getLogger('console')
|
||||
while True:
|
||||
msg = console.readline()
|
||||
console_logger.debug(msg.strip())
|
||||
if success_message in msg:
|
||||
break
|
||||
if failure_message in msg:
|
||||
fail = 'Failure message found in console: %s' % failure_message
|
||||
self.fail(fail)
|
||||
|
||||
def get_portfwd(self):
|
||||
res = self.vm.command('human-monitor-command',
|
||||
command_line='info usernet')
|
||||
@ -137,7 +125,7 @@ class LinuxSSH(Test):
|
||||
|
||||
self.log.info('VM launched, waiting for sshd')
|
||||
console_pattern = 'Starting OpenBSD Secure Shell server: sshd'
|
||||
self.wait_for_console_pattern(console_pattern)
|
||||
wait_for_console_pattern(self, console_pattern, 'Oops')
|
||||
self.log.info('sshd ready')
|
||||
|
||||
self.ssh_connect('root', 'root')
|
||||
@ -145,7 +133,7 @@ class LinuxSSH(Test):
|
||||
def shutdown_via_ssh(self):
|
||||
self.ssh_command('poweroff')
|
||||
self.ssh_disconnect_vm()
|
||||
self.wait_for_console_pattern('Power down')
|
||||
wait_for_console_pattern(self, 'Power down', 'Oops')
|
||||
|
||||
def ssh_command_output_contains(self, cmd, exp):
|
||||
stdout, _ = self.ssh_command(cmd)
|
||||
|
Loading…
Reference in New Issue
Block a user