Changes in tests/test.py [f806b61:35a408b7]

File:

• tests/test.py (modified) (2 diffs)

tests/test.py

@@ -202 +202 @@
                 if error :
                         text = text + '\n' + error
-                        out = sys.stderr
-
-                print(text, file = out)
-                sys.stdout.flush()
+
+                return retcode == TestResult.SUCCESS, text
+        except KeyboardInterrupt:
+                return False, ""
+        except:
+                print("Unexpected error in worker thread", file=sys.stderr)
                 sys.stderr.flush()
-
-                return retcode != TestResult.SUCCESS
-        except KeyboardInterrupt:
-                False
+                return False, ""
+
 
 # run the given list of tests with the given parameters
@@ -220 +220 @@
         pool = multiprocessing.Pool(jobs)
 
+        failed = False
+
         # for each test to run
         try :
-                results = pool.map_async(
+                num = len(tests)
+                fancy = sys.stdout.isatty()
+                results = pool.imap_unordered(
                         run_test_worker,
                         tests,
                         chunksize = 1
-                ).get(settings.timeout.total)
+                )
+
+                for i, (succ, txt) in enumerate(timed(results, timeout = settings.timeout.total), 1) :
+                        if not succ :
+                                failed = True
+
+                        print("       " + txt)
+
+                        if(fancy and i != num):
+                                print("%d/%d" % (i, num), end='\r')
+                                sys.stdout.flush()
+
         except KeyboardInterrupt:
+                print("Tests interrupted by user", file=sys.stderr)
                 pool.terminate()
-                print("Tests interrupted by user")
-                sys.exit(1)
+                pool.join()
+                failed = True
+        except multiprocessing.TimeoutError:
+                print("ERROR: Test suite timed out", file=sys.stderr)
+                pool.terminate()
+                pool.join()
+                failed = True
+                killgroup() # needed to cleanly kill all children
+
 
         # clean the workspace
         make('clean', output=subprocess.DEVNULL, error=subprocess.DEVNULL)
 
-        for failed in results:
-                if failed :
-                        return 1
-
-        return 0
+        return 1 if failed else 0