insanity: Add a way to execute code in a monitor between the last Test.stop() call and the actual teardown

2 files changed, 12 insertions, 0 deletions

diff --git a/insanity/monitor.py b/insanity/monitor.py
index 29bd009..6904caf 100644
--- a/insanity/monitor.py
+++ b/insanity/monitor.py
@@ -140,6 +140,13 @@ class Monitor(object):
         """
         return True
 
+    def prepareTearDown(self):
+        """
+        This method is called between the last iteration call of stop and finallize
+        the teardown with tearDown
+        """
+        pass
+
     def tearDown(self):
         """
         Clean up the monitor.
diff --git a/insanity/test.py b/insanity/test.py
index 57f13b7..c0e06b6 100644
--- a/insanity/test.py
+++ b/insanity/test.py
@@ -394,6 +394,11 @@ class Test(gobject.GObject):
 
         Your teardown MUST happen in a synchronous fashion.
         """
+
+        # Tell monitors that we are starting the teardown process
+        for instance in self._monitorinstances:
+            instance.prepareTearDown()
+
         self.tearDownVmethod()
 
         stoptime = time.time()