Fix assertions regarding magic methods function body that was not executed (GH-14154)

This commit is contained in:
Xtreak 2019-09-09 14:34:57 +05:30 committed by Lisa Roach
parent 526a01467b
commit aa51508274
1 changed files with 2 additions and 9 deletions

View File

@ -361,17 +361,14 @@ class AsyncArguments(unittest.TestCase):
class AsyncContextManagerTest(unittest.TestCase): class AsyncContextManagerTest(unittest.TestCase):
class WithAsyncContextManager: class WithAsyncContextManager:
def __init__(self):
self.entered = False
self.exited = False
async def __aenter__(self, *args, **kwargs): async def __aenter__(self, *args, **kwargs):
self.entered = True
return self return self
async def __aexit__(self, *args, **kwargs): async def __aexit__(self, *args, **kwargs):
self.exited = True pass
def test_magic_methods_are_async_mocks(self): def test_magic_methods_are_async_mocks(self):
mock = MagicMock(self.WithAsyncContextManager()) mock = MagicMock(self.WithAsyncContextManager())
@ -390,11 +387,7 @@ class AsyncContextManagerTest(unittest.TestCase):
return result return result
result = asyncio.run(use_context_manager()) result = asyncio.run(use_context_manager())
self.assertFalse(instance.entered)
self.assertFalse(instance.exited)
self.assertTrue(called) self.assertTrue(called)
self.assertTrue(mock_instance.entered)
self.assertTrue(mock_instance.exited)
self.assertTrue(mock_instance.__aenter__.called) self.assertTrue(mock_instance.__aenter__.called)
self.assertTrue(mock_instance.__aexit__.called) self.assertTrue(mock_instance.__aexit__.called)
self.assertIsNot(mock_instance, result) self.assertIsNot(mock_instance, result)