Add decorator method for catching exceptions and aborting

unittests/atc.py

@@ -24,6 +24,7 @@
 
 __version__ = "0.0.1"
 
+import functools
 import os
 import sys
 import unittest
@@ -101,6 +102,16 @@ def TestDone(self, passed = True):
             # wx.GetApp().GetMainLoop().Exit(30)
             sys.exit(1)
 
+def TestDependent(func):
+    @functools.wraps(func)
+    def method(*args, **kwargs):
+        try:
+            func(*args, **kwargs)
+        except Exception as e:
+            print("Unbound exception caught in test procedure:\n%s\n%s" % (e.__class__, str(e)), file = sys.stderr)
+            # give full stack trace
+            args[0].TestDone(False)
+    return method
 
 def CreateApp(frame):
     class TestApp(wx.App):
@@ -164,5 +175,5 @@ class ApplicationTestCase(unittest.TestCase):
 
     return ApplicationTestCase
 
-if __name__ == "__main__":
-    pass
+
+

unittests/test_atc.py

@@ -1,9 +1,6 @@
 from unittests import atc
 import wx
 import unittest
-import random
-
-random.seed()
 
 class PanelColorChangeTester(wx.Panel, atc.TestWidget):
     def __init__(self, parent):
@@ -20,8 +17,15 @@ def test_fail(self):
         print("Failing test")
         self.TestDone(False)
 
+    @atc.TestDependent
+    def test_abort(self):
+        print("Aborting test case")
+        raise AttributeError("Unbound attribute error")
+
+
 testcase = atc.CreateATC(PanelColorChangeTester)
 
 if __name__ == "__main__":
+    testcase.test_abort(testcase)
     unittest.main()