| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | # mock.py | 
					
						
							|  |  |  | # Test tools for mocking and patching. | 
					
						
							| 
									
										
										
										
											2012-03-14 13:30:29 -07:00
										 |  |  | # Maintained by Michael Foord | 
					
						
							|  |  |  | # Backport for other versions of Python available from | 
					
						
							| 
									
										
										
										
											2018-06-11 00:45:50 -04:00
										 |  |  | # https://pypi.org/project/mock | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | __all__ = ( | 
					
						
							|  |  |  |     'Mock', | 
					
						
							|  |  |  |     'MagicMock', | 
					
						
							|  |  |  |     'patch', | 
					
						
							|  |  |  |     'sentinel', | 
					
						
							|  |  |  |     'DEFAULT', | 
					
						
							|  |  |  |     'ANY', | 
					
						
							|  |  |  |     'call', | 
					
						
							|  |  |  |     'create_autospec', | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |     'AsyncMock', | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     'FILTER_DIR', | 
					
						
							|  |  |  |     'NonCallableMock', | 
					
						
							|  |  |  |     'NonCallableMagicMock', | 
					
						
							|  |  |  |     'mock_open', | 
					
						
							|  |  |  |     'PropertyMock', | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  |     'seal', | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | import asyncio | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  | import contextlib | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  | import io | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | import inspect | 
					
						
							|  |  |  | import pprint | 
					
						
							|  |  |  | import sys | 
					
						
							| 
									
										
										
										
											2014-04-14 16:25:20 -04:00
										 |  |  | import builtins | 
					
						
							| 
									
										
										
										
											2021-07-21 12:47:44 +01:00
										 |  |  | import pkgutil | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  | from asyncio import iscoroutinefunction | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | from types import CodeType, ModuleType, MethodType | 
					
						
							| 
									
										
										
										
											2018-10-28 21:37:10 +01:00
										 |  |  | from unittest.util import safe_repr | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  | from functools import wraps, partial | 
					
						
							| 
									
										
										
										
											2022-10-28 03:51:18 -04:00
										 |  |  | from threading import RLock | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-09 23:45:50 -04:00
										 |  |  | class InvalidSpecError(Exception): | 
					
						
							|  |  |  |     """Indicates that an invalid value was used as a mock spec.""" | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-04-14 16:25:20 -04:00
										 |  |  | _builtins = {name for name in dir(builtins) if not name.startswith('_')} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | FILTER_DIR = True | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-05-27 18:17:07 +10:00
										 |  |  | # Workaround for issue #12370 | 
					
						
							|  |  |  | # Without this, the __class__ properties wouldn't be set correctly | 
					
						
							|  |  |  | _safe_super = super | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | def _is_async_obj(obj): | 
					
						
							| 
									
										
										
										
											2019-09-10 12:18:40 +01:00
										 |  |  |     if _is_instance_mock(obj) and not isinstance(obj, AsyncMock): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         return False | 
					
						
							| 
									
										
										
										
											2020-01-25 04:17:47 -06:00
										 |  |  |     if hasattr(obj, '__func__'): | 
					
						
							|  |  |  |         obj = getattr(obj, '__func__') | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |     return iscoroutinefunction(obj) or inspect.isawaitable(obj) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-27 18:26:23 +05:30
										 |  |  | def _is_async_func(func): | 
					
						
							|  |  |  |     if getattr(func, '__code__', None): | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |         return iscoroutinefunction(func) | 
					
						
							| 
									
										
										
										
											2019-05-27 18:26:23 +05:30
										 |  |  |     else: | 
					
						
							|  |  |  |         return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | def _is_instance_mock(obj): | 
					
						
							|  |  |  |     # can't use isinstance on Mock objects because they override __class__ | 
					
						
							|  |  |  |     # The base class for all mocks is NonCallableMock | 
					
						
							|  |  |  |     return issubclass(type(obj), NonCallableMock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _is_exception(obj): | 
					
						
							|  |  |  |     return ( | 
					
						
							| 
									
										
										
										
											2019-05-01 08:48:44 +01:00
										 |  |  |         isinstance(obj, BaseException) or | 
					
						
							|  |  |  |         isinstance(obj, type) and issubclass(obj, BaseException) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-07-22 13:08:22 +05:30
										 |  |  | def _extract_mock(obj): | 
					
						
							|  |  |  |     # Autospecced functions will return a FunctionType with "mock" attribute | 
					
						
							|  |  |  |     # which is the actual mock object that needs to be used. | 
					
						
							|  |  |  |     if isinstance(obj, FunctionTypes) and hasattr(obj, 'mock'): | 
					
						
							|  |  |  |         return obj.mock | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         return obj | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  | def _get_signature_object(func, as_instance, eat_self): | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     Given an arbitrary, possibly callable object, try to create a suitable | 
					
						
							|  |  |  |     signature object. | 
					
						
							|  |  |  |     Return a (reduced func, signature) tuple, or None. | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     if isinstance(func, type) and not as_instance: | 
					
						
							|  |  |  |         # If it's a type and should be modelled as a type, use __init__. | 
					
						
							| 
									
										
										
										
											2019-05-01 23:04:04 +01:00
										 |  |  |         func = func.__init__ | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         # Skip the `self` argument in __init__ | 
					
						
							|  |  |  |         eat_self = True | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     elif not isinstance(func, FunctionTypes): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         # If we really want to model an instance of the passed type, | 
					
						
							|  |  |  |         # __call__ should be looked up, not __init__. | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         try: | 
					
						
							|  |  |  |             func = func.__call__ | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             return None | 
					
						
							|  |  |  |     if eat_self: | 
					
						
							|  |  |  |         sig_func = partial(func, None) | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         sig_func = func | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     try: | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         return func, inspect.signature(sig_func) | 
					
						
							|  |  |  |     except ValueError: | 
					
						
							|  |  |  |         # Certain callable types are not supported by inspect.signature() | 
					
						
							|  |  |  |         return None | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _check_signature(func, mock, skipfirst, instance=False): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     sig = _get_signature_object(func, instance, skipfirst) | 
					
						
							|  |  |  |     if sig is None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     func, sig = sig | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def checksig(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         sig.bind(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     _copy_func_details(func, checksig) | 
					
						
							|  |  |  |     type(mock)._mock_check_sig = checksig | 
					
						
							| 
									
										
										
										
											2018-12-12 13:24:54 +05:30
										 |  |  |     type(mock).__signature__ = sig | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _copy_func_details(func, funcopy): | 
					
						
							|  |  |  |     # we explicitly don't copy func.__dict__ into this copy as it would | 
					
						
							|  |  |  |     # expose original attributes that should be mocked | 
					
						
							| 
									
										
										
										
											2016-12-15 05:21:44 +03:00
										 |  |  |     for attribute in ( | 
					
						
							|  |  |  |         '__name__', '__doc__', '__text_signature__', | 
					
						
							|  |  |  |         '__module__', '__defaults__', '__kwdefaults__', | 
					
						
							|  |  |  |     ): | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             setattr(funcopy, attribute, getattr(func, attribute)) | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							|  |  |  |             pass | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _callable(obj): | 
					
						
							|  |  |  |     if isinstance(obj, type): | 
					
						
							|  |  |  |         return True | 
					
						
							| 
									
										
										
										
											2019-04-22 08:00:23 +05:30
										 |  |  |     if isinstance(obj, (staticmethod, classmethod, MethodType)): | 
					
						
							|  |  |  |         return _callable(obj.__func__) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     if getattr(obj, '__call__', None) is not None: | 
					
						
							|  |  |  |         return True | 
					
						
							|  |  |  |     return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _is_list(obj): | 
					
						
							|  |  |  |     # checks for list or tuples | 
					
						
							|  |  |  |     # XXXX badly named! | 
					
						
							|  |  |  |     return type(obj) in (list, tuple) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _instance_callable(obj): | 
					
						
							|  |  |  |     """Given an object, return True if the object is callable.
 | 
					
						
							|  |  |  |     For classes, return True if instances would be callable."""
 | 
					
						
							|  |  |  |     if not isinstance(obj, type): | 
					
						
							|  |  |  |         # already an instance | 
					
						
							|  |  |  |         return getattr(obj, '__call__', None) is not None | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 14:40:22 -07:00
										 |  |  |     # *could* be broken by a class overriding __mro__ or __dict__ via | 
					
						
							|  |  |  |     # a metaclass | 
					
						
							|  |  |  |     for base in (obj,) + obj.__mro__: | 
					
						
							|  |  |  |         if base.__dict__.get('__call__') is not None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             return True | 
					
						
							|  |  |  |     return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _set_signature(mock, original, instance=False): | 
					
						
							|  |  |  |     # creates a function with signature (*args, **kwargs) that delegates to a | 
					
						
							|  |  |  |     # mock. It still does signature checking by calling a lambda with the same | 
					
						
							|  |  |  |     # signature as the original. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     skipfirst = isinstance(original, type) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     result = _get_signature_object(original, instance, skipfirst) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     if result is None: | 
					
						
							| 
									
										
										
										
											2017-07-19 17:01:14 -07:00
										 |  |  |         return mock | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     func, sig = result | 
					
						
							|  |  |  |     def checksig(*args, **kwargs): | 
					
						
							|  |  |  |         sig.bind(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     _copy_func_details(func, checksig) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     name = original.__name__ | 
					
						
							|  |  |  |     if not name.isidentifier(): | 
					
						
							|  |  |  |         name = 'funcopy' | 
					
						
							| 
									
										
										
										
											2012-03-25 18:16:07 +01:00
										 |  |  |     context = {'_checksig_': checksig, 'mock': mock} | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     src = """def %s(*args, **kwargs):
 | 
					
						
							| 
									
										
										
										
											2012-03-25 18:16:07 +01:00
										 |  |  |     _checksig_(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     return mock(*args, **kwargs)""" % name
 | 
					
						
							|  |  |  |     exec (src, context) | 
					
						
							|  |  |  |     funcopy = context[name] | 
					
						
							| 
									
										
										
										
											2018-12-12 13:24:54 +05:30
										 |  |  |     _setup_func(funcopy, mock, sig) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     return funcopy | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-12 13:24:54 +05:30
										 |  |  | def _setup_func(funcopy, mock, sig): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     funcopy.mock = mock | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def assert_called_with(*args, **kwargs): | 
					
						
							|  |  |  |         return mock.assert_called_with(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2016-10-06 14:31:23 -07:00
										 |  |  |     def assert_called(*args, **kwargs): | 
					
						
							|  |  |  |         return mock.assert_called(*args, **kwargs) | 
					
						
							|  |  |  |     def assert_not_called(*args, **kwargs): | 
					
						
							|  |  |  |         return mock.assert_not_called(*args, **kwargs) | 
					
						
							|  |  |  |     def assert_called_once(*args, **kwargs): | 
					
						
							|  |  |  |         return mock.assert_called_once(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def assert_called_once_with(*args, **kwargs): | 
					
						
							|  |  |  |         return mock.assert_called_once_with(*args, **kwargs) | 
					
						
							|  |  |  |     def assert_has_calls(*args, **kwargs): | 
					
						
							|  |  |  |         return mock.assert_has_calls(*args, **kwargs) | 
					
						
							|  |  |  |     def assert_any_call(*args, **kwargs): | 
					
						
							|  |  |  |         return mock.assert_any_call(*args, **kwargs) | 
					
						
							|  |  |  |     def reset_mock(): | 
					
						
							|  |  |  |         funcopy.method_calls = _CallList() | 
					
						
							|  |  |  |         funcopy.mock_calls = _CallList() | 
					
						
							|  |  |  |         mock.reset_mock() | 
					
						
							|  |  |  |         ret = funcopy.return_value | 
					
						
							|  |  |  |         if _is_instance_mock(ret) and not ret is mock: | 
					
						
							|  |  |  |             ret.reset_mock() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     funcopy.called = False | 
					
						
							|  |  |  |     funcopy.call_count = 0 | 
					
						
							|  |  |  |     funcopy.call_args = None | 
					
						
							|  |  |  |     funcopy.call_args_list = _CallList() | 
					
						
							|  |  |  |     funcopy.method_calls = _CallList() | 
					
						
							|  |  |  |     funcopy.mock_calls = _CallList() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     funcopy.return_value = mock.return_value | 
					
						
							|  |  |  |     funcopy.side_effect = mock.side_effect | 
					
						
							|  |  |  |     funcopy._mock_children = mock._mock_children | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     funcopy.assert_called_with = assert_called_with | 
					
						
							|  |  |  |     funcopy.assert_called_once_with = assert_called_once_with | 
					
						
							|  |  |  |     funcopy.assert_has_calls = assert_has_calls | 
					
						
							|  |  |  |     funcopy.assert_any_call = assert_any_call | 
					
						
							|  |  |  |     funcopy.reset_mock = reset_mock | 
					
						
							| 
									
										
										
										
											2016-10-06 14:31:23 -07:00
										 |  |  |     funcopy.assert_called = assert_called | 
					
						
							|  |  |  |     funcopy.assert_not_called = assert_not_called | 
					
						
							|  |  |  |     funcopy.assert_called_once = assert_called_once | 
					
						
							| 
									
										
										
										
											2018-12-12 13:24:54 +05:30
										 |  |  |     funcopy.__signature__ = sig | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     mock._mock_delegate = funcopy | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-27 18:26:23 +05:30
										 |  |  | def _setup_async_mock(mock): | 
					
						
							|  |  |  |     mock._is_coroutine = asyncio.coroutines._is_coroutine | 
					
						
							|  |  |  |     mock.await_count = 0 | 
					
						
							|  |  |  |     mock.await_args = None | 
					
						
							|  |  |  |     mock.await_args_list = _CallList() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     # Mock is not configured yet so the attributes are set | 
					
						
							|  |  |  |     # to a function and then the corresponding mock helper function | 
					
						
							|  |  |  |     # is called when the helper is accessed similar to _setup_func. | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def wrapper(attr, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-05-27 18:26:23 +05:30
										 |  |  |         return getattr(mock.mock, attr)(*args, **kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for attribute in ('assert_awaited', | 
					
						
							|  |  |  |                       'assert_awaited_once', | 
					
						
							|  |  |  |                       'assert_awaited_with', | 
					
						
							|  |  |  |                       'assert_awaited_once_with', | 
					
						
							|  |  |  |                       'assert_any_await', | 
					
						
							|  |  |  |                       'assert_has_awaits', | 
					
						
							|  |  |  |                       'assert_not_awaited'): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # setattr(mock, attribute, wrapper) causes late binding | 
					
						
							|  |  |  |         # hence attribute will always be the last value in the loop | 
					
						
							|  |  |  |         # Use partial(wrapper, attribute) to ensure the attribute is bound | 
					
						
							|  |  |  |         # correctly. | 
					
						
							|  |  |  |         setattr(mock, attribute, partial(wrapper, attribute)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | def _is_magic(name): | 
					
						
							|  |  |  |     return '__%s__' % name[2:-2] == name | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class _SentinelObject(object): | 
					
						
							|  |  |  |     "A unique, named, sentinel object." | 
					
						
							|  |  |  |     def __init__(self, name): | 
					
						
							|  |  |  |         self.name = name | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         return 'sentinel.%s' % self.name | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-01-11 20:13:03 +02:00
										 |  |  |     def __reduce__(self): | 
					
						
							|  |  |  |         return 'sentinel.%s' % self.name | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | class _Sentinel(object): | 
					
						
							|  |  |  |     """Access attributes to return a named object, usable as a sentinel.""" | 
					
						
							|  |  |  |     def __init__(self): | 
					
						
							|  |  |  |         self._sentinels = {} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __getattr__(self, name): | 
					
						
							|  |  |  |         if name == '__bases__': | 
					
						
							|  |  |  |             # Without this help(unittest.mock) raises an exception | 
					
						
							|  |  |  |             raise AttributeError | 
					
						
							|  |  |  |         return self._sentinels.setdefault(name, _SentinelObject(name)) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-01-11 20:13:03 +02:00
										 |  |  |     def __reduce__(self): | 
					
						
							|  |  |  |         return 'sentinel' | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | sentinel = _Sentinel() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | DEFAULT = sentinel.DEFAULT | 
					
						
							|  |  |  | _missing = sentinel.MISSING | 
					
						
							|  |  |  | _deleted = sentinel.DELETED | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-12-11 10:28:14 +02:00
										 |  |  | _allowed_names = { | 
					
						
							|  |  |  |     'return_value', '_mock_return_value', 'side_effect', | 
					
						
							|  |  |  |     '_mock_side_effect', '_mock_parent', '_mock_new_parent', | 
					
						
							|  |  |  |     '_mock_name', '_mock_new_name' | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _delegating_property(name): | 
					
						
							|  |  |  |     _allowed_names.add(name) | 
					
						
							|  |  |  |     _the_name = '_mock_' + name | 
					
						
							|  |  |  |     def _get(self, name=name, _the_name=_the_name): | 
					
						
							|  |  |  |         sig = self._mock_delegate | 
					
						
							|  |  |  |         if sig is None: | 
					
						
							|  |  |  |             return getattr(self, _the_name) | 
					
						
							|  |  |  |         return getattr(sig, name) | 
					
						
							|  |  |  |     def _set(self, value, name=name, _the_name=_the_name): | 
					
						
							|  |  |  |         sig = self._mock_delegate | 
					
						
							|  |  |  |         if sig is None: | 
					
						
							|  |  |  |             self.__dict__[_the_name] = value | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             setattr(sig, name, value) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     return property(_get, _set) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class _CallList(list): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __contains__(self, value): | 
					
						
							|  |  |  |         if not isinstance(value, list): | 
					
						
							|  |  |  |             return list.__contains__(self, value) | 
					
						
							|  |  |  |         len_value = len(value) | 
					
						
							|  |  |  |         len_self = len(self) | 
					
						
							|  |  |  |         if len_value > len_self: | 
					
						
							|  |  |  |             return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for i in range(0, len_self - len_value + 1): | 
					
						
							|  |  |  |             sub_list = self[i:i+len_value] | 
					
						
							|  |  |  |             if sub_list == value: | 
					
						
							|  |  |  |                 return True | 
					
						
							|  |  |  |         return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         return pprint.pformat(list(self)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _check_and_set_parent(parent, value, name, new_name): | 
					
						
							| 
									
										
										
										
											2019-07-22 13:08:22 +05:30
										 |  |  |     value = _extract_mock(value) | 
					
						
							| 
									
										
										
										
											2019-02-26 03:16:34 +05:30
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     if not _is_instance_mock(value): | 
					
						
							|  |  |  |         return False | 
					
						
							|  |  |  |     if ((value._mock_name or value._mock_new_name) or | 
					
						
							|  |  |  |         (value._mock_parent is not None) or | 
					
						
							|  |  |  |         (value._mock_new_parent is not None)): | 
					
						
							|  |  |  |         return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     _parent = parent | 
					
						
							|  |  |  |     while _parent is not None: | 
					
						
							|  |  |  |         # setting a mock (value) as a child or return value of itself | 
					
						
							|  |  |  |         # should not modify the mock | 
					
						
							|  |  |  |         if _parent is value: | 
					
						
							|  |  |  |             return False | 
					
						
							|  |  |  |         _parent = _parent._mock_new_parent | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if new_name: | 
					
						
							|  |  |  |         value._mock_new_parent = parent | 
					
						
							|  |  |  |         value._mock_new_name = new_name | 
					
						
							|  |  |  |     if name: | 
					
						
							|  |  |  |         value._mock_parent = parent | 
					
						
							|  |  |  |         value._mock_name = name | 
					
						
							|  |  |  |     return True | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-04-14 16:09:42 -04:00
										 |  |  | # Internal class to identify if we wrapped an iterator object or not. | 
					
						
							|  |  |  | class _MockIter(object): | 
					
						
							|  |  |  |     def __init__(self, obj): | 
					
						
							|  |  |  |         self.obj = iter(obj) | 
					
						
							|  |  |  |     def __next__(self): | 
					
						
							|  |  |  |         return next(self.obj) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | class Base(object): | 
					
						
							|  |  |  |     _mock_return_value = DEFAULT | 
					
						
							|  |  |  |     _mock_side_effect = None | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def __init__(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class NonCallableMock(Base): | 
					
						
							|  |  |  |     """A non-callable version of `Mock`""" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-28 03:51:18 -04:00
										 |  |  |     # Store a mutex as a class attribute in order to protect concurrent access | 
					
						
							|  |  |  |     # to mock attributes. Using a class attribute allows all NonCallableMock | 
					
						
							|  |  |  |     # instances to share the mutex for simplicity. | 
					
						
							|  |  |  |     # | 
					
						
							|  |  |  |     # See https://github.com/python/cpython/issues/98624 for why this is | 
					
						
							|  |  |  |     # necessary. | 
					
						
							|  |  |  |     _lock = RLock() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def __new__(cls, /, *args, **kw): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         # every instance has its own class | 
					
						
							|  |  |  |         # so we can create magic methods on the | 
					
						
							|  |  |  |         # class without stomping on other mocks | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         bases = (cls,) | 
					
						
							| 
									
										
										
										
											2019-09-13 18:40:56 +02:00
										 |  |  |         if not issubclass(cls, AsyncMockMixin): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |             # Check if spec is an async object or function | 
					
						
							| 
									
										
										
										
											2019-09-13 18:40:56 +02:00
										 |  |  |             bound_args = _MOCK_SIG.bind_partial(cls, *args, **kw).arguments | 
					
						
							|  |  |  |             spec_arg = bound_args.get('spec_set', bound_args.get('spec')) | 
					
						
							| 
									
										
										
										
											2020-12-06 11:59:36 +02:00
										 |  |  |             if spec_arg is not None and _is_async_obj(spec_arg): | 
					
						
							| 
									
										
										
										
											2019-09-13 18:40:56 +02:00
										 |  |  |                 bases = (AsyncMockMixin, cls) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         new = type(cls.__name__, bases, {'__doc__': cls.__doc__}) | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |         instance = _safe_super(NonCallableMock, cls).__new__(new) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return instance | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__( | 
					
						
							|  |  |  |             self, spec=None, wraps=None, name=None, spec_set=None, | 
					
						
							|  |  |  |             parent=None, _spec_state=None, _new_name='', _new_parent=None, | 
					
						
							| 
									
										
										
										
											2014-04-16 23:32:21 +05:30
										 |  |  |             _spec_as_instance=False, _eat_self=None, unsafe=False, **kwargs | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         ): | 
					
						
							|  |  |  |         if _new_parent is None: | 
					
						
							|  |  |  |             _new_parent = parent | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         __dict__ = self.__dict__ | 
					
						
							|  |  |  |         __dict__['_mock_parent'] = parent | 
					
						
							|  |  |  |         __dict__['_mock_name'] = name | 
					
						
							|  |  |  |         __dict__['_mock_new_name'] = _new_name | 
					
						
							|  |  |  |         __dict__['_mock_new_parent'] = _new_parent | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  |         __dict__['_mock_sealed'] = False | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         if spec_set is not None: | 
					
						
							|  |  |  |             spec = spec_set | 
					
						
							|  |  |  |             spec_set = True | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         if _eat_self is None: | 
					
						
							|  |  |  |             _eat_self = parent is not None | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         self._mock_add_spec(spec, spec_set, _spec_as_instance, _eat_self) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         __dict__['_mock_children'] = {} | 
					
						
							|  |  |  |         __dict__['_mock_wraps'] = wraps | 
					
						
							|  |  |  |         __dict__['_mock_delegate'] = None | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         __dict__['_mock_called'] = False | 
					
						
							|  |  |  |         __dict__['_mock_call_args'] = None | 
					
						
							|  |  |  |         __dict__['_mock_call_count'] = 0 | 
					
						
							|  |  |  |         __dict__['_mock_call_args_list'] = _CallList() | 
					
						
							|  |  |  |         __dict__['_mock_mock_calls'] = _CallList() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         __dict__['method_calls'] = _CallList() | 
					
						
							| 
									
										
										
										
											2014-04-16 23:32:21 +05:30
										 |  |  |         __dict__['_mock_unsafe'] = unsafe | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         if kwargs: | 
					
						
							|  |  |  |             self.configure_mock(**kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-05-27 18:17:07 +10:00
										 |  |  |         _safe_super(NonCallableMock, self).__init__( | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             spec, wraps, name, spec_set, parent, | 
					
						
							|  |  |  |             _spec_state | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def attach_mock(self, mock, attribute): | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         Attach a mock as an attribute of this one, replacing its name and | 
					
						
							|  |  |  |         parent. Calls to the attached mock will be recorded in the | 
					
						
							|  |  |  |         `method_calls` and `mock_calls` attributes of this one."""
 | 
					
						
							| 
									
										
										
										
											2019-07-22 13:08:22 +05:30
										 |  |  |         inner_mock = _extract_mock(mock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         inner_mock._mock_parent = None | 
					
						
							|  |  |  |         inner_mock._mock_new_parent = None | 
					
						
							|  |  |  |         inner_mock._mock_name = '' | 
					
						
							|  |  |  |         inner_mock._mock_new_name = None | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         setattr(self, attribute, mock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def mock_add_spec(self, spec, spec_set=False): | 
					
						
							|  |  |  |         """Add a spec to a mock. `spec` can either be an object or a
 | 
					
						
							|  |  |  |         list of strings. Only attributes on the `spec` can be fetched as | 
					
						
							|  |  |  |         attributes from the mock. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If `spec_set` is True then only attributes on the spec can be set."""
 | 
					
						
							|  |  |  |         self._mock_add_spec(spec, spec_set) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     def _mock_add_spec(self, spec, spec_set, _spec_as_instance=False, | 
					
						
							|  |  |  |                        _eat_self=False): | 
					
						
							| 
									
										
										
										
											2022-02-03 03:41:19 -05:00
										 |  |  |         if _is_instance_mock(spec): | 
					
						
							|  |  |  |             raise InvalidSpecError(f'Cannot spec a Mock object. [object={spec!r}]') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         _spec_class = None | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         _spec_signature = None | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         _spec_asyncs = [] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for attr in dir(spec): | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |             if iscoroutinefunction(getattr(spec, attr, None)): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |                 _spec_asyncs.append(attr) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         if spec is not None and not _is_list(spec): | 
					
						
							|  |  |  |             if isinstance(spec, type): | 
					
						
							|  |  |  |                 _spec_class = spec | 
					
						
							|  |  |  |             else: | 
					
						
							| 
									
										
										
										
											2019-05-01 23:04:04 +01:00
										 |  |  |                 _spec_class = type(spec) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             res = _get_signature_object(spec, | 
					
						
							|  |  |  |                                         _spec_as_instance, _eat_self) | 
					
						
							|  |  |  |             _spec_signature = res and res[1] | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |             spec = dir(spec) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         __dict__ = self.__dict__ | 
					
						
							|  |  |  |         __dict__['_spec_class'] = _spec_class | 
					
						
							|  |  |  |         __dict__['_spec_set'] = spec_set | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         __dict__['_spec_signature'] = _spec_signature | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         __dict__['_mock_methods'] = spec | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         __dict__['_spec_asyncs'] = _spec_asyncs | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def __get_return_value(self): | 
					
						
							|  |  |  |         ret = self._mock_return_value | 
					
						
							|  |  |  |         if self._mock_delegate is not None: | 
					
						
							|  |  |  |             ret = self._mock_delegate.return_value | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if ret is DEFAULT: | 
					
						
							|  |  |  |             ret = self._get_child_mock( | 
					
						
							|  |  |  |                 _new_parent=self, _new_name='()' | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  |             self.return_value = ret | 
					
						
							|  |  |  |         return ret | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __set_return_value(self, value): | 
					
						
							|  |  |  |         if self._mock_delegate is not None: | 
					
						
							|  |  |  |             self._mock_delegate.return_value = value | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             self._mock_return_value = value | 
					
						
							|  |  |  |             _check_and_set_parent(self, value, None, '()') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     __return_value_doc = "The value to be returned when the mock is called." | 
					
						
							|  |  |  |     return_value = property(__get_return_value, __set_return_value, | 
					
						
							|  |  |  |                             __return_value_doc) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     @property | 
					
						
							|  |  |  |     def __class__(self): | 
					
						
							|  |  |  |         if self._spec_class is None: | 
					
						
							|  |  |  |             return type(self) | 
					
						
							|  |  |  |         return self._spec_class | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     called = _delegating_property('called') | 
					
						
							|  |  |  |     call_count = _delegating_property('call_count') | 
					
						
							|  |  |  |     call_args = _delegating_property('call_args') | 
					
						
							|  |  |  |     call_args_list = _delegating_property('call_args_list') | 
					
						
							|  |  |  |     mock_calls = _delegating_property('mock_calls') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __get_side_effect(self): | 
					
						
							|  |  |  |         delegated = self._mock_delegate | 
					
						
							|  |  |  |         if delegated is None: | 
					
						
							|  |  |  |             return self._mock_side_effect | 
					
						
							| 
									
										
										
										
											2014-04-14 16:09:42 -04:00
										 |  |  |         sf = delegated.side_effect | 
					
						
							| 
									
										
										
										
											2015-07-14 13:51:40 +12:00
										 |  |  |         if (sf is not None and not callable(sf) | 
					
						
							|  |  |  |                 and not isinstance(sf, _MockIter) and not _is_exception(sf)): | 
					
						
							| 
									
										
										
										
											2014-04-14 16:09:42 -04:00
										 |  |  |             sf = _MockIter(sf) | 
					
						
							|  |  |  |             delegated.side_effect = sf | 
					
						
							|  |  |  |         return sf | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def __set_side_effect(self, value): | 
					
						
							|  |  |  |         value = _try_iter(value) | 
					
						
							|  |  |  |         delegated = self._mock_delegate | 
					
						
							|  |  |  |         if delegated is None: | 
					
						
							|  |  |  |             self._mock_side_effect = value | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             delegated.side_effect = value | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     side_effect = property(__get_side_effect, __set_side_effect) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-06-02 10:20:16 -07:00
										 |  |  |     def reset_mock(self,  visited=None,*, return_value=False, side_effect=False): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         "Restore the mock object to its initial state." | 
					
						
							| 
									
										
										
										
											2015-07-15 11:42:28 +12:00
										 |  |  |         if visited is None: | 
					
						
							|  |  |  |             visited = [] | 
					
						
							|  |  |  |         if id(self) in visited: | 
					
						
							|  |  |  |             return | 
					
						
							|  |  |  |         visited.append(id(self)) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         self.called = False | 
					
						
							|  |  |  |         self.call_args = None | 
					
						
							|  |  |  |         self.call_count = 0 | 
					
						
							|  |  |  |         self.mock_calls = _CallList() | 
					
						
							|  |  |  |         self.call_args_list = _CallList() | 
					
						
							|  |  |  |         self.method_calls = _CallList() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-06-02 10:20:16 -07:00
										 |  |  |         if return_value: | 
					
						
							|  |  |  |             self._mock_return_value = DEFAULT | 
					
						
							|  |  |  |         if side_effect: | 
					
						
							|  |  |  |             self._mock_side_effect = None | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         for child in self._mock_children.values(): | 
					
						
							| 
									
										
										
										
											2018-12-01 15:33:54 +05:30
										 |  |  |             if isinstance(child, _SpecState) or child is _deleted: | 
					
						
							| 
									
										
										
										
											2012-06-09 17:31:59 +01:00
										 |  |  |                 continue | 
					
						
							| 
									
										
										
										
											2020-01-25 16:44:46 +01:00
										 |  |  |             child.reset_mock(visited, return_value=return_value, side_effect=side_effect) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         ret = self._mock_return_value | 
					
						
							|  |  |  |         if _is_instance_mock(ret) and ret is not self: | 
					
						
							| 
									
										
										
										
											2015-07-15 11:42:28 +12:00
										 |  |  |             ret.reset_mock(visited) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def configure_mock(self, /, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         """Set attributes on the mock through keyword arguments.
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         Attributes plus return values and side effects can be set on child | 
					
						
							|  |  |  |         mocks using standard dot notation and unpacking a dictionary in the | 
					
						
							|  |  |  |         method call: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         >>> attrs = {'method.return_value': 3, 'other.side_effect': KeyError} | 
					
						
							|  |  |  |         >>> mock.configure_mock(**attrs)"""
 | 
					
						
							|  |  |  |         for arg, val in sorted(kwargs.items(), | 
					
						
							|  |  |  |                                # we sort on the number of dots so that | 
					
						
							|  |  |  |                                # attributes are set before we set attributes on | 
					
						
							|  |  |  |                                # attributes | 
					
						
							|  |  |  |                                key=lambda entry: entry[0].count('.')): | 
					
						
							|  |  |  |             args = arg.split('.') | 
					
						
							|  |  |  |             final = args.pop() | 
					
						
							|  |  |  |             obj = self | 
					
						
							|  |  |  |             for entry in args: | 
					
						
							|  |  |  |                 obj = getattr(obj, entry) | 
					
						
							|  |  |  |             setattr(obj, final, val) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __getattr__(self, name): | 
					
						
							| 
									
										
										
										
											2014-04-16 23:32:21 +05:30
										 |  |  |         if name in {'_mock_methods', '_mock_unsafe'}: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             raise AttributeError(name) | 
					
						
							|  |  |  |         elif self._mock_methods is not None: | 
					
						
							|  |  |  |             if name not in self._mock_methods or name in _all_magics: | 
					
						
							|  |  |  |                 raise AttributeError("Mock object has no attribute %r" % name) | 
					
						
							|  |  |  |         elif _is_magic(name): | 
					
						
							|  |  |  |             raise AttributeError(name) | 
					
						
							| 
									
										
										
										
											2014-04-16 23:32:21 +05:30
										 |  |  |         if not self._mock_unsafe: | 
					
						
							| 
									
										
										
										
											2020-11-05 18:04:38 +01:00
										 |  |  |             if name.startswith(('assert', 'assret', 'asert', 'aseert', 'assrt')): | 
					
						
							| 
									
										
										
										
											2020-12-10 19:35:28 +01:00
										 |  |  |                 raise AttributeError( | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |                     f"{name!r} is not a valid assertion. Use a spec " | 
					
						
							|  |  |  |                     f"for the mock if {name!r} is meant to be an attribute.") | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-10-28 03:51:18 -04:00
										 |  |  |         with NonCallableMock._lock: | 
					
						
							|  |  |  |             result = self._mock_children.get(name) | 
					
						
							|  |  |  |             if result is _deleted: | 
					
						
							|  |  |  |                 raise AttributeError(name) | 
					
						
							|  |  |  |             elif result is None: | 
					
						
							|  |  |  |                 wraps = None | 
					
						
							|  |  |  |                 if self._mock_wraps is not None: | 
					
						
							|  |  |  |                     # XXXX should we get the attribute without triggering code | 
					
						
							|  |  |  |                     # execution? | 
					
						
							|  |  |  |                     wraps = getattr(self._mock_wraps, name) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 result = self._get_child_mock( | 
					
						
							|  |  |  |                     parent=self, name=name, wraps=wraps, _new_name=name, | 
					
						
							|  |  |  |                     _new_parent=self | 
					
						
							| 
									
										
										
										
											2021-04-09 23:45:50 -04:00
										 |  |  |                 ) | 
					
						
							| 
									
										
										
										
											2022-10-28 03:51:18 -04:00
										 |  |  |                 self._mock_children[name]  = result | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             elif isinstance(result, _SpecState): | 
					
						
							|  |  |  |                 try: | 
					
						
							|  |  |  |                     result = create_autospec( | 
					
						
							|  |  |  |                         result.spec, result.spec_set, result.instance, | 
					
						
							|  |  |  |                         result.parent, result.name | 
					
						
							|  |  |  |                     ) | 
					
						
							|  |  |  |                 except InvalidSpecError: | 
					
						
							|  |  |  |                     target_name = self.__dict__['_mock_name'] or self | 
					
						
							|  |  |  |                     raise InvalidSpecError( | 
					
						
							|  |  |  |                         f'Cannot autospec attr {name!r} from target ' | 
					
						
							|  |  |  |                         f'{target_name!r} as it has already been mocked out. ' | 
					
						
							|  |  |  |                         f'[target={self!r}, attr={result.spec!r}]') | 
					
						
							|  |  |  |                 self._mock_children[name]  = result | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         return result | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  |     def _extract_mock_name(self): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         _name_list = [self._mock_new_name] | 
					
						
							|  |  |  |         _parent = self._mock_new_parent | 
					
						
							|  |  |  |         last = self | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         dot = '.' | 
					
						
							|  |  |  |         if _name_list == ['()']: | 
					
						
							|  |  |  |             dot = '' | 
					
						
							| 
									
										
										
										
											2019-05-01 23:04:04 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         while _parent is not None: | 
					
						
							|  |  |  |             last = _parent | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             _name_list.append(_parent._mock_new_name + dot) | 
					
						
							|  |  |  |             dot = '.' | 
					
						
							|  |  |  |             if _parent._mock_new_name == '()': | 
					
						
							|  |  |  |                 dot = '' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             _parent = _parent._mock_new_parent | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         _name_list = list(reversed(_name_list)) | 
					
						
							|  |  |  |         _first = last._mock_name or 'mock' | 
					
						
							|  |  |  |         if len(_name_list) > 1: | 
					
						
							|  |  |  |             if _name_list[1] not in ('()', '().'): | 
					
						
							|  |  |  |                 _first += '.' | 
					
						
							|  |  |  |         _name_list[0] = _first | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  |         return ''.join(_name_list) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         name = self._extract_mock_name() | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         name_string = '' | 
					
						
							|  |  |  |         if name not in ('mock', 'mock.'): | 
					
						
							|  |  |  |             name_string = ' name=%r' % name | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         spec_string = '' | 
					
						
							|  |  |  |         if self._spec_class is not None: | 
					
						
							|  |  |  |             spec_string = ' spec=%r' | 
					
						
							|  |  |  |             if self._spec_set: | 
					
						
							|  |  |  |                 spec_string = ' spec_set=%r' | 
					
						
							|  |  |  |             spec_string = spec_string % self._spec_class.__name__ | 
					
						
							|  |  |  |         return "<%s%s%s id='%s'>" % ( | 
					
						
							|  |  |  |             type(self).__name__, | 
					
						
							|  |  |  |             name_string, | 
					
						
							|  |  |  |             spec_string, | 
					
						
							|  |  |  |             id(self) | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __dir__(self): | 
					
						
							| 
									
										
										
										
											2012-03-14 14:56:54 -07:00
										 |  |  |         """Filter the output of `dir(mock)` to only useful members.""" | 
					
						
							| 
									
										
										
										
											2012-03-25 18:16:07 +01:00
										 |  |  |         if not FILTER_DIR: | 
					
						
							|  |  |  |             return object.__dir__(self) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         extras = self._mock_methods or [] | 
					
						
							|  |  |  |         from_type = dir(type(self)) | 
					
						
							|  |  |  |         from_dict = list(self.__dict__) | 
					
						
							| 
									
										
										
										
											2019-04-30 19:56:36 +01:00
										 |  |  |         from_child_mocks = [ | 
					
						
							|  |  |  |             m_name for m_name, m_value in self._mock_children.items() | 
					
						
							|  |  |  |             if m_value is not _deleted] | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 18:16:07 +01:00
										 |  |  |         from_type = [e for e in from_type if not e.startswith('_')] | 
					
						
							|  |  |  |         from_dict = [e for e in from_dict if not e.startswith('_') or | 
					
						
							|  |  |  |                      _is_magic(e)] | 
					
						
							| 
									
										
										
										
											2019-04-30 19:56:36 +01:00
										 |  |  |         return sorted(set(extras + from_type + from_dict + from_child_mocks)) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __setattr__(self, name, value): | 
					
						
							|  |  |  |         if name in _allowed_names: | 
					
						
							|  |  |  |             # property setters go through here | 
					
						
							|  |  |  |             return object.__setattr__(self, name, value) | 
					
						
							|  |  |  |         elif (self._spec_set and self._mock_methods is not None and | 
					
						
							|  |  |  |             name not in self._mock_methods and | 
					
						
							|  |  |  |             name not in self.__dict__): | 
					
						
							|  |  |  |             raise AttributeError("Mock object has no attribute '%s'" % name) | 
					
						
							|  |  |  |         elif name in _unsupported_magics: | 
					
						
							|  |  |  |             msg = 'Attempting to set unsupported magic method %r.' % name | 
					
						
							|  |  |  |             raise AttributeError(msg) | 
					
						
							|  |  |  |         elif name in _all_magics: | 
					
						
							|  |  |  |             if self._mock_methods is not None and name not in self._mock_methods: | 
					
						
							|  |  |  |                 raise AttributeError("Mock object has no attribute '%s'" % name) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             if not _is_instance_mock(value): | 
					
						
							|  |  |  |                 setattr(type(self), name, _get_method(name, value)) | 
					
						
							|  |  |  |                 original = value | 
					
						
							|  |  |  |                 value = lambda *args, **kw: original(self, *args, **kw) | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 # only set _new_name and not name so that mock_calls is tracked | 
					
						
							|  |  |  |                 # but not method calls | 
					
						
							|  |  |  |                 _check_and_set_parent(self, value, None, name) | 
					
						
							|  |  |  |                 setattr(type(self), name, value) | 
					
						
							| 
									
										
										
										
											2012-06-09 17:31:59 +01:00
										 |  |  |                 self._mock_children[name] = value | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         elif name == '__class__': | 
					
						
							|  |  |  |             self._spec_class = value | 
					
						
							|  |  |  |             return | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             if _check_and_set_parent(self, value, name, name): | 
					
						
							|  |  |  |                 self._mock_children[name] = value | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  |         if self._mock_sealed and not hasattr(self, name): | 
					
						
							|  |  |  |             mock_name = f'{self._extract_mock_name()}.{name}' | 
					
						
							|  |  |  |             raise AttributeError(f'Cannot set {mock_name}') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return object.__setattr__(self, name, value) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __delattr__(self, name): | 
					
						
							|  |  |  |         if name in _all_magics and name in type(self).__dict__: | 
					
						
							|  |  |  |             delattr(type(self), name) | 
					
						
							|  |  |  |             if name not in self.__dict__: | 
					
						
							|  |  |  |                 # for magic methods that are still MagicProxy objects and | 
					
						
							|  |  |  |                 # not set on the instance itself | 
					
						
							|  |  |  |                 return | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         obj = self._mock_children.get(name, _missing) | 
					
						
							| 
									
										
										
										
											2019-01-21 08:57:46 +00:00
										 |  |  |         if name in self.__dict__: | 
					
						
							| 
									
										
										
										
											2019-04-14 00:42:33 +05:30
										 |  |  |             _safe_super(NonCallableMock, self).__delattr__(name) | 
					
						
							| 
									
										
										
										
											2019-01-21 08:57:46 +00:00
										 |  |  |         elif obj is _deleted: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             raise AttributeError(name) | 
					
						
							|  |  |  |         if obj is not _missing: | 
					
						
							|  |  |  |             del self._mock_children[name] | 
					
						
							|  |  |  |         self._mock_children[name] = _deleted | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _format_mock_call_signature(self, args, kwargs): | 
					
						
							|  |  |  |         name = self._mock_name or 'mock' | 
					
						
							|  |  |  |         return _format_call_signature(name, args, kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-29 12:32:26 +05:30
										 |  |  |     def _format_mock_failure_message(self, args, kwargs, action='call'): | 
					
						
							|  |  |  |         message = 'expected %s not found.\nExpected: %s\nActual: %s' | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         expected_string = self._format_mock_call_signature(args, kwargs) | 
					
						
							|  |  |  |         call_args = self.call_args | 
					
						
							|  |  |  |         actual_string = self._format_mock_call_signature(*call_args) | 
					
						
							| 
									
										
										
										
											2019-05-29 12:32:26 +05:30
										 |  |  |         return message % (action, expected_string, actual_string) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-08-29 11:39:01 +05:30
										 |  |  |     def _get_call_signature_from_name(self, name): | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         * If call objects are asserted against a method/function like obj.meth1 | 
					
						
							|  |  |  |         then there could be no name for the call object to lookup. Hence just | 
					
						
							|  |  |  |         return the spec_signature of the method/function being asserted against. | 
					
						
							|  |  |  |         * If the name is not empty then remove () and split by '.' to get | 
					
						
							|  |  |  |         list of names to iterate through the children until a potential | 
					
						
							|  |  |  |         match is found. A child mock is created only during attribute access | 
					
						
							|  |  |  |         so if we get a _SpecState then no attributes of the spec were accessed | 
					
						
							|  |  |  |         and can be safely exited. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if not name: | 
					
						
							|  |  |  |             return self._spec_signature | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         sig = None | 
					
						
							|  |  |  |         names = name.replace('()', '').split('.') | 
					
						
							|  |  |  |         children = self._mock_children | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         for name in names: | 
					
						
							|  |  |  |             child = children.get(name) | 
					
						
							|  |  |  |             if child is None or isinstance(child, _SpecState): | 
					
						
							|  |  |  |                 break | 
					
						
							|  |  |  |             else: | 
					
						
							| 
									
										
										
										
											2020-01-24 18:44:29 +05:30
										 |  |  |                 # If an autospecced object is attached using attach_mock the | 
					
						
							|  |  |  |                 # child would be a function with mock object as attribute from | 
					
						
							|  |  |  |                 # which signature has to be derived. | 
					
						
							|  |  |  |                 child = _extract_mock(child) | 
					
						
							| 
									
										
										
										
											2019-08-29 11:39:01 +05:30
										 |  |  |                 children = child._mock_children | 
					
						
							|  |  |  |                 sig = child._spec_signature | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         return sig | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     def _call_matcher(self, _call): | 
					
						
							|  |  |  |         """
 | 
					
						
							| 
									
										
										
										
											2016-07-11 07:51:37 +00:00
										 |  |  |         Given a call (or simply an (args, kwargs) tuple), return a | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         comparison key suitable for matching with other calls. | 
					
						
							|  |  |  |         This is a best effort method which relies on the spec's signature, | 
					
						
							|  |  |  |         if available, or falls back on the arguments themselves. | 
					
						
							|  |  |  |         """
 | 
					
						
							| 
									
										
										
										
											2019-08-29 11:39:01 +05:30
										 |  |  | 
 | 
					
						
							|  |  |  |         if isinstance(_call, tuple) and len(_call) > 2: | 
					
						
							|  |  |  |             sig = self._get_call_signature_from_name(_call[0]) | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             sig = self._spec_signature | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         if sig is not None: | 
					
						
							|  |  |  |             if len(_call) == 2: | 
					
						
							|  |  |  |                 name = '' | 
					
						
							|  |  |  |                 args, kwargs = _call | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 name, args, kwargs = _call | 
					
						
							|  |  |  |             try: | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |                 bound_call = sig.bind(*args, **kwargs) | 
					
						
							|  |  |  |                 return call(name, bound_call.args, bound_call.kwargs) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             except TypeError as e: | 
					
						
							|  |  |  |                 return e.with_traceback(None) | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             return _call | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_not_called(self): | 
					
						
							| 
									
										
										
										
											2014-04-17 01:36:14 +05:30
										 |  |  |         """assert that the mock was never called.
 | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if self.call_count != 0: | 
					
						
							| 
									
										
										
										
											2018-10-28 21:37:10 +01:00
										 |  |  |             msg = ("Expected '%s' to not have been called. Called %s times.%s" | 
					
						
							|  |  |  |                    % (self._mock_name or 'mock', | 
					
						
							|  |  |  |                       self.call_count, | 
					
						
							|  |  |  |                       self._calls_repr())) | 
					
						
							| 
									
										
										
										
											2016-03-11 22:17:48 +01:00
										 |  |  |             raise AssertionError(msg) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_called(self): | 
					
						
							| 
									
										
										
										
											2016-03-11 22:17:48 +01:00
										 |  |  |         """assert that the mock was called at least once
 | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if self.call_count == 0: | 
					
						
							|  |  |  |             msg = ("Expected '%s' to have been called." % | 
					
						
							| 
									
										
										
										
											2019-09-17 06:16:08 -05:00
										 |  |  |                    (self._mock_name or 'mock')) | 
					
						
							| 
									
										
										
										
											2016-03-11 22:17:48 +01:00
										 |  |  |             raise AssertionError(msg) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_called_once(self): | 
					
						
							| 
									
										
										
										
											2016-03-11 22:17:48 +01:00
										 |  |  |         """assert that the mock was called only once.
 | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if not self.call_count == 1: | 
					
						
							| 
									
										
										
										
											2018-10-28 21:37:10 +01:00
										 |  |  |             msg = ("Expected '%s' to have been called once. Called %s times.%s" | 
					
						
							|  |  |  |                    % (self._mock_name or 'mock', | 
					
						
							|  |  |  |                       self.call_count, | 
					
						
							|  |  |  |                       self._calls_repr())) | 
					
						
							| 
									
										
										
										
											2014-04-17 01:36:14 +05:30
										 |  |  |             raise AssertionError(msg) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_called_with(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-08-29 08:15:53 +02:00
										 |  |  |         """assert that the last call was made with the specified arguments.
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         Raises an AssertionError if the args and keyword args passed in are | 
					
						
							|  |  |  |         different to the last call to the mock."""
 | 
					
						
							|  |  |  |         if self.call_args is None: | 
					
						
							|  |  |  |             expected = self._format_mock_call_signature(args, kwargs) | 
					
						
							| 
									
										
										
										
											2019-02-13 18:22:29 -08:00
										 |  |  |             actual = 'not called.' | 
					
						
							|  |  |  |             error_message = ('expected call not found.\nExpected: %s\nActual: %s' | 
					
						
							|  |  |  |                     % (expected, actual)) | 
					
						
							|  |  |  |             raise AssertionError(error_message) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         def _error_message(): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             msg = self._format_mock_failure_message(args, kwargs) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             return msg | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         expected = self._call_matcher(_Call((args, kwargs), two=True)) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         actual = self._call_matcher(self.call_args) | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         if actual != expected: | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             cause = expected if isinstance(expected, Exception) else None | 
					
						
							|  |  |  |             raise AssertionError(_error_message()) from cause | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_called_once_with(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2017-02-23 15:57:25 +01:00
										 |  |  |         """assert that the mock was called exactly once and that that call was
 | 
					
						
							|  |  |  |         with the specified arguments."""
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         if not self.call_count == 1: | 
					
						
							| 
									
										
										
										
											2018-10-28 21:37:10 +01:00
										 |  |  |             msg = ("Expected '%s' to be called once. Called %s times.%s" | 
					
						
							|  |  |  |                    % (self._mock_name or 'mock', | 
					
						
							|  |  |  |                       self.call_count, | 
					
						
							|  |  |  |                       self._calls_repr())) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             raise AssertionError(msg) | 
					
						
							|  |  |  |         return self.assert_called_with(*args, **kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def assert_has_calls(self, calls, any_order=False): | 
					
						
							|  |  |  |         """assert the mock has been called with the specified calls.
 | 
					
						
							|  |  |  |         The `mock_calls` list is checked for the calls. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If `any_order` is False (the default) then the calls must be | 
					
						
							|  |  |  |         sequential. There can be extra calls before or after the | 
					
						
							|  |  |  |         specified calls. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If `any_order` is True then the calls can be in any order, but | 
					
						
							|  |  |  |         they must all appear in `mock_calls`."""
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         expected = [self._call_matcher(c) for c in calls] | 
					
						
							| 
									
										
										
										
											2019-09-24 15:08:31 -04:00
										 |  |  |         cause = next((e for e in expected if isinstance(e, Exception)), None) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         all_calls = _CallList(self._call_matcher(c) for c in self.mock_calls) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         if not any_order: | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             if expected not in all_calls: | 
					
						
							| 
									
										
										
										
											2019-09-24 15:08:31 -04:00
										 |  |  |                 if cause is None: | 
					
						
							|  |  |  |                     problem = 'Calls not found.' | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     problem = ('Error processing expected calls.\n' | 
					
						
							|  |  |  |                                'Errors: {}').format( | 
					
						
							|  |  |  |                                    [e if isinstance(e, Exception) else None | 
					
						
							|  |  |  |                                     for e in expected]) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 raise AssertionError( | 
					
						
							| 
									
										
										
										
											2019-09-24 15:08:31 -04:00
										 |  |  |                     f'{problem}\n' | 
					
						
							| 
									
										
										
										
											2019-09-24 18:04:29 -04:00
										 |  |  |                     f'Expected: {_CallList(calls)}' | 
					
						
							|  |  |  |                     f'{self._calls_repr(prefix="Actual").rstrip(".")}' | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |                 ) from cause | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             return | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         all_calls = list(all_calls) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         not_found = [] | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         for kall in expected: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             try: | 
					
						
							|  |  |  |                 all_calls.remove(kall) | 
					
						
							|  |  |  |             except ValueError: | 
					
						
							|  |  |  |                 not_found.append(kall) | 
					
						
							|  |  |  |         if not_found: | 
					
						
							|  |  |  |             raise AssertionError( | 
					
						
							| 
									
										
										
										
											2018-08-17 15:09:58 -04:00
										 |  |  |                 '%r does not contain all of %r in its call list, ' | 
					
						
							|  |  |  |                 'found %r instead' % (self._mock_name or 'mock', | 
					
						
							|  |  |  |                                       tuple(not_found), all_calls) | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             ) from cause | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_any_call(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         """assert the mock has been called with the specified arguments.
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         The assert passes if the mock has *ever* been called, unlike | 
					
						
							|  |  |  |         `assert_called_with` and `assert_called_once_with` that only pass if | 
					
						
							|  |  |  |         the call is the most recent one."""
 | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         expected = self._call_matcher(_Call((args, kwargs), two=True)) | 
					
						
							|  |  |  |         cause = expected if isinstance(expected, Exception) else None | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         actual = [self._call_matcher(c) for c in self.call_args_list] | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         if cause or expected not in _AnyComparer(actual): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             expected_string = self._format_mock_call_signature(args, kwargs) | 
					
						
							|  |  |  |             raise AssertionError( | 
					
						
							|  |  |  |                 '%s call not found' % expected_string | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             ) from cause | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def _get_child_mock(self, /, **kw): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         """Create the child mocks for attributes and return value.
 | 
					
						
							|  |  |  |         By default child mocks will be the same type as the parent. | 
					
						
							|  |  |  |         Subclasses of Mock may want to override this to customize the way | 
					
						
							|  |  |  |         child mocks are made. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         For non-callable mocks the callable variant will be used (rather than | 
					
						
							|  |  |  |         any custom subclass)."""
 | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         _new_name = kw.get("_new_name") | 
					
						
							|  |  |  |         if _new_name in self.__dict__['_spec_asyncs']: | 
					
						
							|  |  |  |             return AsyncMock(**kw) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-09-14 13:20:40 +03:00
										 |  |  |         if self._mock_sealed: | 
					
						
							|  |  |  |             attribute = f".{kw['name']}" if "name" in kw else "()" | 
					
						
							|  |  |  |             mock_name = self._extract_mock_name() + attribute | 
					
						
							|  |  |  |             raise AttributeError(mock_name) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         _type = type(self) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         if issubclass(_type, MagicMock) and _new_name in _async_method_magics: | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |             # Any asynchronous magic becomes an AsyncMock | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |             klass = AsyncMock | 
					
						
							| 
									
										
										
										
											2019-09-19 21:04:18 -07:00
										 |  |  |         elif issubclass(_type, AsyncMockMixin): | 
					
						
							| 
									
										
										
										
											2019-09-29 21:56:47 -07:00
										 |  |  |             if (_new_name in _all_sync_magics or | 
					
						
							|  |  |  |                     self._mock_methods and _new_name in self._mock_methods): | 
					
						
							|  |  |  |                 # Any synchronous method on AsyncMock becomes a MagicMock | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |                 klass = MagicMock | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 klass = AsyncMock | 
					
						
							| 
									
										
										
										
											2019-09-19 21:04:18 -07:00
										 |  |  |         elif not issubclass(_type, CallableMixin): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             if issubclass(_type, NonCallableMagicMock): | 
					
						
							|  |  |  |                 klass = MagicMock | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |             elif issubclass(_type, NonCallableMock): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 klass = Mock | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             klass = _type.__mro__[1] | 
					
						
							|  |  |  |         return klass(**kw) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-10-28 21:37:10 +01:00
										 |  |  |     def _calls_repr(self, prefix="Calls"): | 
					
						
							|  |  |  |         """Renders self.mock_calls as a string.
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         Example: "\nCalls: [call(1), call(2)]." | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If self.mock_calls is empty, an empty string is returned. The | 
					
						
							|  |  |  |         output will be truncated if very long. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if not self.mock_calls: | 
					
						
							|  |  |  |             return "" | 
					
						
							|  |  |  |         return f"\n{prefix}: {safe_repr(self.mock_calls)}." | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-13 18:40:56 +02:00
										 |  |  | _MOCK_SIG = inspect.signature(NonCallableMock.__init__) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  | class _AnyComparer(list): | 
					
						
							|  |  |  |     """A list which checks if it contains a call which may have an
 | 
					
						
							|  |  |  |     argument of ANY, flipping the components of item and self from | 
					
						
							|  |  |  |     their traditional locations so that ANY is guaranteed to be on | 
					
						
							|  |  |  |     the left."""
 | 
					
						
							|  |  |  |     def __contains__(self, item): | 
					
						
							|  |  |  |         for _call in self: | 
					
						
							| 
									
										
										
										
											2020-01-29 16:24:54 +00:00
										 |  |  |             assert len(item) == len(_call) | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |             if all([ | 
					
						
							|  |  |  |                 expected == actual | 
					
						
							|  |  |  |                 for expected, actual in zip(item, _call) | 
					
						
							|  |  |  |             ]): | 
					
						
							|  |  |  |                 return True | 
					
						
							|  |  |  |         return False | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | def _try_iter(obj): | 
					
						
							|  |  |  |     if obj is None: | 
					
						
							|  |  |  |         return obj | 
					
						
							|  |  |  |     if _is_exception(obj): | 
					
						
							|  |  |  |         return obj | 
					
						
							|  |  |  |     if _callable(obj): | 
					
						
							|  |  |  |         return obj | 
					
						
							|  |  |  |     try: | 
					
						
							|  |  |  |         return iter(obj) | 
					
						
							|  |  |  |     except TypeError: | 
					
						
							|  |  |  |         # XXXX backwards compatibility | 
					
						
							|  |  |  |         # but this will blow up on first call - so maybe we should fail early? | 
					
						
							|  |  |  |         return obj | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class CallableMixin(Base): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, spec=None, side_effect=None, return_value=DEFAULT, | 
					
						
							|  |  |  |                  wraps=None, name=None, spec_set=None, parent=None, | 
					
						
							|  |  |  |                  _spec_state=None, _new_name='', _new_parent=None, **kwargs): | 
					
						
							|  |  |  |         self.__dict__['_mock_return_value'] = return_value | 
					
						
							| 
									
										
										
										
											2012-05-27 18:17:07 +10:00
										 |  |  |         _safe_super(CallableMixin, self).__init__( | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             spec, wraps, name, spec_set, parent, | 
					
						
							|  |  |  |             _spec_state, _new_name, _new_parent, **kwargs | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.side_effect = side_effect | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def _mock_check_sig(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         # stub method that can be replaced with one with a specific signature | 
					
						
							|  |  |  |         pass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def __call__(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         # can't use self in-case a function / method we are mocking uses self | 
					
						
							|  |  |  |         # in the signature | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |         self._mock_check_sig(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2019-09-23 20:49:40 -07:00
										 |  |  |         self._increment_mock_call(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |         return self._mock_call(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def _mock_call(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-09-23 20:49:40 -07:00
										 |  |  |         return self._execute_mock_call(*args, **kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _increment_mock_call(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         self.called = True | 
					
						
							|  |  |  |         self.call_count += 1 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |         # handle call_args | 
					
						
							| 
									
										
										
										
											2019-09-23 20:49:40 -07:00
										 |  |  |         # needs to be set here so assertions on call arguments pass before | 
					
						
							|  |  |  |         # execution in the case of awaited calls | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |         _call = _Call((args, kwargs), two=True) | 
					
						
							|  |  |  |         self.call_args = _call | 
					
						
							|  |  |  |         self.call_args_list.append(_call) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |         # initial stuff for method_calls: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         do_method_calls = self._mock_parent is not None | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |         method_call_name = self._mock_name | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |         # initial stuff for mock_calls: | 
					
						
							|  |  |  |         mock_call_name = self._mock_new_name | 
					
						
							|  |  |  |         is_a_call = mock_call_name == '()' | 
					
						
							|  |  |  |         self.mock_calls.append(_Call(('', args, kwargs))) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |         # follow up the chain of mocks: | 
					
						
							|  |  |  |         _new_parent = self._mock_new_parent | 
					
						
							|  |  |  |         while _new_parent is not None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |             # handle method_calls: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             if do_method_calls: | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |                 _new_parent.method_calls.append(_Call((method_call_name, args, kwargs))) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 do_method_calls = _new_parent._mock_parent is not None | 
					
						
							|  |  |  |                 if do_method_calls: | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |                     method_call_name = _new_parent._mock_name + '.' + method_call_name | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |             # handle mock_calls: | 
					
						
							|  |  |  |             this_mock_call = _Call((mock_call_name, args, kwargs)) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             _new_parent.mock_calls.append(this_mock_call) | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |             if _new_parent._mock_new_name: | 
					
						
							|  |  |  |                 if is_a_call: | 
					
						
							|  |  |  |                     dot = '' | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     dot = '.' | 
					
						
							|  |  |  |                 is_a_call = _new_parent._mock_new_name == '()' | 
					
						
							|  |  |  |                 mock_call_name = _new_parent._mock_new_name + dot + mock_call_name | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             # follow the parental chain: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             _new_parent = _new_parent._mock_new_parent | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-23 20:49:40 -07:00
										 |  |  |     def _execute_mock_call(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |         # separate from _increment_mock_call so that awaited functions are | 
					
						
							|  |  |  |         # executed separately from their call, also AsyncMock overrides this method | 
					
						
							| 
									
										
										
										
											2019-09-23 20:49:40 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         effect = self.side_effect | 
					
						
							|  |  |  |         if effect is not None: | 
					
						
							|  |  |  |             if _is_exception(effect): | 
					
						
							|  |  |  |                 raise effect | 
					
						
							| 
									
										
										
										
											2018-12-08 11:25:02 +00:00
										 |  |  |             elif not _callable(effect): | 
					
						
							| 
									
										
										
										
											2012-04-21 15:52:11 +01:00
										 |  |  |                 result = next(effect) | 
					
						
							|  |  |  |                 if _is_exception(result): | 
					
						
							|  |  |  |                     raise result | 
					
						
							| 
									
										
										
										
											2018-12-08 11:25:02 +00:00
										 |  |  |             else: | 
					
						
							|  |  |  |                 result = effect(*args, **kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             if result is not DEFAULT: | 
					
						
							| 
									
										
										
										
											2012-04-21 15:52:11 +01:00
										 |  |  |                 return result | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-08 11:25:02 +00:00
										 |  |  |         if self._mock_return_value is not DEFAULT: | 
					
						
							|  |  |  |             return self.return_value | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-08 11:25:02 +00:00
										 |  |  |         if self._mock_wraps is not None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             return self._mock_wraps(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2018-12-08 11:25:02 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |         return self.return_value | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class Mock(CallableMixin, NonCallableMock): | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     Create a new `Mock` object. `Mock` takes several optional arguments | 
					
						
							|  |  |  |     that specify the behaviour of the Mock object: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     * `spec`: This can be either a list of strings or an existing object (a | 
					
						
							|  |  |  |       class or instance) that acts as the specification for the mock object. If | 
					
						
							|  |  |  |       you pass in an object then a list of strings is formed by calling dir on | 
					
						
							|  |  |  |       the object (excluding unsupported magic attributes and methods). Accessing | 
					
						
							|  |  |  |       any attribute not in this list will raise an `AttributeError`. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |       If `spec` is an object (rather than a list of strings) then | 
					
						
							|  |  |  |       `mock.__class__` returns the class of the spec object. This allows mocks | 
					
						
							|  |  |  |       to pass `isinstance` tests. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     * `spec_set`: A stricter variant of `spec`. If used, attempting to *set* | 
					
						
							|  |  |  |       or get an attribute on the mock that isn't on the object passed as | 
					
						
							|  |  |  |       `spec_set` will raise an `AttributeError`. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     * `side_effect`: A function to be called whenever the Mock is called. See | 
					
						
							|  |  |  |       the `side_effect` attribute. Useful for raising exceptions or | 
					
						
							|  |  |  |       dynamically changing return values. The function is called with the same | 
					
						
							|  |  |  |       arguments as the mock, and unless it returns `DEFAULT`, the return | 
					
						
							|  |  |  |       value of this function is used as the return value. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-04-21 15:52:11 +01:00
										 |  |  |       If `side_effect` is an iterable then each call to the mock will return | 
					
						
							|  |  |  |       the next value from the iterable. If any of the members of the iterable | 
					
						
							|  |  |  |       are exceptions they will be raised instead of returned. | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     * `return_value`: The value returned when the mock is called. By default | 
					
						
							|  |  |  |       this is a new Mock (created on first access). See the | 
					
						
							|  |  |  |       `return_value` attribute. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-07-05 02:52:32 -04:00
										 |  |  |     * `unsafe`: By default, accessing any attribute whose name starts with | 
					
						
							|  |  |  |       *assert*, *assret*, *asert*, *aseert* or *assrt* will raise an | 
					
						
							|  |  |  |        AttributeError. Passing `unsafe=True` will allow access to | 
					
						
							|  |  |  |       these attributes. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-04-13 20:51:20 +01:00
										 |  |  |     * `wraps`: Item for the mock object to wrap. If `wraps` is not None then | 
					
						
							|  |  |  |       calling the Mock will pass the call through to the wrapped object | 
					
						
							|  |  |  |       (returning the real result). Attribute access on the mock will return a | 
					
						
							|  |  |  |       Mock object that wraps the corresponding attribute of the wrapped object | 
					
						
							|  |  |  |       (so attempting to access an attribute that doesn't exist will raise an | 
					
						
							|  |  |  |       `AttributeError`). | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |       If the mock has an explicit `return_value` set then calls are not passed | 
					
						
							|  |  |  |       to the wrapped object and the `return_value` is returned instead. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     * `name`: If the mock has a name then it will be used in the repr of the | 
					
						
							|  |  |  |       mock. This can be useful for debugging. The name is propagated to child | 
					
						
							|  |  |  |       mocks. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Mocks can also be called with arbitrary keyword arguments. These will be | 
					
						
							|  |  |  |     used to set attributes on the mock after it is created. | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  | # _check_spec_arg_typos takes kwargs from commands like patch and checks that | 
					
						
							|  |  |  | # they don't contain common misspellings of arguments related to autospeccing. | 
					
						
							|  |  |  | def _check_spec_arg_typos(kwargs_to_check): | 
					
						
							|  |  |  |     typos = ("autospect", "auto_spec", "set_spec") | 
					
						
							|  |  |  |     for typo in typos: | 
					
						
							|  |  |  |         if typo in kwargs_to_check: | 
					
						
							|  |  |  |             raise RuntimeError( | 
					
						
							|  |  |  |                 f"{typo!r} might be a typo; use unsafe=True if this is intended" | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | class _patch(object): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     attribute_name = None | 
					
						
							| 
									
										
										
										
											2014-04-15 17:21:08 -04:00
										 |  |  |     _active_patches = [] | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def __init__( | 
					
						
							|  |  |  |             self, getter, attribute, new, spec, create, | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |             spec_set, autospec, new_callable, kwargs, *, unsafe=False | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         ): | 
					
						
							|  |  |  |         if new_callable is not None: | 
					
						
							|  |  |  |             if new is not DEFAULT: | 
					
						
							|  |  |  |                 raise ValueError( | 
					
						
							|  |  |  |                     "Cannot use 'new' and 'new_callable' together" | 
					
						
							|  |  |  |                 ) | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |             if autospec is not None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 raise ValueError( | 
					
						
							|  |  |  |                     "Cannot use 'autospec' and 'new_callable' together" | 
					
						
							|  |  |  |                 ) | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |         if not unsafe: | 
					
						
							|  |  |  |             _check_spec_arg_typos(kwargs) | 
					
						
							| 
									
										
										
										
											2021-04-09 23:45:50 -04:00
										 |  |  |         if _is_instance_mock(spec): | 
					
						
							|  |  |  |             raise InvalidSpecError( | 
					
						
							|  |  |  |                 f'Cannot spec attr {attribute!r} as the spec ' | 
					
						
							|  |  |  |                 f'has already been mocked out. [spec={spec!r}]') | 
					
						
							|  |  |  |         if _is_instance_mock(spec_set): | 
					
						
							|  |  |  |             raise InvalidSpecError( | 
					
						
							|  |  |  |                 f'Cannot spec attr {attribute!r} as the spec_set ' | 
					
						
							|  |  |  |                 f'target has already been mocked out. [spec_set={spec_set!r}]') | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         self.getter = getter | 
					
						
							|  |  |  |         self.attribute = attribute | 
					
						
							|  |  |  |         self.new = new | 
					
						
							|  |  |  |         self.new_callable = new_callable | 
					
						
							|  |  |  |         self.spec = spec | 
					
						
							|  |  |  |         self.create = create | 
					
						
							|  |  |  |         self.has_local = False | 
					
						
							|  |  |  |         self.spec_set = spec_set | 
					
						
							|  |  |  |         self.autospec = autospec | 
					
						
							|  |  |  |         self.kwargs = kwargs | 
					
						
							|  |  |  |         self.additional_patchers = [] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def copy(self): | 
					
						
							|  |  |  |         patcher = _patch( | 
					
						
							|  |  |  |             self.getter, self.attribute, self.new, self.spec, | 
					
						
							|  |  |  |             self.create, self.spec_set, | 
					
						
							|  |  |  |             self.autospec, self.new_callable, self.kwargs | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  |         patcher.attribute_name = self.attribute_name | 
					
						
							|  |  |  |         patcher.additional_patchers = [ | 
					
						
							|  |  |  |             p.copy() for p in self.additional_patchers | 
					
						
							|  |  |  |         ] | 
					
						
							|  |  |  |         return patcher | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __call__(self, func): | 
					
						
							|  |  |  |         if isinstance(func, type): | 
					
						
							|  |  |  |             return self.decorate_class(func) | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  |         if inspect.iscoroutinefunction(func): | 
					
						
							|  |  |  |             return self.decorate_async_callable(func) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return self.decorate_callable(func) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def decorate_class(self, klass): | 
					
						
							|  |  |  |         for attr in dir(klass): | 
					
						
							|  |  |  |             if not attr.startswith(patch.TEST_PREFIX): | 
					
						
							|  |  |  |                 continue | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             attr_value = getattr(klass, attr) | 
					
						
							|  |  |  |             if not hasattr(attr_value, "__call__"): | 
					
						
							|  |  |  |                 continue | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             patcher = self.copy() | 
					
						
							|  |  |  |             setattr(klass, attr, patcher(attr_value)) | 
					
						
							|  |  |  |         return klass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  |     @contextlib.contextmanager | 
					
						
							|  |  |  |     def decoration_helper(self, patched, args, keywargs): | 
					
						
							|  |  |  |         extra_args = [] | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |         with contextlib.ExitStack() as exit_stack: | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  |             for patching in patched.patchings: | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |                 arg = exit_stack.enter_context(patching) | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  |                 if patching.attribute_name is not None: | 
					
						
							|  |  |  |                     keywargs.update(arg) | 
					
						
							|  |  |  |                 elif patching.new is DEFAULT: | 
					
						
							|  |  |  |                     extra_args.append(arg) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             args += tuple(extra_args) | 
					
						
							|  |  |  |             yield (args, keywargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def decorate_callable(self, func): | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  |         # NB. Keep the method in sync with decorate_async_callable() | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         if hasattr(func, 'patchings'): | 
					
						
							|  |  |  |             func.patchings.append(self) | 
					
						
							|  |  |  |             return func | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         @wraps(func) | 
					
						
							|  |  |  |         def patched(*args, **keywargs): | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  |             with self.decoration_helper(patched, | 
					
						
							|  |  |  |                                         args, | 
					
						
							|  |  |  |                                         keywargs) as (newargs, newkeywargs): | 
					
						
							|  |  |  |                 return func(*newargs, **newkeywargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-28 12:37:39 +05:30
										 |  |  |         patched.patchings = [self] | 
					
						
							|  |  |  |         return patched | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def decorate_async_callable(self, func): | 
					
						
							|  |  |  |         # NB. Keep the method in sync with decorate_callable() | 
					
						
							|  |  |  |         if hasattr(func, 'patchings'): | 
					
						
							|  |  |  |             func.patchings.append(self) | 
					
						
							|  |  |  |             return func | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         @wraps(func) | 
					
						
							|  |  |  |         async def patched(*args, **keywargs): | 
					
						
							|  |  |  |             with self.decoration_helper(patched, | 
					
						
							|  |  |  |                                         args, | 
					
						
							|  |  |  |                                         keywargs) as (newargs, newkeywargs): | 
					
						
							|  |  |  |                 return await func(*newargs, **newkeywargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         patched.patchings = [self] | 
					
						
							|  |  |  |         return patched | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def get_original(self): | 
					
						
							|  |  |  |         target = self.getter() | 
					
						
							|  |  |  |         name = self.attribute | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         original = DEFAULT | 
					
						
							|  |  |  |         local = False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             original = target.__dict__[name] | 
					
						
							|  |  |  |         except (AttributeError, KeyError): | 
					
						
							|  |  |  |             original = getattr(target, name, DEFAULT) | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             local = True | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-04-14 16:25:20 -04:00
										 |  |  |         if name in _builtins and isinstance(target, ModuleType): | 
					
						
							|  |  |  |             self.create = True | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         if not self.create and original is DEFAULT: | 
					
						
							|  |  |  |             raise AttributeError( | 
					
						
							|  |  |  |                 "%s does not have the attribute %r" % (target, name) | 
					
						
							|  |  |  |             ) | 
					
						
							|  |  |  |         return original, local | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __enter__(self): | 
					
						
							|  |  |  |         """Perform the patch.""" | 
					
						
							|  |  |  |         new, spec, spec_set = self.new, self.spec, self.spec_set | 
					
						
							|  |  |  |         autospec, kwargs = self.autospec, self.kwargs | 
					
						
							|  |  |  |         new_callable = self.new_callable | 
					
						
							|  |  |  |         self.target = self.getter() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |         # normalise False to None | 
					
						
							|  |  |  |         if spec is False: | 
					
						
							|  |  |  |             spec = None | 
					
						
							|  |  |  |         if spec_set is False: | 
					
						
							|  |  |  |             spec_set = None | 
					
						
							|  |  |  |         if autospec is False: | 
					
						
							|  |  |  |             autospec = None | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if spec is not None and autospec is not None: | 
					
						
							|  |  |  |             raise TypeError("Can't specify spec and autospec") | 
					
						
							|  |  |  |         if ((spec is not None or autospec is not None) and | 
					
						
							|  |  |  |             spec_set not in (True, None)): | 
					
						
							|  |  |  |             raise TypeError("Can't provide explicit spec_set *and* spec or autospec") | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         original, local = self.get_original() | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |         if new is DEFAULT and autospec is None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             inherit = False | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |             if spec is True: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 # set spec to the object we are replacing | 
					
						
							|  |  |  |                 spec = original | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |                 if spec_set is True: | 
					
						
							|  |  |  |                     spec_set = original | 
					
						
							|  |  |  |                     spec = None | 
					
						
							|  |  |  |             elif spec is not None: | 
					
						
							|  |  |  |                 if spec_set is True: | 
					
						
							|  |  |  |                     spec_set = spec | 
					
						
							|  |  |  |                     spec = None | 
					
						
							|  |  |  |             elif spec_set is True: | 
					
						
							|  |  |  |                 spec_set = original | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |             if spec is not None or spec_set is not None: | 
					
						
							|  |  |  |                 if original is DEFAULT: | 
					
						
							|  |  |  |                     raise TypeError("Can't use 'spec' with create=True") | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 if isinstance(original, type): | 
					
						
							|  |  |  |                     # If we're patching out a class and there is a spec | 
					
						
							|  |  |  |                     inherit = True | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |             if spec is None and _is_async_obj(original): | 
					
						
							|  |  |  |                 Klass = AsyncMock | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 Klass = MagicMock | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             _kwargs = {} | 
					
						
							|  |  |  |             if new_callable is not None: | 
					
						
							|  |  |  |                 Klass = new_callable | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |             elif spec is not None or spec_set is not None: | 
					
						
							| 
									
										
										
										
											2012-03-25 19:53:18 +01:00
										 |  |  |                 this_spec = spec | 
					
						
							|  |  |  |                 if spec_set is not None: | 
					
						
							|  |  |  |                     this_spec = spec_set | 
					
						
							|  |  |  |                 if _is_list(this_spec): | 
					
						
							|  |  |  |                     not_callable = '__call__' not in this_spec | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     not_callable = not callable(this_spec) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |                 if _is_async_obj(this_spec): | 
					
						
							|  |  |  |                     Klass = AsyncMock | 
					
						
							|  |  |  |                 elif not_callable: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                     Klass = NonCallableMagicMock | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             if spec is not None: | 
					
						
							|  |  |  |                 _kwargs['spec'] = spec | 
					
						
							|  |  |  |             if spec_set is not None: | 
					
						
							|  |  |  |                 _kwargs['spec_set'] = spec_set | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             # add a name to mocks | 
					
						
							|  |  |  |             if (isinstance(Klass, type) and | 
					
						
							|  |  |  |                 issubclass(Klass, NonCallableMock) and self.attribute): | 
					
						
							|  |  |  |                 _kwargs['name'] = self.attribute | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             _kwargs.update(kwargs) | 
					
						
							|  |  |  |             new = Klass(**_kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             if inherit and _is_instance_mock(new): | 
					
						
							|  |  |  |                 # we can only tell if the instance should be callable if the | 
					
						
							|  |  |  |                 # spec is not a list | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |                 this_spec = spec | 
					
						
							|  |  |  |                 if spec_set is not None: | 
					
						
							|  |  |  |                     this_spec = spec_set | 
					
						
							|  |  |  |                 if (not _is_list(this_spec) and not | 
					
						
							|  |  |  |                     _instance_callable(this_spec)): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                     Klass = NonCallableMagicMock | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |                 _kwargs.pop('name') | 
					
						
							|  |  |  |                 new.return_value = Klass(_new_parent=new, _new_name='()', | 
					
						
							|  |  |  |                                          **_kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |         elif autospec is not None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             # spec is ignored, new *must* be default, spec_set is treated | 
					
						
							|  |  |  |             # as a boolean. Should we check spec is not None and that spec_set | 
					
						
							|  |  |  |             # is a bool? | 
					
						
							|  |  |  |             if new is not DEFAULT: | 
					
						
							|  |  |  |                 raise TypeError( | 
					
						
							|  |  |  |                     "autospec creates the mock for you. Can't specify " | 
					
						
							|  |  |  |                     "autospec and new." | 
					
						
							|  |  |  |                 ) | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |             if original is DEFAULT: | 
					
						
							| 
									
										
										
										
											2012-03-25 19:07:33 +01:00
										 |  |  |                 raise TypeError("Can't use 'autospec' with create=True") | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             spec_set = bool(spec_set) | 
					
						
							|  |  |  |             if autospec is True: | 
					
						
							|  |  |  |                 autospec = original | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-04-09 23:45:50 -04:00
										 |  |  |             if _is_instance_mock(self.target): | 
					
						
							|  |  |  |                 raise InvalidSpecError( | 
					
						
							|  |  |  |                     f'Cannot autospec attr {self.attribute!r} as the patch ' | 
					
						
							|  |  |  |                     f'target has already been mocked out. ' | 
					
						
							|  |  |  |                     f'[target={self.target!r}, attr={autospec!r}]') | 
					
						
							|  |  |  |             if _is_instance_mock(autospec): | 
					
						
							|  |  |  |                 target_name = getattr(self.target, '__name__', self.target) | 
					
						
							|  |  |  |                 raise InvalidSpecError( | 
					
						
							|  |  |  |                     f'Cannot autospec attr {self.attribute!r} from target ' | 
					
						
							|  |  |  |                     f'{target_name!r} as it has already been mocked out. ' | 
					
						
							|  |  |  |                     f'[target={self.target!r}, attr={autospec!r}]') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             new = create_autospec(autospec, spec_set=spec_set, | 
					
						
							|  |  |  |                                   _name=self.attribute, **kwargs) | 
					
						
							|  |  |  |         elif kwargs: | 
					
						
							|  |  |  |             # can't set keyword args when we aren't creating the mock | 
					
						
							|  |  |  |             # XXXX If new is a Mock we could call new.configure_mock(**kwargs) | 
					
						
							|  |  |  |             raise TypeError("Can't pass kwargs to a mock we aren't creating") | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         new_attr = new | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         self.temp_original = original | 
					
						
							|  |  |  |         self.is_local = local | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |         self._exit_stack = contextlib.ExitStack() | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             setattr(self.target, self.attribute, new_attr) | 
					
						
							|  |  |  |             if self.attribute_name is not None: | 
					
						
							|  |  |  |                 extra_args = {} | 
					
						
							|  |  |  |                 if self.new is DEFAULT: | 
					
						
							|  |  |  |                     extra_args[self.attribute_name] =  new | 
					
						
							|  |  |  |                 for patching in self.additional_patchers: | 
					
						
							|  |  |  |                     arg = self._exit_stack.enter_context(patching) | 
					
						
							|  |  |  |                     if patching.new is DEFAULT: | 
					
						
							|  |  |  |                         extra_args.update(arg) | 
					
						
							|  |  |  |                 return extra_args | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |             return new | 
					
						
							|  |  |  |         except: | 
					
						
							|  |  |  |             if not self.__exit__(*sys.exc_info()): | 
					
						
							|  |  |  |                 raise | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |     def __exit__(self, *exc_info): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         """Undo the patch.""" | 
					
						
							|  |  |  |         if self.is_local and self.temp_original is not DEFAULT: | 
					
						
							|  |  |  |             setattr(self.target, self.attribute, self.temp_original) | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             delattr(self.target, self.attribute) | 
					
						
							| 
									
										
										
										
											2016-01-08 23:43:29 -08:00
										 |  |  |             if not self.create and (not hasattr(self.target, self.attribute) or | 
					
						
							|  |  |  |                         self.attribute in ('__doc__', '__module__', | 
					
						
							|  |  |  |                                            '__defaults__', '__annotations__', | 
					
						
							|  |  |  |                                            '__kwdefaults__')): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 # needed for proxy objects like django settings | 
					
						
							|  |  |  |                 setattr(self.target, self.attribute, self.temp_original) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         del self.temp_original | 
					
						
							|  |  |  |         del self.is_local | 
					
						
							|  |  |  |         del self.target | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |         exit_stack = self._exit_stack | 
					
						
							|  |  |  |         del self._exit_stack | 
					
						
							|  |  |  |         return exit_stack.__exit__(*exc_info) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-06-10 20:36:32 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def start(self): | 
					
						
							|  |  |  |         """Activate a patch, returning any created mock.""" | 
					
						
							|  |  |  |         result = self.__enter__() | 
					
						
							| 
									
										
										
										
											2014-04-15 17:21:08 -04:00
										 |  |  |         self._active_patches.append(self) | 
					
						
							| 
									
										
										
										
											2012-06-10 20:36:32 +01:00
										 |  |  |         return result | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def stop(self): | 
					
						
							|  |  |  |         """Stop an active patch.""" | 
					
						
							| 
									
										
										
										
											2014-04-15 17:21:08 -04:00
										 |  |  |         try: | 
					
						
							|  |  |  |             self._active_patches.remove(self) | 
					
						
							|  |  |  |         except ValueError: | 
					
						
							|  |  |  |             # If the patch hasn't been started this will fail | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |             return None | 
					
						
							| 
									
										
										
										
											2014-04-15 17:21:08 -04:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |         return self.__exit__(None, None, None) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _get_target(target): | 
					
						
							|  |  |  |     try: | 
					
						
							|  |  |  |         target, attribute = target.rsplit('.', 1) | 
					
						
							| 
									
										
										
										
											2022-01-23 18:42:41 +00:00
										 |  |  |     except (TypeError, ValueError, AttributeError): | 
					
						
							|  |  |  |         raise TypeError( | 
					
						
							|  |  |  |             f"Need a valid target to patch. You supplied: {target!r}") | 
					
						
							| 
									
										
										
										
											2021-07-21 12:47:44 +01:00
										 |  |  |     return partial(pkgutil.resolve_name, target), attribute | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _patch_object( | 
					
						
							|  |  |  |         target, attribute, new=DEFAULT, spec=None, | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  |         create=False, spec_set=None, autospec=None, | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |         new_callable=None, *, unsafe=False, **kwargs | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     ): | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     patch the named member (`attribute`) on an object (`target`) with a mock | 
					
						
							|  |  |  |     object. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `patch.object` can be used as a decorator, class decorator or a context | 
					
						
							|  |  |  |     manager. Arguments `new`, `spec`, `create`, `spec_set`, | 
					
						
							|  |  |  |     `autospec` and `new_callable` have the same meaning as for `patch`. Like | 
					
						
							|  |  |  |     `patch`, `patch.object` takes arbitrary keyword arguments for configuring | 
					
						
							|  |  |  |     the mock object it creates. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     When used as a class decorator `patch.object` honours `patch.TEST_PREFIX` | 
					
						
							|  |  |  |     for choosing which methods to wrap. | 
					
						
							|  |  |  |     """
 | 
					
						
							| 
									
										
										
										
											2019-12-08 12:14:38 -08:00
										 |  |  |     if type(target) is str: | 
					
						
							|  |  |  |         raise TypeError( | 
					
						
							|  |  |  |             f"{target!r} must be the actual object to be patched, not a str" | 
					
						
							|  |  |  |         ) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     getter = lambda: target | 
					
						
							|  |  |  |     return _patch( | 
					
						
							|  |  |  |         getter, attribute, new, spec, create, | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |         spec_set, autospec, new_callable, kwargs, unsafe=unsafe | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 18:57:58 +01:00
										 |  |  | def _patch_multiple(target, spec=None, create=False, spec_set=None, | 
					
						
							|  |  |  |                     autospec=None, new_callable=None, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     """Perform multiple patches in a single call. It takes the object to be
 | 
					
						
							|  |  |  |     patched (either as an object or a string to fetch the object by importing) | 
					
						
							|  |  |  |     and keyword arguments for the patches:: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         with patch.multiple(settings, FIRST_PATCH='one', SECOND_PATCH='two'): | 
					
						
							|  |  |  |             ... | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Use `DEFAULT` as the value if you want `patch.multiple` to create | 
					
						
							|  |  |  |     mocks for you. In this case the created mocks are passed into a decorated | 
					
						
							|  |  |  |     function by keyword, and a dictionary is returned when `patch.multiple` is | 
					
						
							|  |  |  |     used as a context manager. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `patch.multiple` can be used as a decorator, class decorator or a context | 
					
						
							|  |  |  |     manager. The arguments `spec`, `spec_set`, `create`, | 
					
						
							|  |  |  |     `autospec` and `new_callable` have the same meaning as for `patch`. These | 
					
						
							|  |  |  |     arguments will be applied to *all* patches done by `patch.multiple`. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     When used as a class decorator `patch.multiple` honours `patch.TEST_PREFIX` | 
					
						
							|  |  |  |     for choosing which methods to wrap. | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     if type(target) is str: | 
					
						
							| 
									
										
										
										
											2021-07-21 12:47:44 +01:00
										 |  |  |         getter = partial(pkgutil.resolve_name, target) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     else: | 
					
						
							|  |  |  |         getter = lambda: target | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if not kwargs: | 
					
						
							|  |  |  |         raise ValueError( | 
					
						
							|  |  |  |             'Must supply at least one keyword argument with patch.multiple' | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  |     # need to wrap in a list for python 3, where items is a view | 
					
						
							|  |  |  |     items = list(kwargs.items()) | 
					
						
							|  |  |  |     attribute, new = items[0] | 
					
						
							|  |  |  |     patcher = _patch( | 
					
						
							|  |  |  |         getter, attribute, new, spec, create, spec_set, | 
					
						
							|  |  |  |         autospec, new_callable, {} | 
					
						
							|  |  |  |     ) | 
					
						
							|  |  |  |     patcher.attribute_name = attribute | 
					
						
							|  |  |  |     for attribute, new in items[1:]: | 
					
						
							|  |  |  |         this_patcher = _patch( | 
					
						
							|  |  |  |             getter, attribute, new, spec, create, spec_set, | 
					
						
							|  |  |  |             autospec, new_callable, {} | 
					
						
							|  |  |  |         ) | 
					
						
							|  |  |  |         this_patcher.attribute_name = attribute | 
					
						
							|  |  |  |         patcher.additional_patchers.append(this_patcher) | 
					
						
							|  |  |  |     return patcher | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def patch( | 
					
						
							|  |  |  |         target, new=DEFAULT, spec=None, create=False, | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |         spec_set=None, autospec=None, new_callable=None, *, unsafe=False, **kwargs | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     ): | 
					
						
							|  |  |  |     """
 | 
					
						
							| 
									
										
										
										
											2012-03-28 15:08:08 +01:00
										 |  |  |     `patch` acts as a function decorator, class decorator or a context | 
					
						
							|  |  |  |     manager. Inside the body of the function or with statement, the `target` | 
					
						
							|  |  |  |     is patched with a `new` object. When the function/with statement exits | 
					
						
							|  |  |  |     the patch is undone. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-09 15:18:06 +01:00
										 |  |  |     If `new` is omitted, then the target is replaced with an | 
					
						
							|  |  |  |     `AsyncMock if the patched object is an async function or a | 
					
						
							|  |  |  |     `MagicMock` otherwise. If `patch` is used as a decorator and `new` is | 
					
						
							| 
									
										
										
										
											2012-03-28 15:08:08 +01:00
										 |  |  |     omitted, the created mock is passed in as an extra argument to the | 
					
						
							|  |  |  |     decorated function. If `patch` is used as a context manager the created | 
					
						
							|  |  |  |     mock is returned by the context manager. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `target` should be a string in the form `'package.module.ClassName'`. The | 
					
						
							|  |  |  |     `target` is imported and the specified object replaced with the `new` | 
					
						
							|  |  |  |     object, so the `target` must be importable from the environment you are | 
					
						
							|  |  |  |     calling `patch` from. The target is imported when the decorated function | 
					
						
							|  |  |  |     is executed, not at decoration time. | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     The `spec` and `spec_set` keyword arguments are passed to the `MagicMock` | 
					
						
							|  |  |  |     if patch is creating one for you. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     In addition you can pass `spec=True` or `spec_set=True`, which causes | 
					
						
							|  |  |  |     patch to pass in the object being mocked as the spec/spec_set object. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `new_callable` allows you to specify a different class, or callable object, | 
					
						
							| 
									
										
										
										
											2019-09-09 15:18:06 +01:00
										 |  |  |     that will be called to create the `new` object. By default `AsyncMock` is | 
					
						
							|  |  |  |     used for async functions and `MagicMock` for the rest. | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     A more powerful form of `spec` is `autospec`. If you set `autospec=True` | 
					
						
							| 
									
										
										
										
											2015-07-17 21:58:36 +12:00
										 |  |  |     then the mock will be created with a spec from the object being replaced. | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     All attributes of the mock will also have the spec of the corresponding | 
					
						
							|  |  |  |     attribute of the object being replaced. Methods and functions being | 
					
						
							|  |  |  |     mocked will have their arguments checked and will raise a `TypeError` if | 
					
						
							|  |  |  |     they are called with the wrong signature. For mocks replacing a class, | 
					
						
							|  |  |  |     their return value (the 'instance') will have the same spec as the class. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Instead of `autospec=True` you can pass `autospec=some_object` to use an | 
					
						
							|  |  |  |     arbitrary object as the spec instead of the one being replaced. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     By default `patch` will fail to replace attributes that don't exist. If | 
					
						
							|  |  |  |     you pass in `create=True`, and the attribute doesn't exist, patch will | 
					
						
							|  |  |  |     create the attribute for you when the patched function is called, and | 
					
						
							|  |  |  |     delete it again afterwards. This is useful for writing tests against | 
					
						
							| 
									
										
										
										
											2013-03-11 18:34:00 -04:00
										 |  |  |     attributes that your production code creates at runtime. It is off by | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     default because it can be dangerous. With it switched on you can write | 
					
						
							|  |  |  |     passing tests against APIs that don't actually exist! | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Patch can be used as a `TestCase` class decorator. It works by | 
					
						
							|  |  |  |     decorating each test method in the class. This reduces the boilerplate | 
					
						
							|  |  |  |     code when your test methods share a common patchings set. `patch` finds | 
					
						
							|  |  |  |     tests by looking for method names that start with `patch.TEST_PREFIX`. | 
					
						
							|  |  |  |     By default this is `test`, which matches the way `unittest` finds tests. | 
					
						
							|  |  |  |     You can specify an alternative prefix by setting `patch.TEST_PREFIX`. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Patch can be used as a context manager, with the with statement. Here the | 
					
						
							|  |  |  |     patching applies to the indented block after the with statement. If you | 
					
						
							|  |  |  |     use "as" then the patched object will be bound to the name after the | 
					
						
							|  |  |  |     "as"; very useful if `patch` is creating a mock object for you. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |     Patch will raise a `RuntimeError` if passed some common misspellings of | 
					
						
							|  |  |  |     the arguments autospec and spec_set. Pass the argument `unsafe` with the | 
					
						
							|  |  |  |     value True to disable that check. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     `patch` takes arbitrary keyword arguments. These will be passed to | 
					
						
							| 
									
										
										
										
											2020-01-25 07:53:54 -03:00
										 |  |  |     `AsyncMock` if the patched object is asynchronous, to `MagicMock` | 
					
						
							|  |  |  |     otherwise or to `new_callable` if specified. | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     `patch.dict(...)`, `patch.multiple(...)` and `patch.object(...)` are | 
					
						
							|  |  |  |     available for alternate use-cases. | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     getter, attribute = _get_target(target) | 
					
						
							|  |  |  |     return _patch( | 
					
						
							|  |  |  |         getter, attribute, new, spec, create, | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |         spec_set, autospec, new_callable, kwargs, unsafe=unsafe | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class _patch_dict(object): | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     Patch a dictionary, or dictionary like object, and restore the dictionary | 
					
						
							|  |  |  |     to its original state after the test. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `in_dict` can be a dictionary or a mapping like container. If it is a | 
					
						
							|  |  |  |     mapping then it must at least support getting, setting and deleting items | 
					
						
							|  |  |  |     plus iterating over keys. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `in_dict` can also be a string specifying the name of the dictionary, which | 
					
						
							|  |  |  |     will then be fetched by importing it. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `values` can be a dictionary of values to set in the dictionary. `values` | 
					
						
							|  |  |  |     can also be an iterable of `(key, value)` pairs. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If `clear` is True then the dictionary will be cleared before the new | 
					
						
							|  |  |  |     values are set. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `patch.dict` can also be called with arbitrary keyword arguments to set | 
					
						
							|  |  |  |     values in the dictionary:: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         with patch.dict('sys.modules', mymodule=Mock(), other_module=Mock()): | 
					
						
							|  |  |  |             ... | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     `patch.dict` can be used as a context manager, decorator or class | 
					
						
							|  |  |  |     decorator. When used as a class decorator `patch.dict` honours | 
					
						
							|  |  |  |     `patch.TEST_PREFIX` for choosing which methods to wrap. | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, in_dict, values=(), clear=False, **kwargs): | 
					
						
							|  |  |  |         self.in_dict = in_dict | 
					
						
							|  |  |  |         # support any argument supported by dict(...) constructor | 
					
						
							|  |  |  |         self.values = dict(values) | 
					
						
							|  |  |  |         self.values.update(kwargs) | 
					
						
							|  |  |  |         self.clear = clear | 
					
						
							|  |  |  |         self._original = None | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __call__(self, f): | 
					
						
							|  |  |  |         if isinstance(f, type): | 
					
						
							|  |  |  |             return self.decorate_class(f) | 
					
						
							| 
									
										
										
										
											2022-11-11 11:04:30 +03:00
										 |  |  |         if inspect.iscoroutinefunction(f): | 
					
						
							|  |  |  |             return self.decorate_async_callable(f) | 
					
						
							|  |  |  |         return self.decorate_callable(f) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def decorate_callable(self, f): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         @wraps(f) | 
					
						
							|  |  |  |         def _inner(*args, **kw): | 
					
						
							|  |  |  |             self._patch_dict() | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 return f(*args, **kw) | 
					
						
							|  |  |  |             finally: | 
					
						
							|  |  |  |                 self._unpatch_dict() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         return _inner | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-11-11 11:04:30 +03:00
										 |  |  |     def decorate_async_callable(self, f): | 
					
						
							|  |  |  |         @wraps(f) | 
					
						
							|  |  |  |         async def _inner(*args, **kw): | 
					
						
							|  |  |  |             self._patch_dict() | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 return await f(*args, **kw) | 
					
						
							|  |  |  |             finally: | 
					
						
							|  |  |  |                 self._unpatch_dict() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         return _inner | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def decorate_class(self, klass): | 
					
						
							|  |  |  |         for attr in dir(klass): | 
					
						
							|  |  |  |             attr_value = getattr(klass, attr) | 
					
						
							|  |  |  |             if (attr.startswith(patch.TEST_PREFIX) and | 
					
						
							|  |  |  |                  hasattr(attr_value, "__call__")): | 
					
						
							|  |  |  |                 decorator = _patch_dict(self.in_dict, self.values, self.clear) | 
					
						
							|  |  |  |                 decorated = decorator(attr_value) | 
					
						
							|  |  |  |                 setattr(klass, attr, decorated) | 
					
						
							|  |  |  |         return klass | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __enter__(self): | 
					
						
							|  |  |  |         """Patch the dict.""" | 
					
						
							|  |  |  |         self._patch_dict() | 
					
						
							| 
									
										
										
										
											2019-05-28 13:53:31 +01:00
										 |  |  |         return self.in_dict | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _patch_dict(self): | 
					
						
							|  |  |  |         values = self.values | 
					
						
							| 
									
										
										
										
											2019-02-25 00:24:49 +05:30
										 |  |  |         if isinstance(self.in_dict, str): | 
					
						
							| 
									
										
										
										
											2021-07-21 12:47:44 +01:00
										 |  |  |             self.in_dict = pkgutil.resolve_name(self.in_dict) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         in_dict = self.in_dict | 
					
						
							|  |  |  |         clear = self.clear | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             original = in_dict.copy() | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							|  |  |  |             # dict like object with no copy method | 
					
						
							|  |  |  |             # must support iteration over keys | 
					
						
							|  |  |  |             original = {} | 
					
						
							|  |  |  |             for key in in_dict: | 
					
						
							|  |  |  |                 original[key] = in_dict[key] | 
					
						
							|  |  |  |         self._original = original | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if clear: | 
					
						
							|  |  |  |             _clear_dict(in_dict) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             in_dict.update(values) | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							|  |  |  |             # dict like object with no update method | 
					
						
							|  |  |  |             for key in values: | 
					
						
							|  |  |  |                 in_dict[key] = values[key] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _unpatch_dict(self): | 
					
						
							|  |  |  |         in_dict = self.in_dict | 
					
						
							|  |  |  |         original = self._original | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         _clear_dict(in_dict) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             in_dict.update(original) | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							|  |  |  |             for key in original: | 
					
						
							|  |  |  |                 in_dict[key] = original[key] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __exit__(self, *args): | 
					
						
							|  |  |  |         """Unpatch the dict.""" | 
					
						
							| 
									
										
										
										
											2020-01-29 16:24:54 +00:00
										 |  |  |         if self._original is not None: | 
					
						
							|  |  |  |             self._unpatch_dict() | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return False | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-01-24 08:38:33 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def start(self): | 
					
						
							|  |  |  |         """Activate a patch, returning any created mock.""" | 
					
						
							|  |  |  |         result = self.__enter__() | 
					
						
							|  |  |  |         _patch._active_patches.append(self) | 
					
						
							|  |  |  |         return result | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def stop(self): | 
					
						
							|  |  |  |         """Stop an active patch.""" | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             _patch._active_patches.remove(self) | 
					
						
							|  |  |  |         except ValueError: | 
					
						
							|  |  |  |             # If the patch hasn't been started this will fail | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |             return None | 
					
						
							| 
									
										
										
										
											2020-01-24 08:38:33 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-04-11 10:59:24 +03:00
										 |  |  |         return self.__exit__(None, None, None) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _clear_dict(in_dict): | 
					
						
							|  |  |  |     try: | 
					
						
							|  |  |  |         in_dict.clear() | 
					
						
							|  |  |  |     except AttributeError: | 
					
						
							|  |  |  |         keys = list(in_dict) | 
					
						
							|  |  |  |         for key in keys: | 
					
						
							|  |  |  |             del in_dict[key] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-06-10 20:36:32 +01:00
										 |  |  | def _patch_stopall(): | 
					
						
							| 
									
										
										
										
											2014-04-15 17:21:08 -04:00
										 |  |  |     """Stop all active patches. LIFO to unroll nested patches.""" | 
					
						
							|  |  |  |     for patch in reversed(_patch._active_patches): | 
					
						
							| 
									
										
										
										
											2012-06-10 20:36:32 +01:00
										 |  |  |         patch.stop() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | patch.object = _patch_object | 
					
						
							|  |  |  | patch.dict = _patch_dict | 
					
						
							|  |  |  | patch.multiple = _patch_multiple | 
					
						
							| 
									
										
										
										
											2012-06-10 20:36:32 +01:00
										 |  |  | patch.stopall = _patch_stopall | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | patch.TEST_PREFIX = 'test' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | magic_methods = ( | 
					
						
							|  |  |  |     "lt le gt ge eq ne " | 
					
						
							|  |  |  |     "getitem setitem delitem " | 
					
						
							|  |  |  |     "len contains iter " | 
					
						
							|  |  |  |     "hash str sizeof " | 
					
						
							|  |  |  |     "enter exit " | 
					
						
							| 
									
										
										
										
											2015-03-15 01:51:56 +02:00
										 |  |  |     # we added divmod and rdivmod here instead of numerics | 
					
						
							|  |  |  |     # because there is no idivmod | 
					
						
							|  |  |  |     "divmod rdivmod neg pos abs invert " | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     "complex int float index " | 
					
						
							| 
									
										
										
										
											2018-05-22 13:01:10 -07:00
										 |  |  |     "round trunc floor ceil " | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     "bool next " | 
					
						
							| 
									
										
										
										
											2018-10-25 14:48:58 -07:00
										 |  |  |     "fspath " | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |     "aiter " | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-04-14 11:23:48 -04:00
										 |  |  | numerics = ( | 
					
						
							| 
									
										
										
										
											2021-08-26 21:19:47 +03:00
										 |  |  |     "add sub mul matmul truediv floordiv mod lshift rshift and xor or pow" | 
					
						
							| 
									
										
										
										
											2014-04-14 11:23:48 -04:00
										 |  |  | ) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | inplace = ' '.join('i%s' % n for n in numerics.split()) | 
					
						
							|  |  |  | right = ' '.join('r%s' % n for n in numerics.split()) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | # not including __prepare__, __instancecheck__, __subclasscheck__ | 
					
						
							|  |  |  | # (as they are metaclass methods) | 
					
						
							|  |  |  | # __del__ is not supported at all as it causes problems if it exists | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-12-11 10:28:14 +02:00
										 |  |  | _non_defaults = { | 
					
						
							|  |  |  |     '__get__', '__set__', '__delete__', '__reversed__', '__missing__', | 
					
						
							|  |  |  |     '__reduce__', '__reduce_ex__', '__getinitargs__', '__getnewargs__', | 
					
						
							| 
									
										
										
										
											2022-02-26 00:53:27 +01:00
										 |  |  |     '__getstate__', '__setstate__', '__getformat__', | 
					
						
							| 
									
										
										
										
											2014-12-11 10:28:14 +02:00
										 |  |  |     '__repr__', '__dir__', '__subclasses__', '__format__', | 
					
						
							| 
									
										
										
										
											2019-09-19 21:04:18 -07:00
										 |  |  |     '__getnewargs_ex__', | 
					
						
							| 
									
										
										
										
											2014-12-11 22:25:49 +01:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _get_method(name, func): | 
					
						
							|  |  |  |     "Turns a callable object (like a mock) into a real function" | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def method(self, /, *args, **kw): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return func(self, *args, **kw) | 
					
						
							|  |  |  |     method.__name__ = name | 
					
						
							|  |  |  |     return method | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-12-11 10:28:14 +02:00
										 |  |  | _magics = { | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     '__%s__' % method for method in | 
					
						
							|  |  |  |     ' '.join([magic_methods, numerics, inplace, right]).split() | 
					
						
							| 
									
										
										
										
											2014-12-11 10:28:14 +02:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | # Magic methods used for async `with` statements | 
					
						
							|  |  |  | _async_method_magics = {"__aenter__", "__aexit__", "__anext__"} | 
					
						
							| 
									
										
										
										
											2019-09-19 21:04:18 -07:00
										 |  |  | # Magic methods that are only used with async calls but are synchronous functions themselves | 
					
						
							|  |  |  | _sync_async_magics = {"__aiter__"} | 
					
						
							|  |  |  | _async_magics = _async_method_magics | _sync_async_magics | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-19 21:04:18 -07:00
										 |  |  | _all_sync_magics = _magics | _non_defaults | 
					
						
							|  |  |  | _all_magics = _all_sync_magics | _async_magics | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-12-11 10:28:14 +02:00
										 |  |  | _unsupported_magics = { | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     '__getattr__', '__setattr__', | 
					
						
							| 
									
										
										
										
											2018-11-05 16:20:25 +02:00
										 |  |  |     '__init__', '__new__', '__prepare__', | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     '__instancecheck__', '__subclasscheck__', | 
					
						
							|  |  |  |     '__del__' | 
					
						
							| 
									
										
										
										
											2014-12-11 10:28:14 +02:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | _calculate_return_value = { | 
					
						
							|  |  |  |     '__hash__': lambda self: object.__hash__(self), | 
					
						
							|  |  |  |     '__str__': lambda self: object.__str__(self), | 
					
						
							|  |  |  |     '__sizeof__': lambda self: object.__sizeof__(self), | 
					
						
							| 
									
										
										
										
											2018-10-25 14:48:58 -07:00
										 |  |  |     '__fspath__': lambda self: f"{type(self).__name__}/{self._extract_mock_name()}/{id(self)}", | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | _return_values = { | 
					
						
							| 
									
										
										
										
											2012-03-25 18:16:07 +01:00
										 |  |  |     '__lt__': NotImplemented, | 
					
						
							|  |  |  |     '__gt__': NotImplemented, | 
					
						
							|  |  |  |     '__le__': NotImplemented, | 
					
						
							|  |  |  |     '__ge__': NotImplemented, | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     '__int__': 1, | 
					
						
							|  |  |  |     '__contains__': False, | 
					
						
							|  |  |  |     '__len__': 0, | 
					
						
							|  |  |  |     '__exit__': False, | 
					
						
							|  |  |  |     '__complex__': 1j, | 
					
						
							|  |  |  |     '__float__': 1.0, | 
					
						
							|  |  |  |     '__bool__': True, | 
					
						
							|  |  |  |     '__index__': 1, | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |     '__aexit__': False, | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _get_eq(self): | 
					
						
							|  |  |  |     def __eq__(other): | 
					
						
							|  |  |  |         ret_val = self.__eq__._mock_return_value | 
					
						
							|  |  |  |         if ret_val is not DEFAULT: | 
					
						
							|  |  |  |             return ret_val | 
					
						
							| 
									
										
										
										
											2017-01-21 23:12:58 +02:00
										 |  |  |         if self is other: | 
					
						
							|  |  |  |             return True | 
					
						
							|  |  |  |         return NotImplemented | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     return __eq__ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _get_ne(self): | 
					
						
							|  |  |  |     def __ne__(other): | 
					
						
							|  |  |  |         if self.__ne__._mock_return_value is not DEFAULT: | 
					
						
							|  |  |  |             return DEFAULT | 
					
						
							| 
									
										
										
										
											2017-01-21 23:12:58 +02:00
										 |  |  |         if self is other: | 
					
						
							|  |  |  |             return False | 
					
						
							|  |  |  |         return NotImplemented | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     return __ne__ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _get_iter(self): | 
					
						
							|  |  |  |     def __iter__(): | 
					
						
							|  |  |  |         ret_val = self.__iter__._mock_return_value | 
					
						
							|  |  |  |         if ret_val is DEFAULT: | 
					
						
							|  |  |  |             return iter([]) | 
					
						
							|  |  |  |         # if ret_val was already an iterator, then calling iter on it should | 
					
						
							|  |  |  |         # return the iterator unchanged | 
					
						
							|  |  |  |         return iter(ret_val) | 
					
						
							|  |  |  |     return __iter__ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | def _get_async_iter(self): | 
					
						
							|  |  |  |     def __aiter__(): | 
					
						
							|  |  |  |         ret_val = self.__aiter__._mock_return_value | 
					
						
							|  |  |  |         if ret_val is DEFAULT: | 
					
						
							|  |  |  |             return _AsyncIterator(iter([])) | 
					
						
							|  |  |  |         return _AsyncIterator(iter(ret_val)) | 
					
						
							|  |  |  |     return __aiter__ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | _side_effect_methods = { | 
					
						
							|  |  |  |     '__eq__': _get_eq, | 
					
						
							|  |  |  |     '__ne__': _get_ne, | 
					
						
							|  |  |  |     '__iter__': _get_iter, | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |     '__aiter__': _get_async_iter | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _set_return_value(mock, method, name): | 
					
						
							|  |  |  |     fixed = _return_values.get(name, DEFAULT) | 
					
						
							|  |  |  |     if fixed is not DEFAULT: | 
					
						
							|  |  |  |         method.return_value = fixed | 
					
						
							|  |  |  |         return | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-16 11:34:46 -05:00
										 |  |  |     return_calculator = _calculate_return_value.get(name) | 
					
						
							|  |  |  |     if return_calculator is not None: | 
					
						
							|  |  |  |         return_value = return_calculator(mock) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         method.return_value = return_value | 
					
						
							|  |  |  |         return | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     side_effector = _side_effect_methods.get(name) | 
					
						
							|  |  |  |     if side_effector is not None: | 
					
						
							|  |  |  |         method.side_effect = side_effector(mock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  | class MagicMixin(Base): | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def __init__(self, /, *args, **kw): | 
					
						
							| 
									
										
										
										
											2015-04-13 23:12:42 -07:00
										 |  |  |         self._mock_set_magics()  # make magic work for kwargs in init | 
					
						
							| 
									
										
										
										
											2012-05-27 18:17:07 +10:00
										 |  |  |         _safe_super(MagicMixin, self).__init__(*args, **kw) | 
					
						
							| 
									
										
										
										
											2015-04-13 23:12:42 -07:00
										 |  |  |         self._mock_set_magics()  # fix magic broken by upper level init | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def _mock_set_magics(self): | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |         orig_magics = _magics | _async_method_magics | 
					
						
							|  |  |  |         these_magics = orig_magics | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-04-13 23:12:42 -07:00
										 |  |  |         if getattr(self, "_mock_methods", None) is not None: | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |             these_magics = orig_magics.intersection(self._mock_methods) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |             remove_magics = set() | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |             remove_magics = orig_magics - these_magics | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |             for entry in remove_magics: | 
					
						
							|  |  |  |                 if entry in type(self).__dict__: | 
					
						
							|  |  |  |                     # remove unneeded magic methods | 
					
						
							|  |  |  |                     delattr(self, entry) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # don't overwrite existing attributes if called a second time | 
					
						
							|  |  |  |         these_magics = these_magics - set(type(self).__dict__) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         _type = type(self) | 
					
						
							|  |  |  |         for entry in these_magics: | 
					
						
							|  |  |  |             setattr(_type, entry, MagicProxy(entry, self)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class NonCallableMagicMock(MagicMixin, NonCallableMock): | 
					
						
							|  |  |  |     """A version of `MagicMock` that isn't callable.""" | 
					
						
							|  |  |  |     def mock_add_spec(self, spec, spec_set=False): | 
					
						
							|  |  |  |         """Add a spec to a mock. `spec` can either be an object or a
 | 
					
						
							|  |  |  |         list of strings. Only attributes on the `spec` can be fetched as | 
					
						
							|  |  |  |         attributes from the mock. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If `spec_set` is True then only attributes on the spec can be set."""
 | 
					
						
							|  |  |  |         self._mock_add_spec(spec, spec_set) | 
					
						
							|  |  |  |         self._mock_set_magics() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  | class AsyncMagicMixin(MagicMixin): | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def __init__(self, /, *args, **kw): | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |         self._mock_set_magics()  # make magic work for kwargs in init | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         _safe_super(AsyncMagicMixin, self).__init__(*args, **kw) | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  |         self._mock_set_magics()  # fix magic broken by upper level init | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  | class MagicMock(MagicMixin, Mock): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     """
 | 
					
						
							|  |  |  |     MagicMock is a subclass of Mock with default implementations | 
					
						
							|  |  |  |     of most of the magic methods. You can use MagicMock without having to | 
					
						
							|  |  |  |     configure the magic methods yourself. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If you use the `spec` or `spec_set` arguments then *only* magic | 
					
						
							|  |  |  |     methods that exist in the spec will be created. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Attributes and the return value of a `MagicMock` will also be `MagicMocks`. | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     def mock_add_spec(self, spec, spec_set=False): | 
					
						
							|  |  |  |         """Add a spec to a mock. `spec` can either be an object or a
 | 
					
						
							|  |  |  |         list of strings. Only attributes on the `spec` can be fetched as | 
					
						
							|  |  |  |         attributes from the mock. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If `spec_set` is True then only attributes on the spec can be set."""
 | 
					
						
							|  |  |  |         self._mock_add_spec(spec, spec_set) | 
					
						
							|  |  |  |         self._mock_set_magics() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-28 18:42:44 -07:00
										 |  |  | class MagicProxy(Base): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def __init__(self, name, parent): | 
					
						
							|  |  |  |         self.name = name | 
					
						
							|  |  |  |         self.parent = parent | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def create_mock(self): | 
					
						
							|  |  |  |         entry = self.name | 
					
						
							|  |  |  |         parent = self.parent | 
					
						
							|  |  |  |         m = parent._get_child_mock(name=entry, _new_name=entry, | 
					
						
							|  |  |  |                                    _new_parent=parent) | 
					
						
							|  |  |  |         setattr(parent, entry, m) | 
					
						
							|  |  |  |         _set_return_value(parent, m, entry) | 
					
						
							|  |  |  |         return m | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __get__(self, obj, _type=None): | 
					
						
							|  |  |  |         return self.create_mock() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | class AsyncMockMixin(Base): | 
					
						
							|  |  |  |     await_count = _delegating_property('await_count') | 
					
						
							|  |  |  |     await_args = _delegating_property('await_args') | 
					
						
							|  |  |  |     await_args_list = _delegating_property('await_args_list') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def __init__(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         super().__init__(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |         # iscoroutinefunction() checks _is_coroutine property to say if an | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         # object is a coroutine. Without this check it looks to see if it is a | 
					
						
							|  |  |  |         # function/method, which in this case it is not (since it is an | 
					
						
							|  |  |  |         # AsyncMock). | 
					
						
							|  |  |  |         # It is set through __dict__ because when spec_set is True, this | 
					
						
							|  |  |  |         # attribute is likely undefined. | 
					
						
							|  |  |  |         self.__dict__['_is_coroutine'] = asyncio.coroutines._is_coroutine | 
					
						
							|  |  |  |         self.__dict__['_mock_await_count'] = 0 | 
					
						
							|  |  |  |         self.__dict__['_mock_await_args'] = None | 
					
						
							|  |  |  |         self.__dict__['_mock_await_args_list'] = _CallList() | 
					
						
							|  |  |  |         code_mock = NonCallableMock(spec_set=CodeType) | 
					
						
							|  |  |  |         code_mock.co_flags = inspect.CO_COROUTINE | 
					
						
							|  |  |  |         self.__dict__['__code__'] = code_mock | 
					
						
							| 
									
										
										
										
											2022-06-30 19:08:38 +02:00
										 |  |  |         self.__dict__['__name__'] = 'AsyncMock' | 
					
						
							|  |  |  |         self.__dict__['__defaults__'] = tuple() | 
					
						
							|  |  |  |         self.__dict__['__kwdefaults__'] = {} | 
					
						
							|  |  |  |         self.__dict__['__annotations__'] = None | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |     async def _execute_mock_call(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2020-01-29 16:24:54 +00:00
										 |  |  |         # This is nearly just like super(), except for special handling | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |         # of coroutines | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-03-11 20:36:12 +05:30
										 |  |  |         _call = _Call((args, kwargs), two=True) | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |         self.await_count += 1 | 
					
						
							|  |  |  |         self.await_args = _call | 
					
						
							|  |  |  |         self.await_args_list.append(_call) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |         effect = self.side_effect | 
					
						
							|  |  |  |         if effect is not None: | 
					
						
							|  |  |  |             if _is_exception(effect): | 
					
						
							|  |  |  |                 raise effect | 
					
						
							|  |  |  |             elif not _callable(effect): | 
					
						
							|  |  |  |                 try: | 
					
						
							|  |  |  |                     result = next(effect) | 
					
						
							|  |  |  |                 except StopIteration: | 
					
						
							| 
									
										
										
										
											2021-06-13 10:47:44 +08:00
										 |  |  |                     # It is impossible to propagate a StopIteration | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |                     # through coroutines because of PEP 479 | 
					
						
							|  |  |  |                     raise StopAsyncIteration | 
					
						
							|  |  |  |                 if _is_exception(result): | 
					
						
							|  |  |  |                     raise result | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |             elif iscoroutinefunction(effect): | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |                 result = await effect(*args, **kwargs) | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 result = effect(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |             if result is not DEFAULT: | 
					
						
							|  |  |  |                 return result | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if self._mock_return_value is not DEFAULT: | 
					
						
							|  |  |  |             return self.return_value | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if self._mock_wraps is not None: | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |             if iscoroutinefunction(self._mock_wraps): | 
					
						
							| 
									
										
										
										
											2019-11-20 16:27:51 -08:00
										 |  |  |                 return await self._mock_wraps(*args, **kwargs) | 
					
						
							|  |  |  |             return self._mock_wraps(*args, **kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         return self.return_value | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_awaited(self): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         Assert that the mock was awaited at least once. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if self.await_count == 0: | 
					
						
							|  |  |  |             msg = f"Expected {self._mock_name or 'mock'} to have been awaited." | 
					
						
							|  |  |  |             raise AssertionError(msg) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_awaited_once(self): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         Assert that the mock was awaited exactly once. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if not self.await_count == 1: | 
					
						
							|  |  |  |             msg = (f"Expected {self._mock_name or 'mock'} to have been awaited once." | 
					
						
							|  |  |  |                    f" Awaited {self.await_count} times.") | 
					
						
							|  |  |  |             raise AssertionError(msg) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_awaited_with(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         Assert that the last await was with the specified arguments. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if self.await_args is None: | 
					
						
							|  |  |  |             expected = self._format_mock_call_signature(args, kwargs) | 
					
						
							|  |  |  |             raise AssertionError(f'Expected await: {expected}\nNot awaited') | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         def _error_message(): | 
					
						
							| 
									
										
										
										
											2019-05-29 12:32:26 +05:30
										 |  |  |             msg = self._format_mock_failure_message(args, kwargs, action='await') | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |             return msg | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         expected = self._call_matcher(_Call((args, kwargs), two=True)) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         actual = self._call_matcher(self.await_args) | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         if actual != expected: | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |             cause = expected if isinstance(expected, Exception) else None | 
					
						
							|  |  |  |             raise AssertionError(_error_message()) from cause | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_awaited_once_with(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         Assert that the mock was awaited exactly once and with the specified | 
					
						
							|  |  |  |         arguments. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if not self.await_count == 1: | 
					
						
							|  |  |  |             msg = (f"Expected {self._mock_name or 'mock'} to have been awaited once." | 
					
						
							|  |  |  |                    f" Awaited {self.await_count} times.") | 
					
						
							|  |  |  |             raise AssertionError(msg) | 
					
						
							|  |  |  |         return self.assert_awaited_with(*args, **kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_any_await(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         Assert the mock has ever been awaited with the specified arguments. | 
					
						
							|  |  |  |         """
 | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         expected = self._call_matcher(_Call((args, kwargs), two=True)) | 
					
						
							|  |  |  |         cause = expected if isinstance(expected, Exception) else None | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         actual = [self._call_matcher(c) for c in self.await_args_list] | 
					
						
							| 
									
										
										
										
											2019-09-13 08:54:32 -07:00
										 |  |  |         if cause or expected not in _AnyComparer(actual): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |             expected_string = self._format_mock_call_signature(args, kwargs) | 
					
						
							|  |  |  |             raise AssertionError( | 
					
						
							|  |  |  |                 '%s await not found' % expected_string | 
					
						
							|  |  |  |             ) from cause | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_has_awaits(self, calls, any_order=False): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         Assert the mock has been awaited with the specified calls. | 
					
						
							|  |  |  |         The :attr:`await_args_list` list is checked for the awaits. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If `any_order` is False (the default) then the awaits must be | 
					
						
							|  |  |  |         sequential. There can be extra calls before or after the | 
					
						
							|  |  |  |         specified awaits. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         If `any_order` is True then the awaits can be in any order, but | 
					
						
							|  |  |  |         they must all appear in :attr:`await_args_list`. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         expected = [self._call_matcher(c) for c in calls] | 
					
						
							| 
									
										
										
										
											2019-09-24 15:08:31 -04:00
										 |  |  |         cause = next((e for e in expected if isinstance(e, Exception)), None) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         all_awaits = _CallList(self._call_matcher(c) for c in self.await_args_list) | 
					
						
							|  |  |  |         if not any_order: | 
					
						
							|  |  |  |             if expected not in all_awaits: | 
					
						
							| 
									
										
										
										
											2019-09-24 15:08:31 -04:00
										 |  |  |                 if cause is None: | 
					
						
							|  |  |  |                     problem = 'Awaits not found.' | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     problem = ('Error processing expected awaits.\n' | 
					
						
							|  |  |  |                                'Errors: {}').format( | 
					
						
							|  |  |  |                                    [e if isinstance(e, Exception) else None | 
					
						
							|  |  |  |                                     for e in expected]) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |                 raise AssertionError( | 
					
						
							| 
									
										
										
										
											2019-09-24 15:08:31 -04:00
										 |  |  |                     f'{problem}\n' | 
					
						
							|  |  |  |                     f'Expected: {_CallList(calls)}\n' | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |                     f'Actual: {self.await_args_list}' | 
					
						
							|  |  |  |                 ) from cause | 
					
						
							|  |  |  |             return | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         all_awaits = list(all_awaits) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         not_found = [] | 
					
						
							|  |  |  |         for kall in expected: | 
					
						
							|  |  |  |             try: | 
					
						
							|  |  |  |                 all_awaits.remove(kall) | 
					
						
							|  |  |  |             except ValueError: | 
					
						
							|  |  |  |                 not_found.append(kall) | 
					
						
							|  |  |  |         if not_found: | 
					
						
							|  |  |  |             raise AssertionError( | 
					
						
							|  |  |  |                 '%r not all found in await list' % (tuple(not_found),) | 
					
						
							|  |  |  |             ) from cause | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def assert_not_awaited(self): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         Assert that the mock was never awaited. | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         if self.await_count != 0: | 
					
						
							| 
									
										
										
										
											2019-05-27 18:26:23 +05:30
										 |  |  |             msg = (f"Expected {self._mock_name or 'mock'} to not have been awaited." | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |                    f" Awaited {self.await_count} times.") | 
					
						
							|  |  |  |             raise AssertionError(msg) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def reset_mock(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         """
 | 
					
						
							|  |  |  |         See :func:`.Mock.reset_mock()` | 
					
						
							|  |  |  |         """
 | 
					
						
							|  |  |  |         super().reset_mock(*args, **kwargs) | 
					
						
							|  |  |  |         self.await_count = 0 | 
					
						
							|  |  |  |         self.await_args = None | 
					
						
							|  |  |  |         self.await_args_list = _CallList() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class AsyncMock(AsyncMockMixin, AsyncMagicMixin, Mock): | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     Enhance :class:`Mock` with features allowing to mock | 
					
						
							|  |  |  |     an async function. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     The :class:`AsyncMock` object will behave so the object is | 
					
						
							|  |  |  |     recognized as an async function, and the result of a call is an awaitable: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     >>> mock = AsyncMock() | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |     >>> iscoroutinefunction(mock) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |     True | 
					
						
							|  |  |  |     >>> inspect.isawaitable(mock()) | 
					
						
							|  |  |  |     True | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     The result of ``mock()`` is an async function which will have the outcome | 
					
						
							|  |  |  |     of ``side_effect`` or ``return_value``: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     - if ``side_effect`` is a function, the async function will return the | 
					
						
							|  |  |  |       result of that function, | 
					
						
							|  |  |  |     - if ``side_effect`` is an exception, the async function will raise the | 
					
						
							|  |  |  |       exception, | 
					
						
							|  |  |  |     - if ``side_effect`` is an iterable, the async function will return the | 
					
						
							|  |  |  |       next value of the iterable, however, if the sequence of result is | 
					
						
							|  |  |  |       exhausted, ``StopIteration`` is raised immediately, | 
					
						
							|  |  |  |     - if ``side_effect`` is not defined, the async function will return the | 
					
						
							|  |  |  |       value defined by ``return_value``, hence, by default, the async function | 
					
						
							|  |  |  |       returns a new :class:`AsyncMock` object. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If the outcome of ``side_effect`` or ``return_value`` is an async function, | 
					
						
							|  |  |  |     the mock async function obtained when the mock object is called will be this | 
					
						
							|  |  |  |     async function itself (and not an async function returning an async | 
					
						
							|  |  |  |     function). | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     The test author can also specify a wrapped object with ``wraps``. In this | 
					
						
							|  |  |  |     case, the :class:`Mock` object behavior is the same as with an | 
					
						
							|  |  |  |     :class:`.Mock` object: the wrapped object may have methods | 
					
						
							|  |  |  |     defined as async function functions. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-21 14:17:17 +05:30
										 |  |  |     Based on Martin Richard's asynctest project. | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |     """
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | class _ANY(object): | 
					
						
							|  |  |  |     "A helper object that compares equal to everything." | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __eq__(self, other): | 
					
						
							|  |  |  |         return True | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __ne__(self, other): | 
					
						
							|  |  |  |         return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __repr__(self): | 
					
						
							|  |  |  |         return '<ANY>' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | ANY = _ANY() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _format_call_signature(name, args, kwargs): | 
					
						
							|  |  |  |     message = '%s(%%s)' % name | 
					
						
							|  |  |  |     formatted_args = '' | 
					
						
							|  |  |  |     args_string = ', '.join([repr(arg) for arg in args]) | 
					
						
							|  |  |  |     kwargs_string = ', '.join([ | 
					
						
							| 
									
										
										
										
											2019-09-09 16:25:22 +05:30
										 |  |  |         '%s=%r' % (key, value) for key, value in kwargs.items() | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     ]) | 
					
						
							|  |  |  |     if args_string: | 
					
						
							|  |  |  |         formatted_args = args_string | 
					
						
							|  |  |  |     if kwargs_string: | 
					
						
							|  |  |  |         if formatted_args: | 
					
						
							|  |  |  |             formatted_args += ', ' | 
					
						
							|  |  |  |         formatted_args += kwargs_string | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     return message % formatted_args | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class _Call(tuple): | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     A tuple for holding the results of a call to a mock, either in the form | 
					
						
							|  |  |  |     `(args, kwargs)` or `(name, args, kwargs)`. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If args or kwargs are empty then a call tuple will compare equal to | 
					
						
							|  |  |  |     a tuple without those values. This makes comparisons less verbose:: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         _Call(('name', (), {})) == ('name',) | 
					
						
							|  |  |  |         _Call(('name', (1,), {})) == ('name', (1,)) | 
					
						
							|  |  |  |         _Call(((), {'a': 'b'})) == ({'a': 'b'},) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     The `_Call` object provides a useful shortcut for comparing with call:: | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         _Call(((1, 2), {'a': 3})) == call(1, 2, a=3) | 
					
						
							|  |  |  |         _Call(('foo', (1, 2), {'a': 3})) == call.foo(1, 2, a=3) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If the _Call has no name then it will match any name. | 
					
						
							|  |  |  |     """
 | 
					
						
							| 
									
										
										
										
											2017-01-06 18:15:51 +01:00
										 |  |  |     def __new__(cls, value=(), name='', parent=None, two=False, | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 from_kall=True): | 
					
						
							|  |  |  |         args = () | 
					
						
							|  |  |  |         kwargs = {} | 
					
						
							|  |  |  |         _len = len(value) | 
					
						
							|  |  |  |         if _len == 3: | 
					
						
							|  |  |  |             name, args, kwargs = value | 
					
						
							|  |  |  |         elif _len == 2: | 
					
						
							|  |  |  |             first, second = value | 
					
						
							|  |  |  |             if isinstance(first, str): | 
					
						
							|  |  |  |                 name = first | 
					
						
							|  |  |  |                 if isinstance(second, tuple): | 
					
						
							|  |  |  |                     args = second | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     kwargs = second | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 args, kwargs = first, second | 
					
						
							|  |  |  |         elif _len == 1: | 
					
						
							|  |  |  |             value, = value | 
					
						
							|  |  |  |             if isinstance(value, str): | 
					
						
							|  |  |  |                 name = value | 
					
						
							|  |  |  |             elif isinstance(value, tuple): | 
					
						
							|  |  |  |                 args = value | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 kwargs = value | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if two: | 
					
						
							|  |  |  |             return tuple.__new__(cls, (args, kwargs)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         return tuple.__new__(cls, (name, args, kwargs)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, value=(), name=None, parent=None, two=False, | 
					
						
							|  |  |  |                  from_kall=True): | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |         self._mock_name = name | 
					
						
							|  |  |  |         self._mock_parent = parent | 
					
						
							|  |  |  |         self._mock_from_kall = from_kall | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __eq__(self, other): | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             len_other = len(other) | 
					
						
							|  |  |  |         except TypeError: | 
					
						
							| 
									
										
										
										
											2019-08-08 08:42:54 +03:00
										 |  |  |             return NotImplemented | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         self_name = '' | 
					
						
							|  |  |  |         if len(self) == 2: | 
					
						
							|  |  |  |             self_args, self_kwargs = self | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             self_name, self_args, self_kwargs = self | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |         if (getattr(self, '_mock_parent', None) and getattr(other, '_mock_parent', None) | 
					
						
							|  |  |  |                 and self._mock_parent != other._mock_parent): | 
					
						
							| 
									
										
										
										
											2018-12-03 21:31:37 +00:00
										 |  |  |             return False | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         other_name = '' | 
					
						
							|  |  |  |         if len_other == 0: | 
					
						
							|  |  |  |             other_args, other_kwargs = (), {} | 
					
						
							|  |  |  |         elif len_other == 3: | 
					
						
							|  |  |  |             other_name, other_args, other_kwargs = other | 
					
						
							|  |  |  |         elif len_other == 1: | 
					
						
							|  |  |  |             value, = other | 
					
						
							|  |  |  |             if isinstance(value, tuple): | 
					
						
							|  |  |  |                 other_args = value | 
					
						
							|  |  |  |                 other_kwargs = {} | 
					
						
							|  |  |  |             elif isinstance(value, str): | 
					
						
							|  |  |  |                 other_name = value | 
					
						
							|  |  |  |                 other_args, other_kwargs = (), {} | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 other_args = () | 
					
						
							|  |  |  |                 other_kwargs = value | 
					
						
							| 
									
										
										
										
											2015-09-09 23:35:25 +03:00
										 |  |  |         elif len_other == 2: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             # could be (name, args) or (name, kwargs) or (args, kwargs) | 
					
						
							|  |  |  |             first, second = other | 
					
						
							|  |  |  |             if isinstance(first, str): | 
					
						
							|  |  |  |                 other_name = first | 
					
						
							|  |  |  |                 if isinstance(second, tuple): | 
					
						
							|  |  |  |                     other_args, other_kwargs = second, {} | 
					
						
							|  |  |  |                 else: | 
					
						
							|  |  |  |                     other_args, other_kwargs = (), second | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 other_args, other_kwargs = first, second | 
					
						
							| 
									
										
										
										
											2015-09-09 23:35:25 +03:00
										 |  |  |         else: | 
					
						
							|  |  |  |             return False | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         if self_name and other_name != self_name: | 
					
						
							|  |  |  |             return False | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # this order is important for ANY to work! | 
					
						
							|  |  |  |         return (other_args, other_kwargs) == (self_args, self_kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2016-03-28 00:30:02 +03:00
										 |  |  |     __ne__ = object.__ne__ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def __call__(self, /, *args, **kwargs): | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |         if self._mock_name is None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             return _Call(('', args, kwargs), name='()') | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |         name = self._mock_name + '()' | 
					
						
							|  |  |  |         return _Call((self._mock_name, args, kwargs), name=name, parent=self) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __getattr__(self, attr): | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |         if self._mock_name is None: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             return _Call(name=attr, from_kall=False) | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |         name = '%s.%s' % (self._mock_name, attr) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return _Call(name=name, parent=self, from_kall=False) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-09-11 07:28:06 -07:00
										 |  |  |     def __getattribute__(self, attr): | 
					
						
							|  |  |  |         if attr in tuple.__dict__: | 
					
						
							|  |  |  |             raise AttributeError | 
					
						
							|  |  |  |         return tuple.__getattribute__(self, attr) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-03-22 13:40:40 +05:30
										 |  |  |     def _get_call_arguments(self): | 
					
						
							|  |  |  |         if len(self) == 2: | 
					
						
							|  |  |  |             args, kwargs = self | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             name, args, kwargs = self | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         return args, kwargs | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     @property | 
					
						
							|  |  |  |     def args(self): | 
					
						
							|  |  |  |         return self._get_call_arguments()[0] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     @property | 
					
						
							|  |  |  |     def kwargs(self): | 
					
						
							|  |  |  |         return self._get_call_arguments()[1] | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     def __repr__(self): | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |         if not self._mock_from_kall: | 
					
						
							|  |  |  |             name = self._mock_name or 'call' | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             if name.startswith('()'): | 
					
						
							|  |  |  |                 name = 'call%s' % name | 
					
						
							|  |  |  |             return name | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if len(self) == 2: | 
					
						
							|  |  |  |             name = 'call' | 
					
						
							|  |  |  |             args, kwargs = self | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             name, args, kwargs = self | 
					
						
							|  |  |  |             if not name: | 
					
						
							|  |  |  |                 name = 'call' | 
					
						
							|  |  |  |             elif not name.startswith('()'): | 
					
						
							|  |  |  |                 name = 'call.%s' % name | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 name = 'call%s' % name | 
					
						
							|  |  |  |         return _format_call_signature(name, args, kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def call_list(self): | 
					
						
							|  |  |  |         """For a call object that represents multiple calls, `call_list`
 | 
					
						
							|  |  |  |         returns a list of all the intermediate calls as well as the | 
					
						
							|  |  |  |         final call."""
 | 
					
						
							|  |  |  |         vals = [] | 
					
						
							|  |  |  |         thing = self | 
					
						
							|  |  |  |         while thing is not None: | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |             if thing._mock_from_kall: | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |                 vals.append(thing) | 
					
						
							| 
									
										
										
										
											2018-12-04 11:08:45 +02:00
										 |  |  |             thing = thing._mock_parent | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return _CallList(reversed(vals)) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | call = _Call(from_kall=False) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def create_autospec(spec, spec_set=False, instance=False, _parent=None, | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |                     _name=None, *, unsafe=False, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     """Create a mock object using another object as a spec. Attributes on the
 | 
					
						
							|  |  |  |     mock will use the corresponding attribute on the `spec` object as their | 
					
						
							|  |  |  |     spec. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Functions or methods being mocked will have their arguments checked | 
					
						
							|  |  |  |     to check that they are called with the correct signature. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If `spec_set` is True then attempting to set attributes that don't exist | 
					
						
							|  |  |  |     on the spec object will raise an `AttributeError`. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     If a class is used as a spec then the return value of the mock (the | 
					
						
							|  |  |  |     instance of the class) will have the same spec. You can use a class as the | 
					
						
							|  |  |  |     spec for an instance object by passing `instance=True`. The returned mock | 
					
						
							|  |  |  |     will only be callable if instances of the mock are callable. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |     `create_autospec` will raise a `RuntimeError` if passed some common | 
					
						
							|  |  |  |     misspellings of the arguments autospec and spec_set. Pass the argument | 
					
						
							|  |  |  |     `unsafe` with the value True to disable that check. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     `create_autospec` also takes arbitrary keyword arguments that are passed to | 
					
						
							|  |  |  |     the constructor of the created mock."""
 | 
					
						
							|  |  |  |     if _is_list(spec): | 
					
						
							|  |  |  |         # can't pass a list instance to the mock constructor as it will be | 
					
						
							|  |  |  |         # interpreted as a list of strings | 
					
						
							|  |  |  |         spec = type(spec) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     is_type = isinstance(spec, type) | 
					
						
							| 
									
										
										
										
											2021-04-09 23:45:50 -04:00
										 |  |  |     if _is_instance_mock(spec): | 
					
						
							|  |  |  |         raise InvalidSpecError(f'Cannot autospec a Mock object. ' | 
					
						
							|  |  |  |                                f'[object={spec!r}]') | 
					
						
							| 
									
										
										
										
											2019-05-27 18:26:23 +05:30
										 |  |  |     is_async_func = _is_async_func(spec) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     _kwargs = {'spec': spec} | 
					
						
							|  |  |  |     if spec_set: | 
					
						
							|  |  |  |         _kwargs = {'spec_set': spec} | 
					
						
							|  |  |  |     elif spec is None: | 
					
						
							|  |  |  |         # None we mock with a normal mock without a spec | 
					
						
							|  |  |  |         _kwargs = {} | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     if _kwargs and instance: | 
					
						
							|  |  |  |         _kwargs['_spec_as_instance'] = True | 
					
						
							| 
									
										
										
										
											2020-12-14 19:30:09 +01:00
										 |  |  |     if not unsafe: | 
					
						
							|  |  |  |         _check_spec_arg_typos(kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |     _kwargs.update(kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Klass = MagicMock | 
					
						
							| 
									
										
										
										
											2016-08-15 23:23:40 -07:00
										 |  |  |     if inspect.isdatadescriptor(spec): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         # descriptors don't have a spec | 
					
						
							|  |  |  |         # because we don't know what type they return | 
					
						
							|  |  |  |         _kwargs = {} | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |     elif is_async_func: | 
					
						
							|  |  |  |         if instance: | 
					
						
							|  |  |  |             raise RuntimeError("Instance can not be True when create_autospec " | 
					
						
							|  |  |  |                                "is mocking an async function") | 
					
						
							|  |  |  |         Klass = AsyncMock | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     elif not _callable(spec): | 
					
						
							|  |  |  |         Klass = NonCallableMagicMock | 
					
						
							|  |  |  |     elif is_type and instance and not _instance_callable(spec): | 
					
						
							|  |  |  |         Klass = NonCallableMagicMock | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2014-04-16 01:05:50 +05:30
										 |  |  |     _name = _kwargs.pop('name', _name) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     _new_name = _name | 
					
						
							|  |  |  |     if _parent is None: | 
					
						
							|  |  |  |         # for a top level object no _new_name should be set | 
					
						
							|  |  |  |         _new_name = '' | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     mock = Klass(parent=_parent, _new_parent=_parent, _new_name=_new_name, | 
					
						
							|  |  |  |                  name=_name, **_kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if isinstance(spec, FunctionTypes): | 
					
						
							|  |  |  |         # should only happen at the top level because we don't | 
					
						
							|  |  |  |         # recurse for functions | 
					
						
							|  |  |  |         mock = _set_signature(mock, spec) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |         if is_async_func: | 
					
						
							| 
									
										
										
										
											2019-05-27 18:26:23 +05:30
										 |  |  |             _setup_async_mock(mock) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     else: | 
					
						
							|  |  |  |         _check_signature(spec, mock, is_type, instance) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if _parent is not None and not instance: | 
					
						
							|  |  |  |         _parent._mock_children[_name] = mock | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     if is_type and not instance and 'return_value' not in kwargs: | 
					
						
							|  |  |  |         mock.return_value = create_autospec(spec, spec_set, instance=True, | 
					
						
							|  |  |  |                                             _name='()', _parent=mock) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for entry in dir(spec): | 
					
						
							|  |  |  |         if _is_magic(entry): | 
					
						
							|  |  |  |             # MagicMock already does the useful magic methods for us | 
					
						
							|  |  |  |             continue | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # XXXX do we need a better way of getting attributes without | 
					
						
							|  |  |  |         # triggering code execution (?) Probably not - we need the actual | 
					
						
							|  |  |  |         # object to mock it so we would rather trigger a property than mock | 
					
						
							|  |  |  |         # the property descriptor. Likewise we want to mock out dynamically | 
					
						
							|  |  |  |         # provided attributes. | 
					
						
							| 
									
										
										
										
											2012-04-13 17:39:16 +01:00
										 |  |  |         # XXXX what about attributes that raise exceptions other than | 
					
						
							|  |  |  |         # AttributeError on being fetched? | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         # we could be resilient against it, or catch and propagate the | 
					
						
							|  |  |  |         # exception when the attribute is fetched from the mock | 
					
						
							| 
									
										
										
										
											2012-04-13 17:39:16 +01:00
										 |  |  |         try: | 
					
						
							|  |  |  |             original = getattr(spec, entry) | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							|  |  |  |             continue | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  |         kwargs = {'spec': original} | 
					
						
							|  |  |  |         if spec_set: | 
					
						
							|  |  |  |             kwargs = {'spec_set': original} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         if not isinstance(original, FunctionTypes): | 
					
						
							|  |  |  |             new = _SpecState(original, spec_set, mock, entry, instance) | 
					
						
							|  |  |  |             mock._mock_children[entry] = new | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             parent = mock | 
					
						
							|  |  |  |             if isinstance(spec, FunctionTypes): | 
					
						
							|  |  |  |                 parent = mock.mock | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             skipfirst = _must_skip(spec, entry, is_type) | 
					
						
							|  |  |  |             kwargs['_eat_self'] = skipfirst | 
					
						
							| 
									
										
										
										
											2020-01-27 14:11:19 +00:00
										 |  |  |             if iscoroutinefunction(original): | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  |                 child_klass = AsyncMock | 
					
						
							|  |  |  |             else: | 
					
						
							|  |  |  |                 child_klass = MagicMock | 
					
						
							|  |  |  |             new = child_klass(parent=parent, name=entry, _new_name=entry, | 
					
						
							|  |  |  |                               _new_parent=parent, | 
					
						
							|  |  |  |                               **kwargs) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             mock._mock_children[entry] = new | 
					
						
							| 
									
										
										
										
											2022-11-07 02:24:46 -05:00
										 |  |  |             new.return_value = child_klass() | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |             _check_signature(original, new, skipfirst=skipfirst) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |         # so functions created with _set_signature become instance attributes, | 
					
						
							|  |  |  |         # *plus* their underlying mock exists in _mock_children of the parent | 
					
						
							|  |  |  |         # mock. Adding to _mock_children may be unnecessary where we are also | 
					
						
							|  |  |  |         # setting as an instance attribute? | 
					
						
							|  |  |  |         if isinstance(new, FunctionTypes): | 
					
						
							|  |  |  |             setattr(mock, entry, new) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     return mock | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def _must_skip(spec, entry, is_type): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |     """
 | 
					
						
							|  |  |  |     Return whether we should skip the first argument on spec's `entry` | 
					
						
							|  |  |  |     attribute. | 
					
						
							|  |  |  |     """
 | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     if not isinstance(spec, type): | 
					
						
							|  |  |  |         if entry in getattr(spec, '__dict__', {}): | 
					
						
							|  |  |  |             # instance attribute - shouldn't skip | 
					
						
							|  |  |  |             return False | 
					
						
							|  |  |  |         spec = spec.__class__ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     for klass in spec.__mro__: | 
					
						
							|  |  |  |         result = klass.__dict__.get(entry, DEFAULT) | 
					
						
							|  |  |  |         if result is DEFAULT: | 
					
						
							|  |  |  |             continue | 
					
						
							|  |  |  |         if isinstance(result, (staticmethod, classmethod)): | 
					
						
							|  |  |  |             return False | 
					
						
							| 
									
										
										
										
											2020-01-29 16:43:37 +01:00
										 |  |  |         elif isinstance(result, FunctionTypes): | 
					
						
							| 
									
										
										
										
											2013-02-03 00:23:58 +01:00
										 |  |  |             # Normal method => skip if looked up on type | 
					
						
							|  |  |  |             # (if looked up on instance, self is already skipped) | 
					
						
							|  |  |  |             return is_type | 
					
						
							|  |  |  |         else: | 
					
						
							|  |  |  |             return False | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-01 23:04:04 +01:00
										 |  |  |     # function is a dynamically provided attribute | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     return is_type | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class _SpecState(object): | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     def __init__(self, spec, spec_set=False, parent=None, | 
					
						
							|  |  |  |                  name=None, ids=None, instance=False): | 
					
						
							|  |  |  |         self.spec = spec | 
					
						
							|  |  |  |         self.ids = ids | 
					
						
							|  |  |  |         self.spec_set = spec_set | 
					
						
							|  |  |  |         self.parent = parent | 
					
						
							|  |  |  |         self.instance = instance | 
					
						
							|  |  |  |         self.name = name | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | FunctionTypes = ( | 
					
						
							|  |  |  |     # python function | 
					
						
							|  |  |  |     type(create_autospec), | 
					
						
							|  |  |  |     # instance method | 
					
						
							|  |  |  |     type(ANY.__eq__), | 
					
						
							|  |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 19:03:13 +01:00
										 |  |  | file_spec = None | 
					
						
							| 
									
										
										
										
											2022-02-03 03:41:19 -05:00
										 |  |  | open_spec = None | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2013-03-19 17:22:51 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  | def _to_stream(read_data): | 
					
						
							|  |  |  |     if isinstance(read_data, bytes): | 
					
						
							|  |  |  |         return io.BytesIO(read_data) | 
					
						
							|  |  |  |     else: | 
					
						
							|  |  |  |         return io.StringIO(read_data) | 
					
						
							| 
									
										
										
										
											2012-03-25 19:11:50 +01:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-07-17 20:08:45 +12:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 19:11:50 +01:00
										 |  |  | def mock_open(mock=None, read_data=''): | 
					
						
							| 
									
										
										
										
											2012-03-25 19:07:33 +01:00
										 |  |  |     """
 | 
					
						
							|  |  |  |     A helper function to create a mock to replace the use of `open`. It works | 
					
						
							|  |  |  |     for `open` called directly or used as a context manager. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     The `mock` argument is the mock object to configure. If `None` (the | 
					
						
							|  |  |  |     default) then a `MagicMock` will be created for you, with the API limited | 
					
						
							|  |  |  |     to methods or attributes available on standard file handles. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-12-20 21:30:21 +05:30
										 |  |  |     `read_data` is a string for the `read`, `readline` and `readlines` of the | 
					
						
							| 
									
										
										
										
											2013-03-19 17:22:51 -07:00
										 |  |  |     file handle to return.  This is an empty string by default. | 
					
						
							| 
									
										
										
										
											2012-03-25 19:07:33 +01:00
										 |  |  |     """
 | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  |     _read_data = _to_stream(read_data) | 
					
						
							|  |  |  |     _state = [_read_data, None] | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-07-24 03:48:20 +12:00
										 |  |  |     def _readlines_side_effect(*args, **kwargs): | 
					
						
							|  |  |  |         if handle.readlines.return_value is not None: | 
					
						
							|  |  |  |             return handle.readlines.return_value | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  |         return _state[0].readlines(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2015-07-24 03:48:20 +12:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def _read_side_effect(*args, **kwargs): | 
					
						
							|  |  |  |         if handle.read.return_value is not None: | 
					
						
							|  |  |  |             return handle.read.return_value | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  |         return _state[0].read(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2015-07-24 03:48:20 +12:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  |     def _readline_side_effect(*args, **kwargs): | 
					
						
							| 
									
										
										
										
											2018-09-12 23:21:16 +01:00
										 |  |  |         yield from _iter_side_effect() | 
					
						
							|  |  |  |         while True: | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  |             yield _state[0].readline(*args, **kwargs) | 
					
						
							| 
									
										
										
										
											2018-09-12 23:21:16 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def _iter_side_effect(): | 
					
						
							| 
									
										
										
										
											2015-07-24 03:48:20 +12:00
										 |  |  |         if handle.readline.return_value is not None: | 
					
						
							|  |  |  |             while True: | 
					
						
							|  |  |  |                 yield handle.readline.return_value | 
					
						
							|  |  |  |         for line in _state[0]: | 
					
						
							|  |  |  |             yield line | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-05-23 12:03:25 +02:00
										 |  |  |     def _next_side_effect(): | 
					
						
							|  |  |  |         if handle.readline.return_value is not None: | 
					
						
							|  |  |  |             return handle.readline.return_value | 
					
						
							|  |  |  |         return next(_state[0]) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-25 19:03:13 +01:00
										 |  |  |     global file_spec | 
					
						
							|  |  |  |     if file_spec is None: | 
					
						
							|  |  |  |         import _io | 
					
						
							|  |  |  |         file_spec = list(set(dir(_io.TextIOWrapper)).union(set(dir(_io.BytesIO)))) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-02-03 03:41:19 -05:00
										 |  |  |     global open_spec | 
					
						
							|  |  |  |     if open_spec is None: | 
					
						
							|  |  |  |         import _io | 
					
						
							|  |  |  |         open_spec = list(set(dir(_io.open))) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     if mock is None: | 
					
						
							| 
									
										
										
										
											2022-02-03 03:41:19 -05:00
										 |  |  |         mock = MagicMock(name='open', spec=open_spec) | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-07-24 03:48:20 +12:00
										 |  |  |     handle = MagicMock(spec=file_spec) | 
					
						
							|  |  |  |     handle.__enter__.return_value = handle | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     handle.write.return_value = None | 
					
						
							|  |  |  |     handle.read.return_value = None | 
					
						
							|  |  |  |     handle.readline.return_value = None | 
					
						
							|  |  |  |     handle.readlines.return_value = None | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     handle.read.side_effect = _read_side_effect | 
					
						
							|  |  |  |     _state[1] = _readline_side_effect() | 
					
						
							|  |  |  |     handle.readline.side_effect = _state[1] | 
					
						
							|  |  |  |     handle.readlines.side_effect = _readlines_side_effect | 
					
						
							| 
									
										
										
										
											2018-09-12 23:21:16 +01:00
										 |  |  |     handle.__iter__.side_effect = _iter_side_effect | 
					
						
							| 
									
										
										
										
											2019-05-23 12:03:25 +02:00
										 |  |  |     handle.__next__.side_effect = _next_side_effect | 
					
						
							| 
									
										
										
										
											2015-07-24 03:48:20 +12:00
										 |  |  | 
 | 
					
						
							|  |  |  |     def reset_data(*args, **kwargs): | 
					
						
							| 
									
										
										
										
											2019-05-07 12:48:36 +02:00
										 |  |  |         _state[0] = _to_stream(read_data) | 
					
						
							| 
									
										
										
										
											2015-07-24 03:48:20 +12:00
										 |  |  |         if handle.readline.side_effect == _state[1]: | 
					
						
							|  |  |  |             # Only reset the side effect if the user hasn't overridden it. | 
					
						
							|  |  |  |             _state[1] = _readline_side_effect() | 
					
						
							|  |  |  |             handle.readline.side_effect = _state[1] | 
					
						
							|  |  |  |         return DEFAULT | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     mock.side_effect = reset_data | 
					
						
							|  |  |  |     mock.return_value = handle | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |     return mock | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class PropertyMock(Mock): | 
					
						
							| 
									
										
										
										
											2012-03-25 19:07:33 +01:00
										 |  |  |     """
 | 
					
						
							|  |  |  |     A mock intended to be used as a property, or other descriptor, on a class. | 
					
						
							|  |  |  |     `PropertyMock` provides `__get__` and `__set__` methods so you can specify | 
					
						
							|  |  |  |     a return value when it is fetched. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     Fetching a `PropertyMock` instance from an object calls the mock, with | 
					
						
							|  |  |  |     no args. Setting it calls the mock with the value being set. | 
					
						
							|  |  |  |     """
 | 
					
						
							| 
									
										
										
										
											2019-06-01 11:00:15 +03:00
										 |  |  |     def _get_child_mock(self, /, **kwargs): | 
					
						
							| 
									
										
										
										
											2012-04-13 16:57:22 +01:00
										 |  |  |         return MagicMock(**kwargs) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-08-29 01:27:42 -07:00
										 |  |  |     def __get__(self, obj, obj_type=None): | 
					
						
							| 
									
										
										
										
											2012-03-14 12:24:34 -07:00
										 |  |  |         return self() | 
					
						
							|  |  |  |     def __set__(self, obj, val): | 
					
						
							|  |  |  |         self(val) | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | def seal(mock): | 
					
						
							| 
									
										
										
										
											2018-10-19 22:57:37 +01:00
										 |  |  |     """Disable the automatic generation of child mocks.
 | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  | 
 | 
					
						
							|  |  |  |     Given an input Mock, seals it to ensure no further mocks will be generated | 
					
						
							|  |  |  |     when accessing an attribute that was not already defined. | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2018-10-19 22:57:37 +01:00
										 |  |  |     The operation recursively seals the mock passed in, meaning that | 
					
						
							|  |  |  |     the mock itself, any mocks generated by accessing one of its attributes, | 
					
						
							|  |  |  |     and all assigned mocks without a name or spec will be sealed. | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  |     """
 | 
					
						
							|  |  |  |     mock._mock_sealed = True | 
					
						
							|  |  |  |     for attr in dir(mock): | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             m = getattr(mock, attr) | 
					
						
							|  |  |  |         except AttributeError: | 
					
						
							|  |  |  |             continue | 
					
						
							|  |  |  |         if not isinstance(m, NonCallableMock): | 
					
						
							|  |  |  |             continue | 
					
						
							| 
									
										
										
										
											2021-09-14 13:20:40 +03:00
										 |  |  |         if isinstance(m._mock_children.get(attr), _SpecState): | 
					
						
							|  |  |  |             continue | 
					
						
							| 
									
										
										
										
											2017-10-17 12:35:11 +01:00
										 |  |  |         if m._mock_new_parent is mock: | 
					
						
							|  |  |  |             seal(m) | 
					
						
							| 
									
										
										
										
											2019-05-20 09:19:53 -07:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | class _AsyncIterator: | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     Wraps an iterator in an asynchronous iterator. | 
					
						
							|  |  |  |     """
 | 
					
						
							|  |  |  |     def __init__(self, iterator): | 
					
						
							|  |  |  |         self.iterator = iterator | 
					
						
							|  |  |  |         code_mock = NonCallableMock(spec_set=CodeType) | 
					
						
							|  |  |  |         code_mock.co_flags = inspect.CO_ITERABLE_COROUTINE | 
					
						
							|  |  |  |         self.__dict__['__code__'] = code_mock | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  |     async def __anext__(self): | 
					
						
							|  |  |  |         try: | 
					
						
							|  |  |  |             return next(self.iterator) | 
					
						
							|  |  |  |         except StopIteration: | 
					
						
							|  |  |  |             pass | 
					
						
							|  |  |  |         raise StopAsyncIteration |