diff --git a/Lib/test/test_unittest/testmock/testmagicmethods.py b/Lib/test/test_unittest/testmock/testmagicmethods.py index 108d2fdd86ea48..2a8aa11b3284f6 100644 --- a/Lib/test/test_unittest/testmock/testmagicmethods.py +++ b/Lib/test/test_unittest/testmock/testmagicmethods.py @@ -363,6 +363,14 @@ def test_magic_mock_resets_manual_mocks(self): self.assertEqual(list(iter(mm)), []) self.assertIsInstance(mm.custom(), MagicMock) + def test_magic_mock_resets_manual_mocks_empty_iter(self): + mm = MagicMock() + mm.__iter__.return_value = [] + self.assertEqual(list(iter(mm)), []) + + mm.reset_mock(return_value=True) + self.assertEqual(list(iter(mm)), []) + def test_magic_methods_and_spec(self): class Iterable(object): def __iter__(self): pass