changeset:   100773:dcd3b078ab84
branch:      3.5
parent:      100771:02efd708c5c0
user:        Berker Peksag <berker.peksag@gmail.com>
date:        Mon Mar 28 00:30:02 2016 +0300
files:       Lib/unittest/mock.py Lib/unittest/test/testmock/testmock.py Misc/NEWS
description:
Issue #25195: Fix a regression in mock.MagicMock

_Call is a subclass of tuple (changeset 3603bae63c13 only works
for classes) so we need to implement __ne__ ourselves.

Patch by Andrew Plummer.


diff -r 02efd708c5c0 -r dcd3b078ab84 Lib/unittest/mock.py
--- a/Lib/unittest/mock.py	Sun Mar 27 22:39:14 2016 +0300
+++ b/Lib/unittest/mock.py	Mon Mar 28 00:30:02 2016 +0300
@@ -2029,6 +2029,9 @@
         return (other_args, other_kwargs) == (self_args, self_kwargs)
 
 
+    __ne__ = object.__ne__
+
+
     def __call__(self, *args, **kwargs):
         if self.name is None:
             return _Call(('', args, kwargs), name='()')
diff -r 02efd708c5c0 -r dcd3b078ab84 Lib/unittest/test/testmock/testmock.py
--- a/Lib/unittest/test/testmock/testmock.py	Sun Mar 27 22:39:14 2016 +0300
+++ b/Lib/unittest/test/testmock/testmock.py	Mon Mar 28 00:30:02 2016 +0300
@@ -304,6 +304,17 @@
         # an exception. See issue 24857.
         self.assertFalse(mock.call_args == "a long sequence")
 
+
+    def test_calls_equal_with_any(self):
+        call1 = mock.call(mock.MagicMock())
+        call2 = mock.call(mock.ANY)
+
+        # Check that equality and non-equality is consistent even when
+        # comparing with mock.ANY
+        self.assertTrue(call1 == call2)
+        self.assertFalse(call1 != call2)
+
+
     def test_assert_called_with(self):
         mock = Mock()
         mock()
@@ -319,6 +330,12 @@
         mock.assert_called_with(1, 2, 3, a='fish', b='nothing')
 
 
+    def test_assert_called_with_any(self):
+        m = MagicMock()
+        m(MagicMock())
+        m.assert_called_with(mock.ANY)
+
+
     def test_assert_called_with_function_spec(self):
         def f(a, b, c, d=None):
             pass
diff -r 02efd708c5c0 -r dcd3b078ab84 Misc/NEWS
--- a/Misc/NEWS	Sun Mar 27 22:39:14 2016 +0300
+++ b/Misc/NEWS	Mon Mar 28 00:30:02 2016 +0300
@@ -94,6 +94,10 @@
 Library
 -------
 
+- Issue #25195: Fix a regression in mock.MagicMock. _Call is a subclass of
+  tuple (changeset 3603bae63c13 only works for classes) so we need to
+  implement __ne__ ourselves.  Patch by Andrew Plummer.
+
 - Issue #26644: Raise ValueError rather than SystemError when a negative
   length is passed to SSLSocket.recv() or read().