mirror of
https://github.com/python/cpython.git
synced 2024-11-24 10:24:35 +08:00
PEP 417: Adding unittest.mock
This commit is contained in:
Michael Foord
parent
8d8f110492
commit
345266aa7e
2151
Lib/unittest/mock.py
Normal file
2151
Lib/unittest/mock.py
Normal file
File diff suppressed because it is too large
Load Diff
@ -14,6 +14,7 @@ def suite():
|
||||
__import__(modname)
|
||||
module = sys.modules[modname]
|
||||
suite.addTest(loader.loadTestsFromModule(module))
|
||||
suite.addTest(loader.loadTestsFromName('unittest.test.testmock'))
|
||||
return suite
|
||||
|
||||
|
||||
|
||||
17
Lib/unittest/test/testmock/__init__.py
Normal file
17
Lib/unittest/test/testmock/__init__.py
Normal file
@ -0,0 +1,17 @@
|
||||
import os
|
||||
import sys
|
||||
import unittest
|
||||
|
||||
|
||||
here = os.path.dirname(__file__)
|
||||
loader = unittest.defaultTestLoader
|
||||
|
||||
def load_tests(*args):
|
||||
suite = unittest.TestSuite()
|
||||
for fn in os.listdir(here):
|
||||
if fn.startswith("test") and fn.endswith(".py"):
|
||||
modname = "unittest.test.testmock." + fn[:-3]
|
||||
__import__(modname)
|
||||
module = sys.modules[modname]
|
||||
suite.addTest(loader.loadTestsFromModule(module))
|
||||
return suite
|
||||
23
Lib/unittest/test/testmock/support.py
Normal file
23
Lib/unittest/test/testmock/support.py
Normal file
@ -0,0 +1,23 @@
|
||||
import sys
|
||||
|
||||
def is_instance(obj, klass):
|
||||
"""Version of is_instance that doesn't access __class__"""
|
||||
return issubclass(type(obj), klass)
|
||||
|
||||
|
||||
class SomeClass(object):
|
||||
class_attribute = None
|
||||
|
||||
def wibble(self):
|
||||
pass
|
||||
|
||||
|
||||
class X(object):
|
||||
pass
|
||||
|
||||
|
||||
def examine_warnings(func):
|
||||
def wrapper():
|
||||
with catch_warnings(record=True) as ws:
|
||||
func(ws)
|
||||
return wrapper
|
||||
159
Lib/unittest/test/testmock/testcallable.py
Normal file
159
Lib/unittest/test/testmock/testcallable.py
Normal file
@ -0,0 +1,159 @@
|
||||
# Copyright (C) 2007-2012 Michael Foord & the mock team
|
||||
# E-mail: fuzzyman AT voidspace DOT org DOT uk
|
||||
# http://www.voidspace.org.uk/python/mock/
|
||||
|
||||
import unittest
|
||||
from unittest.test.testmock.support import is_instance, X, SomeClass
|
||||
|
||||
from unittest.mock import (
|
||||
Mock, MagicMock, NonCallableMagicMock,
|
||||
NonCallableMock, patch, create_autospec,
|
||||
CallableMixin
|
||||
)
|
||||
|
||||
|
||||
|
||||
class TestCallable(unittest.TestCase):
|
||||
|
||||
def assertNotCallable(self, mock):
|
||||
self.assertTrue(is_instance(mock, NonCallableMagicMock))
|
||||
self.assertFalse(is_instance(mock, CallableMixin))
|
||||
|
||||
|
||||
def test_non_callable(self):
|
||||
for mock in NonCallableMagicMock(), NonCallableMock():
|
||||
self.assertRaises(TypeError, mock)
|
||||
self.assertFalse(hasattr(mock, '__call__'))
|
||||
self.assertIn(mock.__class__.__name__, repr(mock))
|
||||
|
||||
|
||||
def test_heirarchy(self):
|
||||
self.assertTrue(issubclass(MagicMock, Mock))
|
||||
self.assertTrue(issubclass(NonCallableMagicMock, NonCallableMock))
|
||||
|
||||
|
||||
def test_attributes(self):
|
||||
one = NonCallableMock()
|
||||
self.assertTrue(issubclass(type(one.one), Mock))
|
||||
|
||||
two = NonCallableMagicMock()
|
||||
self.assertTrue(issubclass(type(two.two), MagicMock))
|
||||
|
||||
|
||||
def test_subclasses(self):
|
||||
class MockSub(Mock):
|
||||
pass
|
||||
|
||||
one = MockSub()
|
||||
self.assertTrue(issubclass(type(one.one), MockSub))
|
||||
|
||||
class MagicSub(MagicMock):
|
||||
pass
|
||||
|
||||
two = MagicSub()
|
||||
self.assertTrue(issubclass(type(two.two), MagicSub))
|
||||
|
||||
|
||||
def test_patch_spec(self):
|
||||
patcher = patch('%s.X' % __name__, spec=True)
|
||||
mock = patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
|
||||
instance = mock()
|
||||
mock.assert_called_once_with()
|
||||
|
||||
self.assertNotCallable(instance)
|
||||
self.assertRaises(TypeError, instance)
|
||||
|
||||
|
||||
def test_patch_spec_set(self):
|
||||
patcher = patch('%s.X' % __name__, spec_set=True)
|
||||
mock = patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
|
||||
instance = mock()
|
||||
mock.assert_called_once_with()
|
||||
|
||||
self.assertNotCallable(instance)
|
||||
self.assertRaises(TypeError, instance)
|
||||
|
||||
|
||||
def test_patch_spec_instance(self):
|
||||
patcher = patch('%s.X' % __name__, spec=X())
|
||||
mock = patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
|
||||
self.assertNotCallable(mock)
|
||||
self.assertRaises(TypeError, mock)
|
||||
|
||||
|
||||
def test_patch_spec_set_instance(self):
|
||||
patcher = patch('%s.X' % __name__, spec_set=X())
|
||||
mock = patcher.start()
|
||||
self.addCleanup(patcher.stop)
|
||||
|
||||
self.assertNotCallable(mock)
|
||||
self.assertRaises(TypeError, mock)
|
||||
|
||||
|
||||
def test_patch_spec_callable_class(self):
|
||||
class CallableX(X):
|
||||
def __call__(self):
|
||||
pass
|
||||
|
||||
class Sub(CallableX):
|
||||
pass
|
||||
|
||||
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, OldStyle, OldStyleSub:
|
||||
patcher = patch('%s.X' % __name__, **{arg: Klass})
|
||||
mock = patcher.start()
|
||||
|
||||
try:
|
||||
instance = mock()
|
||||
mock.assert_called_once_with()
|
||||
|
||||
self.assertTrue(is_instance(instance, MagicMock))
|
||||
# inherited spec
|
||||
self.assertRaises(AttributeError, getattr, instance,
|
||||
'foobarbaz')
|
||||
|
||||
result = instance()
|
||||
# instance is callable, result has no spec
|
||||
instance.assert_called_once_with()
|
||||
|
||||
result(3, 2, 1)
|
||||
result.assert_called_once_with(3, 2, 1)
|
||||
result.foo(3, 2, 1)
|
||||
result.foo.assert_called_once_with(3, 2, 1)
|
||||
finally:
|
||||
patcher.stop()
|
||||
|
||||
|
||||
def test_create_autopsec(self):
|
||||
mock = create_autospec(X)
|
||||
instance = mock()
|
||||
self.assertRaises(TypeError, instance)
|
||||
|
||||
mock = create_autospec(X())
|
||||
self.assertRaises(TypeError, mock)
|
||||
|
||||
|
||||
def test_create_autospec_instance(self):
|
||||
mock = create_autospec(SomeClass, instance=True)
|
||||
|
||||
self.assertRaises(TypeError, mock)
|
||||
mock.wibble()
|
||||
mock.wibble.assert_called_once_with()
|
||||
|
||||
self.assertRaises(TypeError, mock.wibble, 'some', 'args')
|
||||
835
Lib/unittest/test/testmock/testhelpers.py
Normal file
835
Lib/unittest/test/testmock/testhelpers.py
Normal file
@ -0,0 +1,835 @@
|
||||
import unittest
|
||||
|
||||
from unittest.mock import (
|
||||
call, _Call, create_autospec, MagicMock,
|
||||
Mock, ANY, _CallList, patch, PropertyMock
|
||||
)
|
||||
|
||||
from datetime import datetime
|
||||
|
||||
class SomeClass(object):
|
||||
def one(self, a, b):
|
||||
pass
|
||||
def two(self):
|
||||
pass
|
||||
def three(self, a=None):
|
||||
pass
|
||||
|
||||
|
||||
|
||||
class AnyTest(unittest.TestCase):
|
||||
|
||||
def test_any(self):
|
||||
self.assertEqual(ANY, object())
|
||||
|
||||
mock = Mock()
|
||||
mock(ANY)
|
||||
mock.assert_called_with(ANY)
|
||||
|
||||
mock = Mock()
|
||||
mock(foo=ANY)
|
||||
mock.assert_called_with(foo=ANY)
|
||||
|
||||
def test_repr(self):
|
||||
self.assertEqual(repr(ANY), '<ANY>')
|
||||
self.assertEqual(str(ANY), '<ANY>')
|
||||
|
||||
|
||||
def test_any_and_datetime(self):
|
||||
mock = Mock()
|
||||
mock(datetime.now(), foo=datetime.now())
|
||||
|
||||
mock.assert_called_with(ANY, foo=ANY)
|
||||
|
||||
|
||||
def test_any_mock_calls_comparison_order(self):
|
||||
mock = Mock()
|
||||
d = datetime.now()
|
||||
class Foo(object):
|
||||
def __eq__(self, other):
|
||||
return False
|
||||
def __ne__(self, other):
|
||||
return True
|
||||
|
||||
for d in datetime.now(), Foo():
|
||||
mock.reset_mock()
|
||||
|
||||
mock(d, foo=d, bar=d)
|
||||
mock.method(d, zinga=d, alpha=d)
|
||||
mock().method(a1=d, z99=d)
|
||||
|
||||
expected = [
|
||||
call(ANY, foo=ANY, bar=ANY),
|
||||
call.method(ANY, zinga=ANY, alpha=ANY),
|
||||
call(), call().method(a1=ANY, z99=ANY)
|
||||
]
|
||||
self.assertEqual(expected, mock.mock_calls)
|
||||
self.assertEqual(mock.mock_calls, expected)
|
||||
|
||||
|
||||
|
||||
class CallTest(unittest.TestCase):
|
||||
|
||||
def test_call_with_call(self):
|
||||
kall = _Call()
|
||||
self.assertEqual(kall, _Call())
|
||||
self.assertEqual(kall, _Call(('',)))
|
||||
self.assertEqual(kall, _Call(((),)))
|
||||
self.assertEqual(kall, _Call(({},)))
|
||||
self.assertEqual(kall, _Call(('', ())))
|
||||
self.assertEqual(kall, _Call(('', {})))
|
||||
self.assertEqual(kall, _Call(('', (), {})))
|
||||
self.assertEqual(kall, _Call(('foo',)))
|
||||
self.assertEqual(kall, _Call(('bar', ())))
|
||||
self.assertEqual(kall, _Call(('baz', {})))
|
||||
self.assertEqual(kall, _Call(('spam', (), {})))
|
||||
|
||||
kall = _Call(((1, 2, 3),))
|
||||
self.assertEqual(kall, _Call(((1, 2, 3),)))
|
||||
self.assertEqual(kall, _Call(('', (1, 2, 3))))
|
||||
self.assertEqual(kall, _Call(((1, 2, 3), {})))
|
||||
self.assertEqual(kall, _Call(('', (1, 2, 3), {})))
|
||||
|
||||
kall = _Call(((1, 2, 4),))
|
||||
self.assertNotEqual(kall, _Call(('', (1, 2, 3))))
|
||||
self.assertNotEqual(kall, _Call(('', (1, 2, 3), {})))
|
||||
|
||||
kall = _Call(('foo', (1, 2, 4),))
|
||||
self.assertNotEqual(kall, _Call(('', (1, 2, 4))))
|
||||
self.assertNotEqual(kall, _Call(('', (1, 2, 4), {})))
|
||||
self.assertNotEqual(kall, _Call(('bar', (1, 2, 4))))
|
||||
self.assertNotEqual(kall, _Call(('bar', (1, 2, 4), {})))
|
||||
|
||||
kall = _Call(({'a': 3},))
|
||||
self.assertEqual(kall, _Call(('', (), {'a': 3})))
|
||||
self.assertEqual(kall, _Call(('', {'a': 3})))
|
||||
self.assertEqual(kall, _Call(((), {'a': 3})))
|
||||
self.assertEqual(kall, _Call(({'a': 3},)))
|
||||
|
||||
|
||||
def test_empty__Call(self):
|
||||
args = _Call()
|
||||
|
||||
self.assertEqual(args, ())
|
||||
self.assertEqual(args, ('foo',))
|
||||
self.assertEqual(args, ((),))
|
||||
self.assertEqual(args, ('foo', ()))
|
||||
self.assertEqual(args, ('foo',(), {}))
|
||||
self.assertEqual(args, ('foo', {}))
|
||||
self.assertEqual(args, ({},))
|
||||
|
||||
|
||||
def test_named_empty_call(self):
|
||||
args = _Call(('foo', (), {}))
|
||||
|
||||
self.assertEqual(args, ('foo',))
|
||||
self.assertEqual(args, ('foo', ()))
|
||||
self.assertEqual(args, ('foo',(), {}))
|
||||
self.assertEqual(args, ('foo', {}))
|
||||
|
||||
self.assertNotEqual(args, ((),))
|
||||
self.assertNotEqual(args, ())
|
||||
self.assertNotEqual(args, ({},))
|
||||
self.assertNotEqual(args, ('bar',))
|
||||
self.assertNotEqual(args, ('bar', ()))
|
||||
self.assertNotEqual(args, ('bar', {}))
|
||||
|
||||
|
||||
def test_call_with_args(self):
|
||||
args = _Call(((1, 2, 3), {}))
|
||||
|
||||
self.assertEqual(args, ((1, 2, 3),))
|
||||
self.assertEqual(args, ('foo', (1, 2, 3)))
|
||||
self.assertEqual(args, ('foo', (1, 2, 3), {}))
|
||||
self.assertEqual(args, ((1, 2, 3), {}))
|
||||
|
||||
|
||||
def test_named_call_with_args(self):
|
||||
args = _Call(('foo', (1, 2, 3), {}))
|
||||
|
||||
self.assertEqual(args, ('foo', (1, 2, 3)))
|
||||
self.assertEqual(args, ('foo', (1, 2, 3), {}))
|
||||
|
||||
self.assertNotEqual(args, ((1, 2, 3),))
|
||||
self.assertNotEqual(args, ((1, 2, 3), {}))
|
||||
|
||||
|
||||
def test_call_with_kwargs(self):
|
||||
args = _Call(((), dict(a=3, b=4)))
|
||||
|
||||
self.assertEqual(args, (dict(a=3, b=4),))
|
||||
self.assertEqual(args, ('foo', dict(a=3, b=4)))
|
||||
self.assertEqual(args, ('foo', (), dict(a=3, b=4)))
|
||||
self.assertEqual(args, ((), dict(a=3, b=4)))
|
||||
|
||||
|
||||
def test_named_call_with_kwargs(self):
|
||||
args = _Call(('foo', (), dict(a=3, b=4)))
|
||||
|
||||
self.assertEqual(args, ('foo', dict(a=3, b=4)))
|
||||
self.assertEqual(args, ('foo', (), dict(a=3, b=4)))
|
||||
|
||||
self.assertNotEqual(args, (dict(a=3, b=4),))
|
||||
self.assertNotEqual(args, ((), dict(a=3, b=4)))
|
||||
|
||||
|
||||
def test_call_with_args_call_empty_name(self):
|
||||
args = _Call(((1, 2, 3), {}))
|
||||
self.assertEqual(args, call(1, 2, 3))
|
||||
self.assertEqual(call(1, 2, 3), args)
|
||||
self.assertTrue(call(1, 2, 3) in [args])
|
||||
|
||||
|
||||
def test_call_ne(self):
|
||||
self.assertNotEqual(_Call(((1, 2, 3),)), call(1, 2))
|
||||
self.assertFalse(_Call(((1, 2, 3),)) != call(1, 2, 3))
|
||||
self.assertTrue(_Call(((1, 2), {})) != call(1, 2, 3))
|
||||
|
||||
|
||||
def test_call_non_tuples(self):
|
||||
kall = _Call(((1, 2, 3),))
|
||||
for value in 1, None, self, int:
|
||||
self.assertNotEqual(kall, value)
|
||||
self.assertFalse(kall == value)
|
||||
|
||||
|
||||
def test_repr(self):
|
||||
self.assertEqual(repr(_Call()), 'call()')
|
||||
self.assertEqual(repr(_Call(('foo',))), 'call.foo()')
|
||||
|
||||
self.assertEqual(repr(_Call(((1, 2, 3), {'a': 'b'}))),
|
||||
"call(1, 2, 3, a='b')")
|
||||
self.assertEqual(repr(_Call(('bar', (1, 2, 3), {'a': 'b'}))),
|
||||
"call.bar(1, 2, 3, a='b')")
|
||||
|
||||
self.assertEqual(repr(call), 'call')
|
||||
self.assertEqual(str(call), 'call')
|
||||
|
||||
self.assertEqual(repr(call()), 'call()')
|
||||
self.assertEqual(repr(call(1)), 'call(1)')
|
||||
self.assertEqual(repr(call(zz='thing')), "call(zz='thing')")
|
||||
|
||||
self.assertEqual(repr(call().foo), 'call().foo')
|
||||
self.assertEqual(repr(call(1).foo.bar(a=3).bing),
|
||||
'call().foo.bar().bing')
|
||||
self.assertEqual(
|
||||
repr(call().foo(1, 2, a=3)),
|
||||
"call().foo(1, 2, a=3)"
|
||||
)
|
||||
self.assertEqual(repr(call()()), "call()()")
|
||||
self.assertEqual(repr(call(1)(2)), "call()(2)")
|
||||
self.assertEqual(
|
||||
repr(call()().bar().baz.beep(1)),
|
||||
"call()().bar().baz.beep(1)"
|
||||
)
|
||||
|
||||
|
||||
def test_call(self):
|
||||
self.assertEqual(call(), ('', (), {}))
|
||||
self.assertEqual(call('foo', 'bar', one=3, two=4),
|
||||
('', ('foo', 'bar'), {'one': 3, 'two': 4}))
|
||||
|
||||
mock = Mock()
|
||||
mock(1, 2, 3)
|
||||
mock(a=3, b=6)
|
||||
self.assertEqual(mock.call_args_list,
|
||||
[call(1, 2, 3), call(a=3, b=6)])
|
||||
|
||||
def test_attribute_call(self):
|
||||
self.assertEqual(call.foo(1), ('foo', (1,), {}))
|
||||
self.assertEqual(call.bar.baz(fish='eggs'),
|
||||
('bar.baz', (), {'fish': 'eggs'}))
|
||||
|
||||
mock = Mock()
|
||||
mock.foo(1, 2 ,3)
|
||||
mock.bar.baz(a=3, b=6)
|
||||
self.assertEqual(mock.method_calls,
|
||||
[call.foo(1, 2, 3), call.bar.baz(a=3, b=6)])
|
||||
|
||||
|
||||
def test_extended_call(self):
|
||||
result = call(1).foo(2).bar(3, a=4)
|
||||
self.assertEqual(result, ('().foo().bar', (3,), dict(a=4)))
|
||||
|
||||
mock = MagicMock()
|
||||
mock(1, 2, a=3, b=4)
|
||||
self.assertEqual(mock.call_args, call(1, 2, a=3, b=4))
|
||||
self.assertNotEqual(mock.call_args, call(1, 2, 3))
|
||||
|
||||
self.assertEqual(mock.call_args_list, [call(1, 2, a=3, b=4)])
|
||||
self.assertEqual(mock.mock_calls, [call(1, 2, a=3, b=4)])
|
||||
|
||||
mock = MagicMock()
|
||||
mock.foo(1).bar()().baz.beep(a=6)
|
||||
|
||||
last_call = call.foo(1).bar()().baz.beep(a=6)
|
||||
self.assertEqual(mock.mock_calls[-1], last_call)
|
||||
self.assertEqual(mock.mock_calls, last_call.call_list())
|
||||
|
||||
|
||||
def test_call_list(self):
|
||||
mock = MagicMock()
|
||||
mock(1)
|
||||
self.assertEqual(call(1).call_list(), mock.mock_calls)
|
||||
|
||||
mock = MagicMock()
|
||||
mock(1).method(2)
|
||||
self.assertEqual(call(1).method(2).call_list(),
|
||||
mock.mock_calls)
|
||||
|
||||
mock = MagicMock()
|
||||
mock(1).method(2)(3)
|
||||
self.assertEqual(call(1).method(2)(3).call_list(),
|
||||
mock.mock_calls)
|
||||
|
||||
mock = MagicMock()
|
||||
int(mock(1).method(2)(3).foo.bar.baz(4)(5))
|
||||
kall = call(1).method(2)(3).foo.bar.baz(4)(5).__int__()
|
||||
self.assertEqual(kall.call_list(), mock.mock_calls)
|
||||
|
||||
|
||||
def test_call_any(self):
|
||||
self.assertEqual(call, ANY)
|
||||
|
||||
m = MagicMock()
|
||||
int(m)
|
||||
self.assertEqual(m.mock_calls, [ANY])
|
||||
self.assertEqual([ANY], m.mock_calls)
|
||||
|
||||
|
||||
def test_two_args_call(self):
|
||||
args = _Call(((1, 2), {'a': 3}), two=True)
|
||||
self.assertEqual(len(args), 2)
|
||||
self.assertEqual(args[0], (1, 2))
|
||||
self.assertEqual(args[1], {'a': 3})
|
||||
|
||||
other_args = _Call(((1, 2), {'a': 3}))
|
||||
self.assertEqual(args, other_args)
|
||||
|
||||
|
||||
class SpecSignatureTest(unittest.TestCase):
|
||||
|
||||
def _check_someclass_mock(self, mock):
|
||||
self.assertRaises(AttributeError, getattr, mock, 'foo')
|
||||
mock.one(1, 2)
|
||||
mock.one.assert_called_with(1, 2)
|
||||
self.assertRaises(AssertionError,
|
||||
mock.one.assert_called_with, 3, 4)
|
||||
self.assertRaises(TypeError, mock.one, 1)
|
||||
|
||||
mock.two()
|
||||
mock.two.assert_called_with()
|
||||
self.assertRaises(AssertionError,
|
||||
mock.two.assert_called_with, 3)
|
||||
self.assertRaises(TypeError, mock.two, 1)
|
||||
|
||||
mock.three()
|
||||
mock.three.assert_called_with()
|
||||
self.assertRaises(AssertionError,
|
||||
mock.three.assert_called_with, 3)
|
||||
self.assertRaises(TypeError, mock.three, 3, 2)
|
||||
|
||||
mock.three(1)
|
||||
mock.three.assert_called_with(1)
|
||||
|
||||
mock.three(a=1)
|
||||
mock.three.assert_called_with(a=1)
|
||||
|
||||
|
||||
def test_basic(self):
|
||||
for spec in (SomeClass, SomeClass()):
|
||||
mock = create_autospec(spec)
|
||||
self._check_someclass_mock(mock)
|
||||
|
||||
|
||||
def test_create_autospec_return_value(self):
|
||||
def f():
|
||||
pass
|
||||
mock = create_autospec(f, return_value='foo')
|
||||
self.assertEqual(mock(), 'foo')
|
||||
|
||||
class Foo(object):
|
||||
pass
|
||||
|
||||
mock = create_autospec(Foo, return_value='foo')
|
||||
self.assertEqual(mock(), 'foo')
|
||||
|
||||
|
||||
def test_mocking_unbound_methods(self):
|
||||
class Foo(object):
|
||||
def foo(self, foo):
|
||||
pass
|
||||
p = patch.object(Foo, 'foo')
|
||||
mock_foo = p.start()
|
||||
Foo().foo(1)
|
||||
|
||||
mock_foo.assert_called_with(1)
|
||||
|
||||
|
||||
def test_create_autospec_unbound_methods(self):
|
||||
# see issue 128
|
||||
# this is expected to fail until the issue is fixed
|
||||
return
|
||||
class Foo(object):
|
||||
def foo(self):
|
||||
pass
|
||||
|
||||
klass = create_autospec(Foo)
|
||||
instance = klass()
|
||||
self.assertRaises(TypeError, instance.foo, 1)
|
||||
|
||||
# Note: no type checking on the "self" parameter
|
||||
klass.foo(1)
|
||||
klass.foo.assert_called_with(1)
|
||||
self.assertRaises(TypeError, klass.foo)
|
||||
|
||||
|
||||
def test_create_autospec_keyword_arguments(self):
|
||||
class Foo(object):
|
||||
a = 3
|
||||
m = create_autospec(Foo, a='3')
|
||||
self.assertEqual(m.a, '3')
|
||||
|
||||
|
||||
def test_function_as_instance_attribute(self):
|
||||
obj = SomeClass()
|
||||
def f(a):
|
||||
pass
|
||||
obj.f = f
|
||||
|
||||
mock = create_autospec(obj)
|
||||
mock.f('bing')
|
||||
mock.f.assert_called_with('bing')
|
||||
|
||||
|
||||
def test_spec_as_list(self):
|
||||
# because spec as a list of strings in the mock constructor means
|
||||
# something very different we treat a list instance as the type.
|
||||
mock = create_autospec([])
|
||||
mock.append('foo')
|
||||
mock.append.assert_called_with('foo')
|
||||
|
||||
self.assertRaises(AttributeError, getattr, mock, 'foo')
|
||||
|
||||
class Foo(object):
|
||||
foo = []
|
||||
|
||||
mock = create_autospec(Foo)
|
||||
mock.foo.append(3)
|
||||
mock.foo.append.assert_called_with(3)
|
||||
self.assertRaises(AttributeError, getattr, mock.foo, 'foo')
|
||||
|
||||
|
||||
def test_attributes(self):
|
||||
class Sub(SomeClass):
|
||||
attr = SomeClass()
|
||||
|
||||
sub_mock = create_autospec(Sub)
|
||||
|
||||
for mock in (sub_mock, sub_mock.attr):
|
||||
self._check_someclass_mock(mock)
|
||||
|
||||
|
||||
def test_builtin_functions_types(self):
|
||||
# we could replace builtin functions / methods with a function
|
||||
# with *args / **kwargs signature. Using the builtin method type
|
||||
# as a spec seems to work fairly well though.
|
||||
class BuiltinSubclass(list):
|
||||
def bar(self, arg):
|
||||
pass
|
||||
sorted = sorted
|
||||
attr = {}
|
||||
|
||||
mock = create_autospec(BuiltinSubclass)
|
||||
mock.append(3)
|
||||
mock.append.assert_called_with(3)
|
||||
self.assertRaises(AttributeError, getattr, mock.append, 'foo')
|
||||
|
||||
mock.bar('foo')
|
||||
mock.bar.assert_called_with('foo')
|
||||
self.assertRaises(TypeError, mock.bar, 'foo', 'bar')
|
||||
self.assertRaises(AttributeError, getattr, mock.bar, 'foo')
|
||||
|
||||
mock.sorted([1, 2])
|
||||
mock.sorted.assert_called_with([1, 2])
|
||||
self.assertRaises(AttributeError, getattr, mock.sorted, 'foo')
|
||||
|
||||
mock.attr.pop(3)
|
||||
mock.attr.pop.assert_called_with(3)
|
||||
self.assertRaises(AttributeError, getattr, mock.attr, 'foo')
|
||||
|
||||
|
||||
def test_method_calls(self):
|
||||
class Sub(SomeClass):
|
||||
attr = SomeClass()
|
||||
|
||||
mock = create_autospec(Sub)
|
||||
mock.one(1, 2)
|
||||
mock.two()
|
||||
mock.three(3)
|
||||
|
||||
expected = [call.one(1, 2), call.two(), call.three(3)]
|
||||
self.assertEqual(mock.method_calls, expected)
|
||||
|
||||
mock.attr.one(1, 2)
|
||||
mock.attr.two()
|
||||
mock.attr.three(3)
|
||||
|
||||
expected.extend(
|
||||
[call.attr.one(1, 2), call.attr.two(), call.attr.three(3)]
|
||||
)
|
||||
self.assertEqual(mock.method_calls, expected)
|
||||
|
||||
|
||||
def test_magic_methods(self):
|
||||
class BuiltinSubclass(list):
|
||||
attr = {}
|
||||
|
||||
mock = create_autospec(BuiltinSubclass)
|
||||
self.assertEqual(list(mock), [])
|
||||
self.assertRaises(TypeError, int, mock)
|
||||
self.assertRaises(TypeError, int, mock.attr)
|
||||
self.assertEqual(list(mock), [])
|
||||
|
||||
self.assertIsInstance(mock['foo'], MagicMock)
|
||||
self.assertIsInstance(mock.attr['foo'], MagicMock)
|
||||
|
||||
|
||||
def test_spec_set(self):
|
||||
class Sub(SomeClass):
|
||||
attr = SomeClass()
|
||||
|
||||
for spec in (Sub, Sub()):
|
||||
mock = create_autospec(spec, spec_set=True)
|
||||
self._check_someclass_mock(mock)
|
||||
|
||||
self.assertRaises(AttributeError, setattr, mock, 'foo', 'bar')
|
||||
self.assertRaises(AttributeError, setattr, mock.attr, 'foo', 'bar')
|
||||
|
||||
|
||||
def test_descriptors(self):
|
||||
class Foo(object):
|
||||
@classmethod
|
||||
def f(cls, a, b):
|
||||
pass
|
||||
@staticmethod
|
||||
def g(a, b):
|
||||
pass
|
||||
|
||||
class Bar(Foo):
|
||||
pass
|
||||
|
||||
class Baz(SomeClass, Bar):
|
||||
pass
|
||||
|
||||
for spec in (Foo, Foo(), Bar, Bar(), Baz, Baz()):
|
||||
mock = create_autospec(spec)
|
||||
mock.f(1, 2)
|
||||
mock.f.assert_called_once_with(1, 2)
|
||||
|
||||
mock.g(3, 4)
|
||||
mock.g.assert_called_once_with(3, 4)
|
||||
|
||||
|
||||
def test_recursive(self):
|
||||
class A(object):
|
||||
def a(self):
|
||||
pass
|
||||
foo = 'foo bar baz'
|
||||
bar = foo
|
||||
|
||||
A.B = A
|
||||
mock = create_autospec(A)
|
||||
|
||||
mock()
|
||||
self.assertFalse(mock.B.called)
|
||||
|
||||
mock.a()
|
||||
mock.B.a()
|
||||
self.assertEqual(mock.method_calls, [call.a(), call.B.a()])
|
||||
|
||||
self.assertIs(A.foo, A.bar)
|
||||
self.assertIsNot(mock.foo, mock.bar)
|
||||
mock.foo.lower()
|
||||
self.assertRaises(AssertionError, mock.bar.lower.assert_called_with)
|
||||
|
||||
|
||||
def test_spec_inheritance_for_classes(self):
|
||||
class Foo(object):
|
||||
def a(self):
|
||||
pass
|
||||
class Bar(object):
|
||||
def f(self):
|
||||
pass
|
||||
|
||||
class_mock = create_autospec(Foo)
|
||||
|
||||
self.assertIsNot(class_mock, class_mock())
|
||||
|
||||
for this_mock in class_mock, class_mock():
|
||||
this_mock.a()
|
||||
this_mock.a.assert_called_with()
|
||||
self.assertRaises(TypeError, this_mock.a, 'foo')
|
||||
self.assertRaises(AttributeError, getattr, this_mock, 'b')
|
||||
|
||||
instance_mock = create_autospec(Foo())
|
||||
instance_mock.a()
|
||||
instance_mock.a.assert_called_with()
|
||||
self.assertRaises(TypeError, instance_mock.a, 'foo')
|
||||
self.assertRaises(AttributeError, getattr, instance_mock, 'b')
|
||||
|
||||
# The return value isn't isn't callable
|
||||
self.assertRaises(TypeError, instance_mock)
|
||||
|
||||
instance_mock.Bar.f()
|
||||
instance_mock.Bar.f.assert_called_with()
|
||||
self.assertRaises(AttributeError, getattr, instance_mock.Bar, 'g')
|
||||
|
||||
instance_mock.Bar().f()
|
||||
instance_mock.Bar().f.assert_called_with()
|
||||
self.assertRaises(AttributeError, getattr, instance_mock.Bar(), 'g')
|
||||
|
||||
|
||||
def test_inherit(self):
|
||||
class Foo(object):
|
||||
a = 3
|
||||
|
||||
Foo.Foo = Foo
|
||||
|
||||
# class
|
||||
mock = create_autospec(Foo)
|
||||
instance = mock()
|
||||
self.assertRaises(AttributeError, getattr, instance, 'b')
|
||||
|
||||
attr_instance = mock.Foo()
|
||||
self.assertRaises(AttributeError, getattr, attr_instance, 'b')
|
||||
|
||||
# instance
|
||||
mock = create_autospec(Foo())
|
||||
self.assertRaises(AttributeError, getattr, mock, 'b')
|
||||
self.assertRaises(TypeError, mock)
|
||||
|
||||
# attribute instance
|
||||
call_result = mock.Foo()
|
||||
self.assertRaises(AttributeError, getattr, call_result, 'b')
|
||||
|
||||
|
||||
def test_builtins(self):
|
||||
# used to fail with infinite recursion
|
||||
create_autospec(1)
|
||||
|
||||
create_autospec(int)
|
||||
create_autospec('foo')
|
||||
create_autospec(str)
|
||||
create_autospec({})
|
||||
create_autospec(dict)
|
||||
create_autospec([])
|
||||
create_autospec(list)
|
||||
create_autospec(set())
|
||||
create_autospec(set)
|
||||
create_autospec(1.0)
|
||||
create_autospec(float)
|
||||
create_autospec(1j)
|
||||
create_autospec(complex)
|
||||
create_autospec(False)
|
||||
create_autospec(True)
|
||||
|
||||
|
||||
def test_function(self):
|
||||
def f(a, b):
|
||||
pass
|
||||
|
||||
mock = create_autospec(f)
|
||||
self.assertRaises(TypeError, mock)
|
||||
mock(1, 2)
|
||||
mock.assert_called_with(1, 2)
|
||||
|
||||
f.f = f
|
||||
mock = create_autospec(f)
|
||||
self.assertRaises(TypeError, mock.f)
|
||||
mock.f(3, 4)
|
||||
mock.f.assert_called_with(3, 4)
|
||||
|
||||
|
||||
def test_signature_class(self):
|
||||
class Foo(object):
|
||||
def __init__(self, a, b=3):
|
||||
pass
|
||||
|
||||
mock = create_autospec(Foo)
|
||||
|
||||
self.assertRaises(TypeError, mock)
|
||||
mock(1)
|
||||
mock.assert_called_once_with(1)
|
||||
|
||||
mock(4, 5)
|
||||
mock.assert_called_with(4, 5)
|
||||
|
||||
|
||||
def test_class_with_no_init(self):
|
||||
# this used to raise an exception
|
||||
# due to trying to get a signature from object.__init__
|
||||
class Foo(object):
|
||||
pass
|
||||
create_autospec(Foo)
|
||||
|
||||
|
||||
def test_signature_callable(self):
|
||||
class Callable(object):
|
||||
def __init__(self):
|
||||
pass
|
||||
def __call__(self, a):
|
||||
pass
|
||||
|
||||
mock = create_autospec(Callable)
|
||||
mock()
|
||||
mock.assert_called_once_with()
|
||||
self.assertRaises(TypeError, mock, 'a')
|
||||
|
||||
instance = mock()
|
||||
self.assertRaises(TypeError, instance)
|
||||
instance(a='a')
|
||||
instance.assert_called_once_with(a='a')
|
||||
instance('a')
|
||||
instance.assert_called_with('a')
|
||||
|
||||
mock = create_autospec(Callable())
|
||||
mock(a='a')
|
||||
mock.assert_called_once_with(a='a')
|
||||
self.assertRaises(TypeError, mock)
|
||||
mock('a')
|
||||
mock.assert_called_with('a')
|
||||
|
||||
|
||||
def test_signature_noncallable(self):
|
||||
class NonCallable(object):
|
||||
def __init__(self):
|
||||
pass
|
||||
|
||||
mock = create_autospec(NonCallable)
|
||||
instance = mock()
|
||||
mock.assert_called_once_with()
|
||||
self.assertRaises(TypeError, mock, 'a')
|
||||
self.assertRaises(TypeError, instance)
|
||||
self.assertRaises(TypeError, instance, 'a')
|
||||
|
||||
mock = create_autospec(NonCallable())
|
||||
self.assertRaises(TypeError, mock)
|
||||
self.assertRaises(TypeError, mock, 'a')
|
||||
|
||||
|
||||
def test_create_autospec_none(self):
|
||||
class Foo(object):
|
||||
bar = None
|
||||
|
||||
mock = create_autospec(Foo)
|
||||
none = mock.bar
|
||||
self.assertNotIsInstance(none, type(None))
|
||||
|
||||
none.foo()
|
||||
none.foo.assert_called_once_with()
|
||||
|
||||
|
||||
def test_autospec_functions_with_self_in_odd_place(self):
|
||||
class Foo(object):
|
||||
def f(a, self):
|
||||
pass
|
||||
|
||||
a = create_autospec(Foo)
|
||||
a.f(self=10)
|
||||
a.f.assert_called_with(self=10)
|
||||
|
||||
|
||||
def test_autospec_property(self):
|
||||
class Foo(object):
|
||||
@property
|
||||
def foo(self):
|
||||
return 3
|
||||
|
||||
foo = create_autospec(Foo)
|
||||
mock_property = foo.foo
|
||||
|
||||
# no spec on properties
|
||||
self.assertTrue(isinstance(mock_property, MagicMock))
|
||||
mock_property(1, 2, 3)
|
||||
mock_property.abc(4, 5, 6)
|
||||
mock_property.assert_called_once_with(1, 2, 3)
|
||||
mock_property.abc.assert_called_once_with(4, 5, 6)
|
||||
|
||||
|
||||
def test_autospec_slots(self):
|
||||
class Foo(object):
|
||||
__slots__ = ['a']
|
||||
|
||||
foo = create_autospec(Foo)
|
||||
mock_slot = foo.a
|
||||
|
||||
# no spec on slots
|
||||
mock_slot(1, 2, 3)
|
||||
mock_slot.abc(4, 5, 6)
|
||||
mock_slot.assert_called_once_with(1, 2, 3)
|
||||
mock_slot.abc.assert_called_once_with(4, 5, 6)
|
||||
|
||||
|
||||
class TestCallList(unittest.TestCase):
|
||||
|
||||
def test_args_list_contains_call_list(self):
|
||||
mock = Mock()
|
||||
self.assertIsInstance(mock.call_args_list, _CallList)
|
||||
|
||||
mock(1, 2)
|
||||
mock(a=3)
|
||||
mock(3, 4)
|
||||
mock(b=6)
|
||||
|
||||
for kall in call(1, 2), call(a=3), call(3, 4), call(b=6):
|
||||
self.assertTrue(kall in mock.call_args_list)
|
||||
|
||||
calls = [call(a=3), call(3, 4)]
|
||||
self.assertTrue(calls in mock.call_args_list)
|
||||
calls = [call(1, 2), call(a=3)]
|
||||
self.assertTrue(calls in mock.call_args_list)
|
||||
calls = [call(3, 4), call(b=6)]
|
||||
self.assertTrue(calls in mock.call_args_list)
|
||||
calls = [call(3, 4)]
|
||||
self.assertTrue(calls in mock.call_args_list)
|
||||
|
||||
self.assertFalse(call('fish') in mock.call_args_list)
|
||||
self.assertFalse([call('fish')] in mock.call_args_list)
|
||||
|
||||
|
||||
def test_call_list_str(self):
|
||||
mock = Mock()
|
||||
mock(1, 2)
|
||||
mock.foo(a=3)
|
||||
mock.foo.bar().baz('fish', cat='dog')
|
||||
|
||||
expected = (
|
||||
"[call(1, 2),\n"
|
||||
" call.foo(a=3),\n"
|
||||
" call.foo.bar(),\n"
|
||||
" call.foo.bar().baz('fish', cat='dog')]"
|
||||
)
|
||||
self.assertEqual(str(mock.mock_calls), expected)
|
||||
|
||||
|
||||
def test_propertymock(self):
|
||||
p = patch('%s.SomeClass.one' % __name__, new_callable=PropertyMock)
|
||||
mock = p.start()
|
||||
try:
|
||||
SomeClass.one
|
||||
mock.assert_called_once_with()
|
||||
|
||||
s = SomeClass()
|
||||
s.one
|
||||
mock.assert_called_with()
|
||||
self.assertEqual(mock.mock_calls, [call(), call()])
|
||||
|
||||
s.one = 3
|
||||
self.assertEqual(mock.mock_calls, [call(), call(), call(3)])
|
||||
finally:
|
||||
p.stop()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
382
Lib/unittest/test/testmock/testmagicmethods.py
Normal file
382
Lib/unittest/test/testmock/testmagicmethods.py
Normal file
@ -0,0 +1,382 @@
|
||||
import unittest
|
||||
import inspect
|
||||
import sys
|
||||
from unittest.mock import Mock, MagicMock, _magics
|
||||
|
||||
|
||||
|
||||
class TestMockingMagicMethods(unittest.TestCase):
|
||||
|
||||
def test_deleting_magic_methods(self):
|
||||
mock = Mock()
|
||||
self.assertFalse(hasattr(mock, '__getitem__'))
|
||||
|
||||
mock.__getitem__ = Mock()
|
||||
self.assertTrue(hasattr(mock, '__getitem__'))
|
||||
|
||||
del mock.__getitem__
|
||||
self.assertFalse(hasattr(mock, '__getitem__'))
|
||||
|
||||
|
||||
def test_magicmock_del(self):
|
||||
mock = MagicMock()
|
||||
# before using getitem
|
||||
del mock.__getitem__
|
||||
self.assertRaises(TypeError, lambda: mock['foo'])
|
||||
|
||||
mock = MagicMock()
|
||||
# this time use it first
|
||||
mock['foo']
|
||||
del mock.__getitem__
|
||||
self.assertRaises(TypeError, lambda: mock['foo'])
|
||||
|
||||
|
||||
def test_magic_method_wrapping(self):
|
||||
mock = Mock()
|
||||
def f(self, name):
|
||||
return self, 'fish'
|
||||
|
||||
mock.__getitem__ = f
|
||||
self.assertFalse(mock.__getitem__ is f)
|
||||
self.assertEqual(mock['foo'], (mock, 'fish'))
|
||||
self.assertEqual(mock.__getitem__('foo'), (mock, 'fish'))
|
||||
|
||||
mock.__getitem__ = mock
|
||||
self.assertTrue(mock.__getitem__ is mock)
|
||||
|
||||
|
||||
def test_magic_methods_isolated_between_mocks(self):
|
||||
mock1 = Mock()
|
||||
mock2 = Mock()
|
||||
|
||||
mock1.__iter__ = Mock(return_value=iter([]))
|
||||
self.assertEqual(list(mock1), [])
|
||||
self.assertRaises(TypeError, lambda: list(mock2))
|
||||
|
||||
|
||||
def test_repr(self):
|
||||
mock = Mock()
|
||||
self.assertEqual(repr(mock), "<Mock id='%s'>" % id(mock))
|
||||
mock.__repr__ = lambda s: 'foo'
|
||||
self.assertEqual(repr(mock), 'foo')
|
||||
|
||||
|
||||
def test_str(self):
|
||||
mock = Mock()
|
||||
self.assertEqual(str(mock), object.__str__(mock))
|
||||
mock.__str__ = lambda s: 'foo'
|
||||
self.assertEqual(str(mock), 'foo')
|
||||
|
||||
|
||||
def test_dict_methods(self):
|
||||
mock = Mock()
|
||||
|
||||
self.assertRaises(TypeError, lambda: mock['foo'])
|
||||
def _del():
|
||||
del mock['foo']
|
||||
def _set():
|
||||
mock['foo'] = 3
|
||||
self.assertRaises(TypeError, _del)
|
||||
self.assertRaises(TypeError, _set)
|
||||
|
||||
_dict = {}
|
||||
def getitem(s, name):
|
||||
return _dict[name]
|
||||
def setitem(s, name, value):
|
||||
_dict[name] = value
|
||||
def delitem(s, name):
|
||||
del _dict[name]
|
||||
|
||||
mock.__setitem__ = setitem
|
||||
mock.__getitem__ = getitem
|
||||
mock.__delitem__ = delitem
|
||||
|
||||
self.assertRaises(KeyError, lambda: mock['foo'])
|
||||
mock['foo'] = 'bar'
|
||||
self.assertEqual(_dict, {'foo': 'bar'})
|
||||
self.assertEqual(mock['foo'], 'bar')
|
||||
del mock['foo']
|
||||
self.assertEqual(_dict, {})
|
||||
|
||||
|
||||
def test_numeric(self):
|
||||
original = mock = Mock()
|
||||
mock.value = 0
|
||||
|
||||
self.assertRaises(TypeError, lambda: mock + 3)
|
||||
|
||||
def add(self, other):
|
||||
mock.value += other
|
||||
return self
|
||||
mock.__add__ = add
|
||||
self.assertEqual(mock + 3, mock)
|
||||
self.assertEqual(mock.value, 3)
|
||||
|
||||
del mock.__add__
|
||||
def iadd(mock):
|
||||
mock += 3
|
||||
self.assertRaises(TypeError, iadd, mock)
|
||||
mock.__iadd__ = add
|
||||
mock += 6
|
||||
self.assertEqual(mock, original)
|
||||
self.assertEqual(mock.value, 9)
|
||||
|
||||
self.assertRaises(TypeError, lambda: 3 + mock)
|
||||
mock.__radd__ = add
|
||||
self.assertEqual(7 + mock, mock)
|
||||
self.assertEqual(mock.value, 16)
|
||||
|
||||
|
||||
def test_hash(self):
|
||||
mock = Mock()
|
||||
# test delegation
|
||||
self.assertEqual(hash(mock), Mock.__hash__(mock))
|
||||
|
||||
def _hash(s):
|
||||
return 3
|
||||
mock.__hash__ = _hash
|
||||
self.assertEqual(hash(mock), 3)
|
||||
|
||||
|
||||
def test_nonzero(self):
|
||||
m = Mock()
|
||||
self.assertTrue(bool(m))
|
||||
|
||||
m.__bool__ = lambda s: False
|
||||
self.assertFalse(bool(m))
|
||||
|
||||
|
||||
def test_comparison(self):
|
||||
mock = Mock()
|
||||
def comp(s, o):
|
||||
return True
|
||||
mock.__lt__ = mock.__gt__ = mock.__le__ = mock.__ge__ = comp
|
||||
self. assertTrue(mock < 3)
|
||||
self. assertTrue(mock > 3)
|
||||
self. assertTrue(mock <= 3)
|
||||
self. assertTrue(mock >= 3)
|
||||
|
||||
|
||||
def test_equality(self):
|
||||
for mock in Mock(), MagicMock():
|
||||
self.assertEqual(mock == mock, True)
|
||||
self.assertIsInstance(mock == mock, bool)
|
||||
self.assertEqual(mock != mock, False)
|
||||
self.assertIsInstance(mock != mock, bool)
|
||||
self.assertEqual(mock == object(), False)
|
||||
self.assertEqual(mock != object(), True)
|
||||
|
||||
def eq(self, other):
|
||||
return other == 3
|
||||
mock.__eq__ = eq
|
||||
self.assertTrue(mock == 3)
|
||||
self.assertFalse(mock == 4)
|
||||
|
||||
def ne(self, other):
|
||||
return other == 3
|
||||
mock.__ne__ = ne
|
||||
self.assertTrue(mock != 3)
|
||||
self.assertFalse(mock != 4)
|
||||
|
||||
mock = MagicMock()
|
||||
mock.__eq__.return_value = True
|
||||
self.assertIsInstance(mock == 3, bool)
|
||||
self.assertEqual(mock == 3, True)
|
||||
|
||||
mock.__ne__.return_value = False
|
||||
self.assertIsInstance(mock != 3, bool)
|
||||
self.assertEqual(mock != 3, False)
|
||||
|
||||
|
||||
def test_len_contains_iter(self):
|
||||
mock = Mock()
|
||||
|
||||
self.assertRaises(TypeError, len, mock)
|
||||
self.assertRaises(TypeError, iter, mock)
|
||||
self.assertRaises(TypeError, lambda: 'foo' in mock)
|
||||
|
||||
mock.__len__ = lambda s: 6
|
||||
self.assertEqual(len(mock), 6)
|
||||
|
||||
mock.__contains__ = lambda s, o: o == 3
|
||||
self.assertTrue(3 in mock)
|
||||
self.assertFalse(6 in mock)
|
||||
|
||||
mock.__iter__ = lambda s: iter('foobarbaz')
|
||||
self.assertEqual(list(mock), list('foobarbaz'))
|
||||
|
||||
|
||||
def test_magicmock(self):
|
||||
mock = MagicMock()
|
||||
|
||||
mock.__iter__.return_value = iter([1, 2, 3])
|
||||
self.assertEqual(list(mock), [1, 2, 3])
|
||||
|
||||
getattr(mock, '__bool__').return_value = False
|
||||
self.assertFalse(hasattr(mock, '__nonzero__'))
|
||||
self.assertFalse(bool(mock))
|
||||
|
||||
for entry in _magics:
|
||||
self.assertTrue(hasattr(mock, entry))
|
||||
self.assertFalse(hasattr(mock, '__imaginery__'))
|
||||
|
||||
|
||||
def test_magic_mock_equality(self):
|
||||
mock = MagicMock()
|
||||
self.assertIsInstance(mock == object(), bool)
|
||||
self.assertIsInstance(mock != object(), bool)
|
||||
|
||||
self.assertEqual(mock == object(), False)
|
||||
self.assertEqual(mock != object(), True)
|
||||
self.assertEqual(mock == mock, True)
|
||||
self.assertEqual(mock != mock, False)
|
||||
|
||||
|
||||
def test_magicmock_defaults(self):
|
||||
mock = MagicMock()
|
||||
self.assertEqual(int(mock), 1)
|
||||
self.assertEqual(complex(mock), 1j)
|
||||
self.assertEqual(float(mock), 1.0)
|
||||
self.assertNotIn(object(), mock)
|
||||
self.assertEqual(len(mock), 0)
|
||||
self.assertEqual(list(mock), [])
|
||||
self.assertEqual(hash(mock), object.__hash__(mock))
|
||||
self.assertEqual(str(mock), object.__str__(mock))
|
||||
self.assertTrue(bool(mock))
|
||||
|
||||
# in Python 3 oct and hex use __index__
|
||||
# so these tests are for __index__ in py3k
|
||||
self.assertEqual(oct(mock), '0o1')
|
||||
self.assertEqual(hex(mock), '0x1')
|
||||
# how to test __sizeof__ ?
|
||||
|
||||
|
||||
def test_magic_methods_and_spec(self):
|
||||
class Iterable(object):
|
||||
def __iter__(self):
|
||||
pass
|
||||
|
||||
mock = Mock(spec=Iterable)
|
||||
self.assertRaises(AttributeError, lambda: mock.__iter__)
|
||||
|
||||
mock.__iter__ = Mock(return_value=iter([]))
|
||||
self.assertEqual(list(mock), [])
|
||||
|
||||
class NonIterable(object):
|
||||
pass
|
||||
mock = Mock(spec=NonIterable)
|
||||
self.assertRaises(AttributeError, lambda: mock.__iter__)
|
||||
|
||||
def set_int():
|
||||
mock.__int__ = Mock(return_value=iter([]))
|
||||
self.assertRaises(AttributeError, set_int)
|
||||
|
||||
mock = MagicMock(spec=Iterable)
|
||||
self.assertEqual(list(mock), [])
|
||||
self.assertRaises(AttributeError, set_int)
|
||||
|
||||
|
||||
def test_magic_methods_and_spec_set(self):
|
||||
class Iterable(object):
|
||||
def __iter__(self):
|
||||
pass
|
||||
|
||||
mock = Mock(spec_set=Iterable)
|
||||
self.assertRaises(AttributeError, lambda: mock.__iter__)
|
||||
|
||||
mock.__iter__ = Mock(return_value=iter([]))
|
||||
self.assertEqual(list(mock), [])
|
||||
|
||||
class NonIterable(object):
|
||||
pass
|
||||
mock = Mock(spec_set=NonIterable)
|
||||
self.assertRaises(AttributeError, lambda: mock.__iter__)
|
||||
|
||||
def set_int():
|
||||
mock.__int__ = Mock(return_value=iter([]))
|
||||
self.assertRaises(AttributeError, set_int)
|
||||
|
||||
mock = MagicMock(spec_set=Iterable)
|
||||
self.assertEqual(list(mock), [])
|
||||
self.assertRaises(AttributeError, set_int)
|
||||
|
||||
|
||||
def test_setting_unsupported_magic_method(self):
|
||||
mock = MagicMock()
|
||||
def set_setattr():
|
||||
mock.__setattr__ = lambda self, name: None
|
||||
self.assertRaisesRegex(AttributeError,
|
||||
"Attempting to set unsupported magic method '__setattr__'.",
|
||||
set_setattr
|
||||
)
|
||||
|
||||
|
||||
def test_attributes_and_return_value(self):
|
||||
mock = MagicMock()
|
||||
attr = mock.foo
|
||||
def _get_type(obj):
|
||||
# the type of every mock (or magicmock) is a custom subclass
|
||||
# so the real type is the second in the mro
|
||||
return type(obj).__mro__[1]
|
||||
self.assertEqual(_get_type(attr), MagicMock)
|
||||
|
||||
returned = mock()
|
||||
self.assertEqual(_get_type(returned), MagicMock)
|
||||
|
||||
|
||||
def test_magic_methods_are_magic_mocks(self):
|
||||
mock = MagicMock()
|
||||
self.assertIsInstance(mock.__getitem__, MagicMock)
|
||||
|
||||
mock[1][2].__getitem__.return_value = 3
|
||||
self.assertEqual(mock[1][2][3], 3)
|
||||
|
||||
|
||||
def test_dir(self):
|
||||
# overriding the default implementation
|
||||
for mock in Mock(), MagicMock():
|
||||
def _dir(self):
|
||||
return ['foo']
|
||||
mock.__dir__ = _dir
|
||||
self.assertEqual(dir(mock), ['foo'])
|
||||
|
||||
|
||||
@unittest.skipIf('PyPy' in sys.version, "This fails differently on pypy")
|
||||
def test_bound_methods(self):
|
||||
m = Mock()
|
||||
|
||||
# XXXX should this be an expected failure instead?
|
||||
|
||||
# this seems like it should work, but is hard to do without introducing
|
||||
# other api inconsistencies. Failure message could be better though.
|
||||
m.__iter__ = [3].__iter__
|
||||
self.assertRaises(TypeError, iter, m)
|
||||
|
||||
|
||||
def test_magic_method_type(self):
|
||||
class Foo(MagicMock):
|
||||
pass
|
||||
|
||||
foo = Foo()
|
||||
self.assertIsInstance(foo.__int__, Foo)
|
||||
|
||||
|
||||
def test_descriptor_from_class(self):
|
||||
m = MagicMock()
|
||||
type(m).__str__.return_value = 'foo'
|
||||
self.assertEqual(str(m), 'foo')
|
||||
|
||||
|
||||
def test_iterable_as_iter_return_value(self):
|
||||
m = MagicMock()
|
||||
m.__iter__.return_value = [1, 2, 3]
|
||||
self.assertEqual(list(m), [1, 2, 3])
|
||||
self.assertEqual(list(m), [1, 2, 3])
|
||||
|
||||
m.__iter__.return_value = iter([4, 5, 6])
|
||||
self.assertEqual(list(m), [4, 5, 6])
|
||||
self.assertEqual(list(m), [])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
1258
Lib/unittest/test/testmock/testmock.py
Normal file
1258
Lib/unittest/test/testmock/testmock.py
Normal file
File diff suppressed because it is too large
Load Diff
1652
Lib/unittest/test/testmock/testpatch.py
Normal file
1652
Lib/unittest/test/testmock/testpatch.py
Normal file
File diff suppressed because it is too large
Load Diff
28
Lib/unittest/test/testmock/testsentinel.py
Normal file
28
Lib/unittest/test/testmock/testsentinel.py
Normal file
@ -0,0 +1,28 @@
|
||||
import unittest
|
||||
from unittest.mock import sentinel, DEFAULT
|
||||
|
||||
|
||||
class SentinelTest(unittest.TestCase):
|
||||
|
||||
def testSentinels(self):
|
||||
self.assertEqual(sentinel.whatever, sentinel.whatever,
|
||||
'sentinel not stored')
|
||||
self.assertNotEqual(sentinel.whatever, sentinel.whateverelse,
|
||||
'sentinel should be unique')
|
||||
|
||||
|
||||
def testSentinelName(self):
|
||||
self.assertEqual(str(sentinel.whatever), 'sentinel.whatever',
|
||||
'sentinel name incorrect')
|
||||
|
||||
|
||||
def testDEFAULT(self):
|
||||
self.assertTrue(DEFAULT is sentinel.DEFAULT)
|
||||
|
||||
def testBases(self):
|
||||
# If this doesn't raise an AttributeError then help(mock) is broken
|
||||
self.assertRaises(AttributeError, lambda: sentinel.__bases__)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
176
Lib/unittest/test/testmock/testwith.py
Normal file
176
Lib/unittest/test/testmock/testwith.py
Normal file
@ -0,0 +1,176 @@
|
||||
import unittest
|
||||
from warnings import catch_warnings
|
||||
|
||||
from unittest.test.testmock.support import is_instance
|
||||
from unittest.mock import MagicMock, Mock, patch, sentinel, mock_open, call
|
||||
|
||||
|
||||
|
||||
something = sentinel.Something
|
||||
something_else = sentinel.SomethingElse
|
||||
|
||||
|
||||
|
||||
class WithTest(unittest.TestCase):
|
||||
|
||||
def test_with_statement(self):
|
||||
with patch('%s.something' % __name__, sentinel.Something2):
|
||||
self.assertEqual(something, sentinel.Something2, "unpatched")
|
||||
self.assertEqual(something, sentinel.Something)
|
||||
|
||||
|
||||
def test_with_statement_exception(self):
|
||||
try:
|
||||
with patch('%s.something' % __name__, sentinel.Something2):
|
||||
self.assertEqual(something, sentinel.Something2, "unpatched")
|
||||
raise Exception('pow')
|
||||
except Exception:
|
||||
pass
|
||||
else:
|
||||
self.fail("patch swallowed exception")
|
||||
self.assertEqual(something, sentinel.Something)
|
||||
|
||||
|
||||
def test_with_statement_as(self):
|
||||
with patch('%s.something' % __name__) as mock_something:
|
||||
self.assertEqual(something, mock_something, "unpatched")
|
||||
self.assertTrue(is_instance(mock_something, MagicMock),
|
||||
"patching wrong type")
|
||||
self.assertEqual(something, sentinel.Something)
|
||||
|
||||
|
||||
def test_patch_object_with_statement(self):
|
||||
class Foo(object):
|
||||
something = 'foo'
|
||||
original = Foo.something
|
||||
with patch.object(Foo, 'something'):
|
||||
self.assertNotEqual(Foo.something, original, "unpatched")
|
||||
self.assertEqual(Foo.something, original)
|
||||
|
||||
|
||||
def test_with_statement_nested(self):
|
||||
with catch_warnings(record=True):
|
||||
with patch('%s.something' % __name__) as mock_something, patch('%s.something_else' % __name__) as mock_something_else:
|
||||
self.assertEqual(something, mock_something, "unpatched")
|
||||
self.assertEqual(something_else, mock_something_else,
|
||||
"unpatched")
|
||||
|
||||
self.assertEqual(something, sentinel.Something)
|
||||
self.assertEqual(something_else, sentinel.SomethingElse)
|
||||
|
||||
|
||||
def test_with_statement_specified(self):
|
||||
with patch('%s.something' % __name__, sentinel.Patched) as mock_something:
|
||||
self.assertEqual(something, mock_something, "unpatched")
|
||||
self.assertEqual(mock_something, sentinel.Patched, "wrong patch")
|
||||
self.assertEqual(something, sentinel.Something)
|
||||
|
||||
|
||||
def testContextManagerMocking(self):
|
||||
mock = Mock()
|
||||
mock.__enter__ = Mock()
|
||||
mock.__exit__ = Mock()
|
||||
mock.__exit__.return_value = False
|
||||
|
||||
with mock as m:
|
||||
self.assertEqual(m, mock.__enter__.return_value)
|
||||
mock.__enter__.assert_called_with()
|
||||
mock.__exit__.assert_called_with(None, None, None)
|
||||
|
||||
|
||||
def test_context_manager_with_magic_mock(self):
|
||||
mock = MagicMock()
|
||||
|
||||
with self.assertRaises(TypeError):
|
||||
with mock:
|
||||
'foo' + 3
|
||||
mock.__enter__.assert_called_with()
|
||||
self.assertTrue(mock.__exit__.called)
|
||||
|
||||
|
||||
def test_with_statement_same_attribute(self):
|
||||
with patch('%s.something' % __name__, sentinel.Patched) as mock_something:
|
||||
self.assertEqual(something, mock_something, "unpatched")
|
||||
|
||||
with patch('%s.something' % __name__) as mock_again:
|
||||
self.assertEqual(something, mock_again, "unpatched")
|
||||
|
||||
self.assertEqual(something, mock_something,
|
||||
"restored with wrong instance")
|
||||
|
||||
self.assertEqual(something, sentinel.Something, "not restored")
|
||||
|
||||
|
||||
def test_with_statement_imbricated(self):
|
||||
with patch('%s.something' % __name__) as mock_something:
|
||||
self.assertEqual(something, mock_something, "unpatched")
|
||||
|
||||
with patch('%s.something_else' % __name__) as mock_something_else:
|
||||
self.assertEqual(something_else, mock_something_else,
|
||||
"unpatched")
|
||||
|
||||
self.assertEqual(something, sentinel.Something)
|
||||
self.assertEqual(something_else, sentinel.SomethingElse)
|
||||
|
||||
|
||||
def test_dict_context_manager(self):
|
||||
foo = {}
|
||||
with patch.dict(foo, {'a': 'b'}):
|
||||
self.assertEqual(foo, {'a': 'b'})
|
||||
self.assertEqual(foo, {})
|
||||
|
||||
with self.assertRaises(NameError):
|
||||
with patch.dict(foo, {'a': 'b'}):
|
||||
self.assertEqual(foo, {'a': 'b'})
|
||||
raise NameError('Konrad')
|
||||
|
||||
self.assertEqual(foo, {})
|
||||
|
||||
|
||||
|
||||
class TestMockOpen(unittest.TestCase):
|
||||
|
||||
def test_mock_open(self):
|
||||
mock = mock_open()
|
||||
with patch('%s.open' % __name__, mock, create=True) as patched:
|
||||
self.assertIs(patched, mock)
|
||||
open('foo')
|
||||
|
||||
mock.assert_called_once_with('foo')
|
||||
|
||||
|
||||
def test_mock_open_context_manager(self):
|
||||
mock = mock_open()
|
||||
handle = mock.return_value
|
||||
with patch('%s.open' % __name__, mock, create=True):
|
||||
with open('foo') as f:
|
||||
f.read()
|
||||
|
||||
expected_calls = [call('foo'), call().__enter__(), call().read(),
|
||||
call().__exit__(None, None, None)]
|
||||
self.assertEqual(mock.mock_calls, expected_calls)
|
||||
self.assertIs(f, handle)
|
||||
|
||||
|
||||
def test_explicit_mock(self):
|
||||
mock = MagicMock()
|
||||
mock_open(mock)
|
||||
|
||||
with patch('%s.open' % __name__, mock, create=True) as patched:
|
||||
self.assertIs(patched, mock)
|
||||
open('foo')
|
||||
|
||||
mock.assert_called_once_with('foo')
|
||||
|
||||
|
||||
def test_read_data(self):
|
||||
mock = mock_open(read_data='foo')
|
||||
with patch('%s.open' % __name__, mock, create=True):
|
||||
h = open('bar')
|
||||
result = h.read()
|
||||
|
||||
self.assertEqual(result, 'foo')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Loading…
Reference in New Issue
Block a user