diff options
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  | 
