From aa515082749687c1e3bc9ec5e2296368191b9f84 Mon Sep 17 00:00:00 2001 From: Xtreak Date: Mon, 9 Sep 2019 14:34:57 +0530 Subject: [PATCH] Fix assertions regarding magic methods function body that was not executed (GH-14154) --- Lib/unittest/test/testmock/testasync.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/Lib/unittest/test/testmock/testasync.py b/Lib/unittest/test/testmock/testasync.py index fa906e4f715..865cfdc0026 100644 --- a/Lib/unittest/test/testmock/testasync.py +++ b/Lib/unittest/test/testmock/testasync.py @@ -361,17 +361,14 @@ class AsyncArguments(unittest.TestCase): class AsyncContextManagerTest(unittest.TestCase): + class WithAsyncContextManager: - def __init__(self): - self.entered = False - self.exited = False async def __aenter__(self, *args, **kwargs): - self.entered = True return self async def __aexit__(self, *args, **kwargs): - self.exited = True + pass def test_magic_methods_are_async_mocks(self): mock = MagicMock(self.WithAsyncContextManager()) @@ -390,11 +387,7 @@ class AsyncContextManagerTest(unittest.TestCase): return result result = asyncio.run(use_context_manager()) - self.assertFalse(instance.entered) - self.assertFalse(instance.exited) self.assertTrue(called) - self.assertTrue(mock_instance.entered) - self.assertTrue(mock_instance.exited) self.assertTrue(mock_instance.__aenter__.called) self.assertTrue(mock_instance.__aexit__.called) self.assertIsNot(mock_instance, result)