Skip to content

Commit

Permalink
pythongh-120732: Fix name passing to Mock, when using kwargs to `…
Browse files Browse the repository at this point in the history
…create_autospec` (pythonGH-120737)

(cherry picked from commit 1e48156)

Co-authored-by: Nikita Sobolev <[email protected]>
  • Loading branch information
sobolevn authored and miss-islington committed Jun 19, 2024
1 parent d6a36c9 commit fa482f3
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 7 deletions.
5 changes: 5 additions & 0 deletions Lib/test/test_unittest/testmock/testmock.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,6 +118,11 @@ def test_create_autospec_should_be_configurable_by_kwargs(self):
# pass kwargs with respect to the parent mock.
self.assertEqual(class_mock().return_value.meth.side_effect, None)

def test_create_autospec_correctly_handles_name(self):
class X: ...
mock = create_autospec(X, spec_set=True, name="Y")
self.assertEqual(mock._mock_name, "Y")

def test_repr(self):
mock = Mock(name='foo')
self.assertIn('foo', repr(mock))
Expand Down
13 changes: 6 additions & 7 deletions Lib/unittest/mock.py
Original file line number Diff line number Diff line change
Expand Up @@ -2725,6 +2725,12 @@ def create_autospec(spec, spec_set=False, instance=False, _parent=None,
if not unsafe:
_check_spec_arg_typos(kwargs)

_name = kwargs.pop('name', _name)
_new_name = _name
if _parent is None:
# for a top level object no _new_name should be set
_new_name = ''

_kwargs.update(kwargs)

Klass = MagicMock
Expand All @@ -2742,13 +2748,6 @@ def create_autospec(spec, spec_set=False, instance=False, _parent=None,
elif is_type and instance and not _instance_callable(spec):
Klass = NonCallableMagicMock

_name = _kwargs.pop('name', _name)

_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)

Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
Fix ``name`` passing to :class:`unittest.mock.Mock` object when using
:func:`unittest.mock.create_autospec`.

0 comments on commit fa482f3

Please sign in to comment.