mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-04 00:48:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Issue #21271: Adds new keyword only parameters in reset_mock call

Browse source 
We now have two keyword only parameters in the reset_mock function to
selectively reset the return_value or the side_effects, or both.
This commit is contained in:
Kushal Das 2016-06-02 10:20:16 -07:00
parent e2e71685f3
commit 9cd39a170b
4 changed files with 36 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

18
Lib/unittest/test/testmock/testmock.py
View file

@ -1277,6 +1277,24 @@ class MockTest(unittest.TestCase):
self.assertEqual(m.method_calls[0], c)
self.assertEqual(m.method_calls[1], i)
def test_reset_return_sideeffect(self):
m = Mock(return_value=10, side_effect=[2,3])
m.reset_mock(return_value=True, side_effect=True)
self.assertIsInstance(m.return_value, Mock)
self.assertEqual(m.side_effect, None)
def test_reset_return(self):
m = Mock(return_value=10, side_effect=[2,3])
m.reset_mock(return_value=True)
self.assertIsInstance(m.return_value, Mock)
self.assertNotEqual(m.side_effect, None)
def test_reset_sideeffect(self):
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_mock_add_spec(self):
class _One(object):
one = 1