MockerTestCase now verifies if the mocker is put in replay

mode in cases where events were recorded.

1 files changed, 7 insertions, 1 deletions

diff --git a/mocker.py b/mocker.py
index 79f7191..e146ae4 100644
--- a/mocker.py
+++ b/mocker.py
@@ -98,7 +98,13 @@ class MockerTestCase(unittest.TestCase):
         if test_method is not None:
             def test_method_wrapper():
                 try:
-                    result = test_method()
+                    try:
+                        result = test_method()
+                    finally:
+                        if (self.mocker.is_recording() and
+                            self.mocker.get_events()):
+                            raise RuntimeError("Mocker must be put in replay "
+                                               "mode with self.mocker.replay()")
                 except:
                     self.__cleanup()
                     raise