mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-07-19 17:25:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8

Closes #21270 : We now override tuple methods in mock.call objects.

Browse source
This commit is contained in:
Kushal Das 2014-09-16 18:33:37 +05:30
parent a0f33759fa
commit a37b958d65
3 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
Lib/unittest/mock.py
View file

@ -2035,6 +2035,12 @@ class _Call(tuple):
return _Call(name=name, parent=self, from_kall=False)
def count(self, *args, **kwargs):
return self.__getattr__('count')(*args, **kwargs)
def index(self, *args, **kwargs):
return self.__getattr__('index')(*args, **kwargs)
def __repr__(self):
if not self.from_kall:
name = self.name or 'call'

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

@ -1213,6 +1213,16 @@ class MockTest(unittest.TestCase):
text = "call(daddy='hero', name='hello')"
self.assertEqual(repr(m.hello.call_args), text)
#Issue21270 overrides tuple methods for mock.call objects
def test_override_tuple_methods(self):
c = call.count()
i = call.index(132,'hello')
m = Mock()
m.count()
m.index(132,"hello")
self.assertEqual(m.method_calls[0], c)
self.assertEqual(m.method_calls[1], i)
def test_mock_add_spec(self):
class _One(object):
one = 1