#!/usr/bin/python3 # This program is free software; you can redistribute it and/or modify it under # the terms of the GNU Lesser General Public License as published by the Free # Software Foundation; either version 3 of the License, or (at your option) any # later version. See http://www.gnu.org/copyleft/lgpl.html for the full text # of the license. __author__ = 'Bastien Nocera' __email__ = 'hadess@hadess.net' __copyright__ = '(c) 2020 Red Hat Inc.' __license__ = 'LGPL 3+' import tempfile import unittest import sys import subprocess import dbus import dbus.mainloop.glib import dbusmock import fcntl import os import time dbus.mainloop.glib.DBusGMainLoop(set_as_default=True) class TestFprintd(dbusmock.DBusTestCase): '''Test fprintd utilities''' @classmethod def setUpClass(klass): klass.start_system_bus() klass.dbus_con = klass.get_dbus(True) klass.sleep_time = 0.5 template_path = './' if 'TOPSRCDIR' in os.environ: template_path = os.environ['TOPSRCDIR'] + '/tests/' klass.template_name = template_path + 'dbusmock/fprintd.py' print ('Using template from %s' % klass.template_name) klass.tools_prefix = '' if 'FPRINT_BUILD_DIR' in os.environ: klass.tools_prefix = os.environ['FPRINT_BUILD_DIR'] + '/../utils/' print ('Using tools from %s' % klass.tools_prefix) else: print ('Using tools from $PATH') klass.wrapper_args = [] klass.valgrind = False if 'VALGRIND' in os.environ: valgrind = os.environ['VALGRIND'] if valgrind is not None: klass.valgrind = True klass.sleep_time *= 4 klass.wrapper_args = ['valgrind', '--leak-check=full'] if os.path.exists(valgrind): klass.wrapper_args += ['--suppressions={}'.format(valgrind)] def setUp(self): (self.p_mock, self.obj_fprintd_manager) = self.spawn_server_template( self.template_name, {}, stdout=subprocess.PIPE) # set log to nonblocking flags = fcntl.fcntl(self.p_mock.stdout, fcntl.F_GETFL) fcntl.fcntl(self.p_mock.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK) self.obj_fprintd_mock = dbus.Interface(self.obj_fprintd_manager, 'net.reactivated.Fprint.Manager.Mock') def tearDown(self): self.p_mock.terminate() self.p_mock.wait() def setup_device(self): device_path = self.obj_fprintd_mock.AddDevice('FDO Trigger Finger Laser Reader', 3, 'swipe') self.device_mock = self.dbus_con.get_object('net.reactivated.Fprint', device_path) self.device_mock.SetEnrolledFingers('toto', ['left-little-finger', 'right-little-finger']) def start_utility_process(self, utility_name, args=[], sleep=True): mock_log = tempfile.NamedTemporaryFile() utility = [ os.path.join(self.tools_prefix, 'fprintd-{}'.format(utility_name)) ] process = subprocess.Popen(self.wrapper_args + utility + args, stdout=mock_log, stderr=subprocess.STDOUT, universal_newlines=True) self.addCleanup(process.terminate) self.addCleanup(process.wait) def print_proc_log(mock_log): with open(mock_log.name) as f: print(f.read()) self.addCleanup(print_proc_log, mock_log) if sleep: time.sleep(self.sleep_time) return (process, mock_log) def run_utility_process(self, utility_name, args=[], sleep=True, timeout=None): proc, mock_log = self.start_utility_process(utility_name, args=args, sleep=sleep) ret = proc.wait(timeout=timeout if timeout is not None else self.sleep_time * 4) self.assertLessEqual(ret, 128) with open(mock_log.name) as f: return f.read() def test_fprintd_enroll(self): self.setup_device() process, mock_log = self.start_utility_process('enroll', ['-f', 'right-index-finger', 'toto']) with open(mock_log.name) as f: out = f.read() self.assertRegex(out, r'right-index-finger') self.device_mock.EmitEnrollStatus('enroll-completed', True) time.sleep(self.sleep_time) with open(mock_log.name) as f: out = f.read() self.assertRegex(out, 'Enroll result: enroll-completed') def test_fprintd_verify(self): self.setup_device() process, mock_log = self.start_utility_process('verify', ['toto']) with open(mock_log.name) as f: out = f.read() self.assertRegex(out, r'left-little-finger') self.assertNotRegex(out, 'Verify result: verify-match \(done\)') self.device_mock.EmitVerifyStatus('verify-match', True) time.sleep(self.sleep_time) with open(mock_log.name) as f: out = f.read() self.assertRegex(out, 'Verify result: verify-match \(done\)') def test_fprintd_verify_script(self): self.setup_device() script = [ ( 'verify-match', True, 2 ) ] self.device_mock.SetVerifyScript(script) process, mock_log = self.start_utility_process('verify', ['toto']) with open(mock_log.name) as f: out = f.read() self.assertRegex(out, r'left-little-finger') self.assertNotRegex(out, 'Verify result: verify-match \(done\)') time.sleep(2) with open(mock_log.name) as f: out = f.read() self.assertRegex(out, 'Verify result: verify-match \(done\)') def test_fprintd_list(self): self.setup_device() # Rick has no fingerprints enrolled out = self.run_utility_process('list', ['rick']) self.assertRegex(out, r'has no fingers enrolled for') # Toto does out = self.run_utility_process('list', ['toto']) self.assertRegex(out, r'right-little-finger') def test_fprintd_delete(self): self.setup_device() # Has fingerprints enrolled out = self.run_utility_process('list', ['toto']) self.assertRegex(out, r'left-little-finger') self.assertRegex(out, r'right-little-finger') # Delete fingerprints out = self.run_utility_process('delete', ['toto']) self.assertRegex(out, r'Fingerprints deleted') # Doesn't have fingerprints out = self.run_utility_process('list', ['toto']) self.assertRegex(out, r'has no fingers enrolled for') if __name__ == '__main__': # avoid writing to stderr unittest.main(testRunner=unittest.TextTestRunner(stream=sys.stdout, verbosity=2))