diff options
author | Gustavo Niemeyer <gustavo@niemeyer.net> | 2008-07-01 14:41:32 -0300 |
---|---|---|
committer | Gustavo Niemeyer <gustavo@niemeyer.net> | 2008-07-01 14:41:32 -0300 |
commit | 8a4227cfcc32fed041462ebf6031d722bb0c39a1 (patch) | |
tree | 3b11999909a02ecbf4b30262d0f007488ece1620 /test.py | |
parent | 9aef58ec39a9395d7427aa3b938e96b9d021467a (diff) | |
download | mocker-8a4227cfcc32fed041462ebf6031d722bb0c39a1.tar.gz |
Moved the __mocker_mock__ cleaning up test from Patcher into its
own method around the Mocker.patch() tests.
Diffstat (limited to 'test.py')
-rwxr-xr-x | test.py | 10 |
1 files changed, 8 insertions, 2 deletions
@@ -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" |