From e48d2b467a68416a792630f83afcb01bf0b3b5c8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Marco=20Trevisan=20=28Trevi=C3=B1o=29?= Date: Thu, 22 Jun 2023 00:10:12 -0400 Subject: [PATCH] tests/virtual-device: Restore default enroll steps on tearDown When running the tests as a single script we may leave the device in an inconsistent state, so reset the enroll steps when done. --- tests/virtual-device.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/virtual-device.py b/tests/virtual-device.py index 4dc8e56e..2d022475 100644 --- a/tests/virtual-device.py +++ b/tests/virtual-device.py @@ -59,6 +59,8 @@ class GLibErrorMessage: class VirtualDeviceBase(unittest.TestCase): + DEFAULT_ENROLL_STEPS = 5 + @classmethod def setUpClass(cls): unittest.TestCase.setUpClass() @@ -101,6 +103,7 @@ class VirtualDeviceBase(unittest.TestCase): def tearDown(self): if self._close_on_teardown: self.assertTrue(self.dev.is_open()) + self.send_command('SET_ENROLL_STAGES', self.DEFAULT_ENROLL_STEPS) self.dev.close_sync() self.assertFalse(self.dev.is_open()) super().tearDown() @@ -339,7 +342,8 @@ class VirtualDevice(VirtualDeviceBase): self.assertEqual(self.dev.get_name(), 'Virtual device for debugging') self.assertTrue(self.dev.is_open()) self.assertEqual(self.dev.get_scan_type(), FPrint.ScanType.SWIPE) - self.assertEqual(self.dev.get_nr_enroll_stages(), 5) + self.assertEqual(self.dev.get_nr_enroll_stages(), + self.DEFAULT_ENROLL_STEPS) self.assertFalse(self.dev.supports_identify()) self.assertFalse(self.dev.supports_capture()) self.assertFalse(self.dev.has_storage())