Mocker.on_restore() removed. Restore isn't performed if replay()

isn't called, and that may not be obvious, so a hook won't be
exposed for now.

3 files changed, 5 insertions, 19 deletions

diff --git a/NEWS b/NEWS
index 59de88f..5066402 100644
--- a/NEWS
+++ b/NEWS
@@ -1,10 +1,14 @@
 0.9 (2007-XX-XX)
 ================
 
-- New MockerTestCase.makeFile() and .makeDir() helpers.  They offer
+- Added MockerTestCase.makeFile() and .makeDir() helpers.  They offer
   easy creation of temporary files/directories with certainty of
   removal after tests finish running.
 
+- Mocker.on_restore() removed.  Restore isn't performed if replay()
+  isn't called, and that may not be obvious, so a hook won't be
+  exposed for now.
+
 
 0.8 (2007-11-11)
 ================
diff --git a/mocker.py b/mocker.py
index c7acd26..cb34664 100644
--- a/mocker.py
+++ b/mocker.py
@@ -554,14 +554,6 @@ class MockerBase(object):
         object.__mocker_mock__ = mock
         return mock
 
-    def on_restore(self, callback):
-        """Run C{callback()} when the environment state is restored.
-
-        @param callback: Any callable.
-        """
-        event = self._get_replay_restore_event()
-        event.add_task(OnRestoreCaller(callback))
-
     def act(self, path):
         """This is called by mock objects whenever something happens to them.
 
diff --git a/test.py b/test.py
index db6527d..236aa0c 100755
--- a/test.py
+++ b/test.py
@@ -645,16 +645,6 @@ class MockerTest(unittest.TestCase):
         self.assertTrue(self.mocker.is_recording())
         self.assertEquals(calls, ["replay", "restore"])
 
-    def test_on_restore(self):
-        calls = []
-        self.mocker.on_restore(lambda: calls.append("callback"))
-        self.mocker.restore()
-        self.assertEquals(calls, [])
-        self.mocker.replay()
-        self.mocker.restore()
-        self.mocker.restore()
-        self.assertEquals(calls, ["callback"])
-
     def test_reset(self):
         calls = []
         event = self.mocker.add_event(Event())