summaryrefslogtreecommitdiff
path: root/mock/tests/testcallable.py
diff options
context:
space:
mode:
Diffstat (limited to 'mock/tests/testcallable.py')
-rw-r--r--mock/tests/testcallable.py11
1 files changed, 9 insertions, 2 deletions
diff --git a/mock/tests/testcallable.py b/mock/tests/testcallable.py
index 41715ed..729947e 100644
--- a/mock/tests/testcallable.py
+++ b/mock/tests/testcallable.py
@@ -8,8 +8,9 @@ from mock.tests.support import is_instance, X, SomeClass
from mock import (
Mock, MagicMock, NonCallableMagicMock,
NonCallableMock, patch, create_autospec,
+ CallableMixin
)
-from mock.mock import CallableMixin
+
class TestCallable(unittest.TestCase):
@@ -105,8 +106,14 @@ class TestCallable(unittest.TestCase):
class Multi(SomeClass, Sub):
pass
+ class OldStyle:
+ def __call__(self): pass
+
+ class OldStyleSub(OldStyle):
+ pass
+
for arg in 'spec', 'spec_set':
- for Klass in CallableX, Sub, Multi:
+ for Klass in CallableX, Sub, Multi, OldStyle, OldStyleSub:
with patch('%s.X' % __name__, **{arg: Klass}) as mock:
instance = mock()
mock.assert_called_once_with()