Commit aa515082 authored by Xtreak's avatar Xtreak Committed by Lisa Roach

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

parent 526a0146
...@@ -361,17 +361,14 @@ class AsyncArguments(unittest.TestCase): ...@@ -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): ...@@ -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)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment