diff options
| author | Vegard Stikbakke <vegard.stikbakke@gmail.com> | 2020-01-25 16:44:46 +0100 |
|---|---|---|
| committer | Chris Withers <chris@withers.org> | 2020-01-25 15:44:46 +0000 |
| commit | aef7dc89879d099dc704bd8037b8a7686fb72838 (patch) | |
| tree | 90685ced511b41f37ea6fc5d6d4eb12f2b7c6575 /Lib/unittest/test/testmock/testmock.py | |
| parent | 9bfb4a7061a3bc4fc5632bccfdf9ed61f62679f7 (diff) | |
| download | cpython-git-aef7dc89879d099dc704bd8037b8a7686fb72838.tar.gz | |
bpo-38932: Mock fully resets child objects on reset_mock(). (GH-17409)
Diffstat (limited to 'Lib/unittest/test/testmock/testmock.py')
| -rw-r--r-- | Lib/unittest/test/testmock/testmock.py | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/Lib/unittest/test/testmock/testmock.py b/Lib/unittest/test/testmock/testmock.py index 1030d12323..1329346ae7 100644 --- a/Lib/unittest/test/testmock/testmock.py +++ b/Lib/unittest/test/testmock/testmock.py @@ -1636,11 +1636,23 @@ class MockTest(unittest.TestCase): self.assertNotEqual(m.side_effect, None) def test_reset_sideeffect(self): - m = Mock(return_value=10, side_effect=[2,3]) + m = Mock(return_value=10, side_effect=[2, 3]) m.reset_mock(side_effect=True) self.assertEqual(m.return_value, 10) self.assertEqual(m.side_effect, None) + def test_reset_return_with_children(self): + m = MagicMock(f=MagicMock(return_value=1)) + self.assertEqual(m.f(), 1) + m.reset_mock(return_value=True) + self.assertNotEqual(m.f(), 1) + + def test_reset_return_with_children_side_effect(self): + m = MagicMock(f=MagicMock(side_effect=[2, 3])) + self.assertNotEqual(m.f.side_effect, None) + m.reset_mock(side_effect=True) + self.assertEqual(m.f.side_effect, None) + def test_mock_add_spec(self): class _One(object): one = 1 |
