diff options
author | Miss Islington (bot) <31488909+miss-islington@users.noreply.github.com> | 2018-12-08 11:41:52 (GMT) |
---|---|---|
committer | Chris Withers <chris@withers.org> | 2018-12-08 11:41:52 (GMT) |
commit | 12b9fb603eea9298c835bae5b8742db4fa52892e (patch) | |
tree | 16f266f891f92fcf85a284807bc586fb42dd16cd /Lib/unittest/mock.py | |
parent | 2d6bc25dbc3dc5662f13917eb759f92842bf6de6 (diff) | |
download | cpython-12b9fb603eea9298c835bae5b8742db4fa52892e.zip cpython-12b9fb603eea9298c835bae5b8742db4fa52892e.tar.gz cpython-12b9fb603eea9298c835bae5b8742db4fa52892e.tar.bz2 |
bpo-35330: Don't call the wrapped object if `side_effect` is set (GH11034)
* tests: Further validate `wraps` functionality in `unittest.mock.Mock`
Add more tests to validate how `wraps` interacts with other features of
mocks.
* Don't call the wrapped object if `side_effect` is set
When a object is wrapped using `Mock(wraps=...)`, if an user sets a
`side_effect` in one of their methods, return the value of `side_effect`
and don't call the original object.
* Refactor what to be called on `mock_call`
When a `Mock` is called, it should return looking up in the following
order: `side_effect`, `return_value`, `wraps`. If any of the first two
return `mock.DEFAULT`, lookup in the next option.
It makes no sense to check for `wraps` returning default, as it is
supposed to be the original implementation and there is nothing to
fallback to.
(cherry picked from commit f05df0a4b679d0acfd0b1fe6187ba2d553b37afa)
Co-authored-by: Mario Corchero <mariocj89@gmail.com>
Diffstat (limited to 'Lib/unittest/mock.py')
-rw-r--r-- | Lib/unittest/mock.py | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/Lib/unittest/mock.py b/Lib/unittest/mock.py index 707ef0b..645d100 100644 --- a/Lib/unittest/mock.py +++ b/Lib/unittest/mock.py @@ -993,28 +993,27 @@ class CallableMixin(Base): break seen.add(_new_parent_id) - ret_val = DEFAULT effect = self.side_effect if effect is not None: if _is_exception(effect): raise effect - - if not _callable(effect): + elif not _callable(effect): result = next(effect) if _is_exception(result): raise result - if result is DEFAULT: - result = self.return_value + else: + result = effect(*args, **kwargs) + + if result is not DEFAULT: return result - ret_val = effect(*args, **kwargs) + if self._mock_return_value is not DEFAULT: + return self.return_value - if (self._mock_wraps is not None and - self._mock_return_value is DEFAULT): + if self._mock_wraps is not None: return self._mock_wraps(*args, **kwargs) - if ret_val is DEFAULT: - ret_val = self.return_value - return ret_val + + return self.return_value |