From efb461c0e1dcd15577a6a072668990facc5533f6 Mon Sep 17 00:00:00 2001 From: Andrei Zeliankou Date: Thu, 14 Nov 2019 17:15:20 +0300 Subject: Tests: run_process() and stop_processes() introduced. --- test/unit/main.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'test/unit') diff --git a/test/unit/main.py b/test/unit/main.py index f5727726..094fdb0e 100644 --- a/test/unit/main.py +++ b/test/unit/main.py @@ -185,6 +185,8 @@ class TestUnit(unittest.TestCase): if self._started: self._stop() + self.stop_processes() + def _run(self): self.unitd = self.pardir + '/build/unitd' @@ -287,6 +289,26 @@ class TestUnit(unittest.TestCase): if found: print('skipped.') + def run_process(self, target): + if not hasattr(self, '_processes'): + self._processes = [] + + process = Process(target=target) + process.start() + + self._processes.append(process) + + def stop_processes(self): + if not hasattr(self, '_processes'): + return + + for process in self._processes: + process.terminate() + process.join(timeout=5) + + if process.is_alive(): + self.fail('Fail to stop process') + def waitforfiles(self, *files): for i in range(50): wait = False -- cgit