4f29a32da8
It is not very useful to just delete the data again after a failure, as it might be useful for debugging. Just store it into an "errors" subdirectory of the PWD in the hope that this is a sane location. Note that it'll error out if the directory already exists, but that should be acceptable in all cases. i.e. it won't exist in the CI and developers can just wipe the directory.
97 lines
3.2 KiB
Python
Executable file
97 lines
3.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import os
|
|
import os.path
|
|
import shutil
|
|
import tempfile
|
|
import subprocess
|
|
|
|
if len(sys.argv) != 2:
|
|
print("You need to specify exactly one argument, the directory with test data")
|
|
|
|
# Check that umockdev is available
|
|
try:
|
|
umockdev_version = subprocess.check_output(['umockdev-run', '--version'])
|
|
version = tuple(int(_) for _ in umockdev_version.split(b'.'))
|
|
if version < (0, 13, 2):
|
|
print('umockdev is too old for test to be reliable, expect random failures!')
|
|
print('Please update umockdev to at least 0.13.2.')
|
|
except FileNotFoundError:
|
|
print('umockdev-run not found, skipping test!')
|
|
print('Please install umockdev.')
|
|
sys.exit(77)
|
|
|
|
edir = os.path.dirname(sys.argv[0])
|
|
ddir = sys.argv[1]
|
|
|
|
tmpdir = tempfile.mkdtemp(prefix='libfprint-umockdev-test-')
|
|
|
|
assert os.path.isdir(ddir)
|
|
assert os.path.isfile(os.path.join(ddir, "device"))
|
|
|
|
def cmp_pngs(png_a, png_b):
|
|
print("Comparing PNGs %s and %s" % (png_a, png_b))
|
|
import cairo
|
|
img_a = cairo.ImageSurface.create_from_png(png_a)
|
|
img_b = cairo.ImageSurface.create_from_png(png_b)
|
|
|
|
assert img_a.get_format() == cairo.FORMAT_RGB24
|
|
assert img_b.get_format() == cairo.FORMAT_RGB24
|
|
assert img_a.get_width() == img_b.get_width()
|
|
assert img_a.get_height() == img_b.get_height()
|
|
assert img_a.get_stride () == img_b.get_stride()
|
|
|
|
data_a = img_a.get_data()
|
|
data_b = img_b.get_data()
|
|
stride = img_a.get_stride()
|
|
|
|
for x in range(img_a.get_width()):
|
|
for y in range(img_a.get_height()):
|
|
# RGB24 format is endian dependent, using +1 means we test either
|
|
# the G or B component, which works on any endian for the greyscale
|
|
# test.
|
|
assert(data_a[y * stride + x * 4 + 1] == data_b[y * stride + x * 4 + 1])
|
|
|
|
def get_umockdev_runner(ioctl_basename):
|
|
ioctl = os.path.join(ddir, "{}.ioctl".format(ioctl_basename))
|
|
device = os.path.join(ddir, "device")
|
|
dev = open(ioctl).readline().strip()
|
|
assert dev.startswith('@DEV ')
|
|
dev = dev[5:]
|
|
|
|
umockdev = ['umockdev-run', '-d', device,
|
|
'-i', "%s=%s" % (dev, ioctl),
|
|
'--']
|
|
wrapper = os.getenv('LIBFPRINT_TEST_WRAPPER')
|
|
return umockdev + (wrapper.split(' ') if wrapper else []) + [sys.executable]
|
|
|
|
def capture():
|
|
subprocess.check_call(get_umockdev_runner("capture") +
|
|
['%s' % os.path.join(edir, "capture.py"),
|
|
'%s' % os.path.join(tmpdir, "capture.png")])
|
|
|
|
assert os.path.isfile(os.path.join(tmpdir, "capture.png"))
|
|
if os.path.isfile(os.path.join(ddir, "capture.png")):
|
|
# Compare the images, they need to be identical
|
|
cmp_pngs(os.path.join(tmpdir, "capture.png"), os.path.join(ddir, "capture.png"))
|
|
|
|
def custom():
|
|
subprocess.check_call(get_umockdev_runner("custom") +
|
|
['%s' % os.path.join(ddir, "custom.py")])
|
|
|
|
try:
|
|
if os.path.exists(os.path.join(ddir, "capture.ioctl")):
|
|
capture()
|
|
|
|
if os.path.exists(os.path.join(ddir, "custom.ioctl")):
|
|
custom()
|
|
|
|
except:
|
|
# Store created output files for inspection (in the build directory)
|
|
outdir = os.path.join('errors', os.path.basename(ddir))
|
|
shutil.copytree(tmpdir, outdir)
|
|
|
|
finally:
|
|
shutil.rmtree(tmpdir)
|
|
|