Commit a37b958d authored by Kushal Das's avatar Kushal Das

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

parent a0f33759
......@@ -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'
......
......@@ -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
......
......@@ -157,6 +157,9 @@ Library
- Issue #12410: imaplib.IMAP4 now supports the context management protocol.
Original patch by Tarek Ziadé.
- Issue #21270: We now override tuple methods in mock.call objects so that
they can be used as normal call attributes.
- Issue #16662: load_tests() is now unconditionally run when it is present in
a package's __init__.py. TestLoader.loadTestsFromModule() still accepts
use_load_tests, but it is deprecated and ignored. A new keyword-only
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment