cookiez/pam-fprint-grosshack
1
0
Fork 0
mirror of https://gitlab.com/mishakmak/pam-fprint-grosshack.git synced 2026-04-09 04:13:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

test_fprintd_utils: Use a non-blocking pipe to read output

Browse Source 
Avoid using a temporary file for reading the utilities output, so we can
read it as it comes, ignoring the previous one, and avoiding open/closing
the file.

To keep the output printing on cleanup working, adding an utility function
that reads the output and save it for later printing
This commit is contained in:
Marco Trevisan (Treviño)
2020-02-14 18:16:23 +01:00
committed by Bastien Nocera
parent 072fbc2b46
commit cfbded36e1
1 changed files with 30 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
tests/test_fprintd_utils.py
View File

@ -75,65 +75,62 @@ class TestFprintd(dbusmock.DBusTestCase):
self.device_mock.SetEnrolledFingers('toto', ['left-little-finger', 'right-little-finger']) self.device_mock.SetEnrolledFingers('toto', ['left-little-finger', 'right-little-finger'])
def start_utility_process(self, utility_name, args=[], sleep=True): 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)) ] utility = [ os.path.join(self.tools_prefix, 'fprintd-{}'.format(utility_name)) ]
process = subprocess.Popen(self.wrapper_args + utility + args, process = subprocess.Popen(self.wrapper_args + utility + args,
stdout=mock_log, stdout=subprocess.PIPE,
stderr=subprocess.STDOUT, stderr=subprocess.STDOUT,
universal_newlines=True) universal_newlines=True)
flags = fcntl.fcntl(process.stdout, fcntl.F_GETFL)
fcntl.fcntl(process.stdout, fcntl.F_SETFL, flags | os.O_NONBLOCK)
self.addCleanup(lambda: print(process.stdout.read()))
self.addCleanup(process.terminate) self.addCleanup(process.terminate)
self.addCleanup(process.wait) 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: if sleep:
time.sleep(self.sleep_time) time.sleep(self.sleep_time)
return (process, mock_log) return process
def get_process_output(self, process):
out = process.stdout.read()
self.addCleanup(print, out)
return out
def run_utility_process(self, utility_name, args=[], sleep=True, timeout=None): 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) proc = 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) ret = proc.wait(timeout=timeout if timeout is not None else self.sleep_time * 4)
self.assertLessEqual(ret, 128) self.assertLessEqual(ret, 128)
with open(mock_log.name) as f: return self.get_process_output(proc)
return f.read()
def test_fprintd_enroll(self): def test_fprintd_enroll(self):
self.setup_device() self.setup_device()
process, mock_log = self.start_utility_process('enroll', ['-f', 'right-index-finger', 'toto']) process = self.start_utility_process('enroll', ['-f', 'right-index-finger', 'toto'])
with open(mock_log.name) as f: out = self.get_process_output(process)
out = f.read()
self.assertRegex(out, r'right-index-finger') self.assertRegex(out, r'right-index-finger')
self.device_mock.EmitEnrollStatus('enroll-completed', True) self.device_mock.EmitEnrollStatus('enroll-completed', True)
time.sleep(self.sleep_time) time.sleep(self.sleep_time)
with open(mock_log.name) as f: out = self.get_process_output(process)
out = f.read()
self.assertRegex(out, 'Enroll result: enroll-completed') self.assertRegex(out, 'Enroll result: enroll-completed')
def test_fprintd_verify(self): def test_fprintd_verify(self):
self.setup_device() self.setup_device()
process, mock_log = self.start_utility_process('verify', ['toto']) process = self.start_utility_process('verify', ['toto'])
with open(mock_log.name) as f: out = self.get_process_output(process)
out = f.read()
self.assertRegex(out, r'left-little-finger') self.assertRegex(out, r'left-little-finger')
self.assertNotRegex(out, 'Verify result: verify-match \(done\)') self.assertNotRegex(out, 'Verify result: verify-match \(done\)')
self.device_mock.EmitVerifyStatus('verify-match', True) self.device_mock.EmitVerifyStatus('verify-match', True)
time.sleep(self.sleep_time) time.sleep(self.sleep_time)
with open(mock_log.name) as f: out = self.get_process_output(process)
out = f.read()
self.assertRegex(out, 'Verify result: verify-match \(done\)') self.assertRegex(out, 'Verify result: verify-match \(done\)')
def test_fprintd_verify_script(self): def test_fprintd_verify_script(self):
@ -143,17 +140,15 @@ class TestFprintd(dbusmock.DBusTestCase):
] ]
self.device_mock.SetVerifyScript(script) self.device_mock.SetVerifyScript(script)
process, mock_log = self.start_utility_process('verify', ['toto']) process = self.start_utility_process('verify', ['toto'])
with open(mock_log.name) as f: out = self.get_process_output(process)
out = f.read()
self.assertRegex(out, r'left-little-finger') self.assertRegex(out, r'left-little-finger')
self.assertNotRegex(out, 'Verify result: verify-match \(done\)') self.assertNotRegex(out, 'Verify result: verify-match \(done\)')
time.sleep(2) time.sleep(2)
with open(mock_log.name) as f: out = self.get_process_output(process)
out = f.read()
self.assertRegex(out, 'Verify result: verify-match \(done\)') self.assertRegex(out, 'Verify result: verify-match \(done\)')
def test_fprintd_list(self): def test_fprintd_list(self):