From dd13d97f602ebb08b0357701a6b22a7441c4abea Mon Sep 17 00:00:00 2001 From: Gustavo Niemeyer Date: Tue, 4 Dec 2007 23:47:03 -0200 Subject: MockerTestCase now verifies if the mocker is put in replay mode in cases where events were recorded. --- mocker.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'mocker.py') 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 -- cgit v1.2.1