From fb643b348bf87eab3bab6748faa1d03657bc2e45 Mon Sep 17 00:00:00 2001 From: Alexandre Abadie Date: Thu, 12 Mar 2020 13:42:35 +0100 Subject: [PATCH] dist/testrunner: add common function to check unittests --- dist/pythonlibs/testrunner/__init__.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/dist/pythonlibs/testrunner/__init__.py b/dist/pythonlibs/testrunner/__init__.py index 31fa260eb..6dcd105a7 100755 --- a/dist/pythonlibs/testrunner/__init__.py +++ b/dist/pythonlibs/testrunner/__init__.py @@ -9,6 +9,7 @@ import os import sys +from functools import partial from traceback import print_tb import pexpect @@ -44,3 +45,16 @@ def run(testfunc, timeout=TIMEOUT, echo=True, traceback=False): print("") teardown_child(child) return 0 + + +def check_unittests(child, timeout=TIMEOUT, nb_tests=None): + _tests = r'\d+' if nb_tests is None else int(nb_tests) + child.expect(r'OK \({} tests\)'.format(_tests), timeout=timeout) + + +def run_check_unittests(timeout=TIMEOUT, echo=True, traceback=False, + nb_tests=None): + _unittests_func = partial(check_unittests, + timeout=timeout, nb_tests=nb_tests) + + return run(_unittests_func, timeout, echo, traceback)