20869f9865
Instead of looking for a specific error, let's relax the pattern because different errors have been seen (I'm consistenly getting 52) and the real goal of this test is to validate the framebuffer operation, and not to reproduce one specific error. Signed-off-by: Cleber Rosa <crosa@redhat.com> Reviewed-by: Philippe Mathieu-Daudé <philmd@redhat.com> Reviewed-by: Thomas Huth <huth@tuxfamily.org> Message-Id: <20190919161400.26399-1-crosa@redhat.com> Signed-off-by: Cleber Rosa <crosa@redhat.com>
122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
# Functional test that boots a VM and run OCR on the framebuffer
|
|
#
|
|
# Copyright (c) Philippe Mathieu-Daudé <f4bug@amsat.org>
|
|
#
|
|
# 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 os
|
|
import re
|
|
import time
|
|
import logging
|
|
import distutils.spawn
|
|
|
|
from avocado_qemu import Test
|
|
from avocado import skipUnless
|
|
from avocado.utils import process
|
|
from avocado.utils.path import find_command, CmdNotFoundError
|
|
|
|
PIL_AVAILABLE = True
|
|
try:
|
|
from PIL import Image
|
|
except ImportError:
|
|
PIL_AVAILABLE = False
|
|
|
|
|
|
def tesseract_available(expected_version):
|
|
try:
|
|
find_command('tesseract')
|
|
except CmdNotFoundError:
|
|
return False
|
|
res = process.run('tesseract --version')
|
|
try:
|
|
version = res.stdout_text.split()[1]
|
|
except IndexError:
|
|
version = res.stderr_text.split()[1]
|
|
return int(version.split('.')[0]) == expected_version
|
|
|
|
match = re.match(r'tesseract\s(\d)', res)
|
|
if match is None:
|
|
return False
|
|
# now this is guaranteed to be a digit
|
|
return int(match.groups()[0]) == expected_version
|
|
|
|
|
|
class NextCubeMachine(Test):
|
|
|
|
timeout = 15
|
|
|
|
def check_bootrom_framebuffer(self, screenshot_path):
|
|
rom_url = ('http://www.nextcomputers.org/NeXTfiles/Software/ROM_Files/'
|
|
'68040_Non-Turbo_Chipset/Rev_2.5_v66.BIN')
|
|
rom_hash = 'b3534796abae238a0111299fc406a9349f7fee24'
|
|
rom_path = self.fetch_asset(rom_url, asset_hash=rom_hash)
|
|
|
|
self.vm.set_machine('next-cube')
|
|
self.vm.add_args('-bios', rom_path)
|
|
self.vm.launch()
|
|
|
|
self.log.info('VM launched, waiting for display')
|
|
# TODO: Use avocado.utils.wait.wait_for to catch the
|
|
# 'displaysurface_create 1120x832' trace-event.
|
|
time.sleep(2)
|
|
|
|
self.vm.command('human-monitor-command',
|
|
command_line='screendump %s' % screenshot_path)
|
|
|
|
@skipUnless(PIL_AVAILABLE, 'Python PIL not installed')
|
|
def test_bootrom_framebuffer_size(self):
|
|
"""
|
|
:avocado: tags=arch:m68k
|
|
:avocado: tags=machine:next_cube
|
|
:avocado: tags=device:framebuffer
|
|
"""
|
|
screenshot_path = os.path.join(self.workdir, "dump.png")
|
|
self.check_bootrom_framebuffer(screenshot_path)
|
|
|
|
width, height = Image.open(screenshot_path).size
|
|
self.assertEqual(width, 1120)
|
|
self.assertEqual(height, 832)
|
|
|
|
@skipUnless(tesseract_available(3), 'tesseract v3 OCR tool not available')
|
|
def test_bootrom_framebuffer_ocr_with_tesseract_v3(self):
|
|
"""
|
|
:avocado: tags=arch:m68k
|
|
:avocado: tags=machine:next_cube
|
|
:avocado: tags=device:framebuffer
|
|
"""
|
|
screenshot_path = os.path.join(self.workdir, "dump.png")
|
|
self.check_bootrom_framebuffer(screenshot_path)
|
|
|
|
console_logger = logging.getLogger('console')
|
|
text = process.run("tesseract %s stdout" % screenshot_path).stdout_text
|
|
for line in text.split('\n'):
|
|
if len(line):
|
|
console_logger.debug(line)
|
|
self.assertIn('Backplane', text)
|
|
self.assertIn('Ethernet address', text)
|
|
|
|
# Tesseract 4 adds a new OCR engine based on LSTM neural networks. The
|
|
# new version is faster and more accurate than version 3. The drawback is
|
|
# that it is still alpha-level software.
|
|
@skipUnless(tesseract_available(4), 'tesseract v4 OCR tool not available')
|
|
def test_bootrom_framebuffer_ocr_with_tesseract_v4(self):
|
|
"""
|
|
:avocado: tags=arch:m68k
|
|
:avocado: tags=machine:next_cube
|
|
:avocado: tags=device:framebuffer
|
|
"""
|
|
screenshot_path = os.path.join(self.workdir, "dump.png")
|
|
self.check_bootrom_framebuffer(screenshot_path)
|
|
|
|
console_logger = logging.getLogger('console')
|
|
proc = process.run("tesseract --oem 1 %s stdout" % screenshot_path)
|
|
text = proc.stdout_text
|
|
for line in text.split('\n'):
|
|
if len(line):
|
|
console_logger.debug(line)
|
|
self.assertIn('Testing the FPU, SCC', text)
|
|
self.assertIn('System test failed. Error code', text)
|
|
self.assertIn('Boot command', text)
|
|
self.assertIn('Next>', text)
|