From 15b41aa7c209f05f0a594926eff1b9a2db2fbeda Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marco=20Trevisan=20=28Trevi=C3=B1o=29?= <mail@3v1n0.net>
Date: Fri, 18 Dec 2020 01:18:21 +0100
Subject: [PATCH] tests/fprintd: Add function and tests to enroll multiple
 users fingers/images

Allow to enroll multiple data in a single shot so that we don't have to
do it for each user, and add a test that uses it to match each possible
combination.
---
 tests/fprintd.py | 61 ++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 56 insertions(+), 5 deletions(-)

diff --git a/tests/fprintd.py b/tests/fprintd.py
index b1c045c..04317b9 100644
--- a/tests/fprintd.py
+++ b/tests/fprintd.py
@@ -578,6 +578,13 @@ class FPrintdVirtualDeviceBaseTest(FPrintdVirtualImageDeviceBaseTests):
 
         super().tearDown()
 
+    def try_release(self):
+        try:
+            self.device.Release()
+        except GLib.GError as e:
+            if not 'net.reactivated.Fprint.Error.ClaimDevice' in e.message:
+                raise(e)
+
     def wait_for_result(self, expected=None, max_wait=-1):
         self._last_result = None
         self._verify_stopped = False
@@ -641,6 +648,33 @@ class FPrintdVirtualDeviceBaseTest(FPrintdVirtualImageDeviceBaseTests):
 
         return (enrolled, enroll_map)
 
+    def enroll_users_images(self, images_override={}):
+        enroll_map = {
+            'test-user1': {'left-thumb': 'whorl'},
+            'test-user2': {'right-index-finger': 'arch'},
+            'test-user3': {'left-little-finger': 'loop-right',
+                           'left-thumb': 'tented_arch'},
+        }
+        enroll_map.update(images_override)
+        enrolled_prints = []
+
+        self.try_release()
+
+        for user, print_map in enroll_map.items():
+            self.device.Claim('(s)', user)
+            for finger, p in print_map.items():
+                self.enroll_image(p, finger=finger)
+                enrolled_prints.append(p)
+            self.device.Release()
+
+        self.assertCountEqual(enrolled_prints, set(enrolled_prints))
+
+        for user in enroll_map:
+            enrolled = self.device.ListEnrolledFingers('(s)', user)
+            self.assertCountEqual(enroll_map[user].keys(), enrolled)
+
+        return enroll_map
+
     def get_secondary_bus_and_device(self, claim=None):
         addr = os.environ['DBUS_SYSTEM_BUS_ADDRESS']
 
@@ -1294,11 +1328,7 @@ class FPrintdVirtualDeviceClaimedTest(FPrintdVirtualDeviceBaseTest):
 
     def tearDown(self):
         self._polkitd_obj.SetAllowed([FprintDevicePermission.enroll])
-        try:
-            self.device.Release()
-        except GLib.GError as e:
-            if not 'net.reactivated.Fprint.Error.ClaimDevice' in e.message:
-                raise(e)
+        self.try_release()
         super().tearDown()
 
     def test_any_finger_enroll_start(self):
@@ -1618,6 +1648,27 @@ class FPrintdVirtualDeviceClaimedTest(FPrintdVirtualDeviceBaseTest):
         self.assertVerifyNoMatch(selected_finger)
         self.device.VerifyStop()
 
+    def test_verify_any_finger_multiple_users(self):
+        enroll_map = self.enroll_users_images()
+        enrolled_users = list(enroll_map)
+
+        for verifying_user in enrolled_users:
+            self.device.Claim('(s)', verifying_user)
+
+            for enrolled_user in enrolled_users:
+                should_match = enrolled_user == verifying_user
+
+                for finger, print in enroll_map[enrolled_user].items():
+                    self.device.VerifyStart('(s)', 'any')
+                    self.send_image(print)
+                    if should_match:
+                        self.assertVerifyMatch(selected_finger='any')
+                    else:
+                        self.assertVerifyNoMatch(selected_finger='any')
+                    self.device.VerifyStop()
+
+            self.device.Release()
+
     def test_verify_finger_not_enrolled(self):
         self.enroll_image('whorl', finger='left-thumb')
         with self.assertFprintError('NoEnrolledPrints'):