tests/fprintd: Add option to claim device for user on enroll_image

This commit is contained in:
Marco Trevisan (Treviño)
2020-12-19 02:01:48 +01:00
parent 15b41aa7c2
commit 8ef255a3bd

View File

@ -605,11 +605,16 @@ class FPrintdVirtualDeviceBaseTest(FPrintdVirtualImageDeviceBaseTests):
if expected is not None: if expected is not None:
self.assertEqual(self._last_result, expected) self.assertEqual(self._last_result, expected)
def enroll_image(self, img, device=None, finger='right-index-finger', expected_result='enroll-completed'): def enroll_image(self, img, device=None, finger='right-index-finger',
self._maybe_reduce_enroll_stages() expected_result='enroll-completed', claim_user=None):
if device is None: if device is None:
device = self.device device = self.device
if claim_user:
device.Claim('(s)', claim_user)
if device is self.device:
self._maybe_reduce_enroll_stages()
device.EnrollStart('(s)', finger) device.EnrollStart('(s)', finger)
while not self.finger_needed: while not self.finger_needed:
@ -629,6 +634,9 @@ class FPrintdVirtualDeviceBaseTest(FPrintdVirtualImageDeviceBaseTests):
self.assertEqual(self._last_result, expected_result) self.assertEqual(self._last_result, expected_result)
self.assertFalse(self.finger_needed) self.assertFalse(self.finger_needed)
if claim_user:
device.Release()
def enroll_multiple_images(self, images_override={}, return_index=-1): def enroll_multiple_images(self, images_override={}, return_index=-1):
enroll_map = { enroll_map = {
'left-thumb': 'whorl', 'left-thumb': 'whorl',