From 8a4227cfcc32fed041462ebf6031d722bb0c39a1 Mon Sep 17 00:00:00 2001 From: Gustavo Niemeyer Date: Tue, 1 Jul 2008 14:41:32 -0300 Subject: Moved the __mocker_mock__ cleaning up test from Patcher into its own method around the Mocker.patch() tests. --- test.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'test.py') diff --git a/test.py b/test.py index 43b0a7b..e98acb8 100755 --- a/test.py +++ b/test.py @@ -1756,6 +1756,14 @@ class MockerTest(TestCase): mock = self.mocker.patch(C, spec=None) self.assertEquals(mock.__mocker_spec__, None) + def test_patch_and_restore_unsets_mocker_mock(self): + class C(object): pass + mock = self.mocker.patch(C) + self.mocker.replay() + self.assertTrue("__mocker_mock__" in C.__dict__) + self.mocker.restore() + self.assertFalse("__mocker_mock__" in C.__dict__) + class ActionTest(TestCase): @@ -3947,8 +3955,6 @@ class PatcherTest(TestCase): self.mocker.replay() self.assertEquals(self.C().method(), "mocked") self.assertRaises(AssertionError, self.C().method) - self.mocker.restore() - self.assertEquals(getattr(self.C, "__mocker_mock__", None), None) def test_recorder_instance_getattr(self): self.C.attr = "original" -- cgit v1.2.1