diff options
author | Xtreak <tir.karthi@gmail.com> | 2019-09-09 14:34:57 +0530 |
---|---|---|
committer | Lisa Roach <lisaroach14@gmail.com> | 2019-09-09 10:04:57 +0100 |
commit | aa515082749687c1e3bc9ec5e2296368191b9f84 (patch) | |
tree | 368050a77c01ec3c2d3337ad9529fbe3c1cb2cd5 | |
parent | 526a01467b3277f9fcf7f91e66c23321caa1245d (diff) | |
download | cpython-git-aa515082749687c1e3bc9ec5e2296368191b9f84.tar.gz |
Fix assertions regarding magic methods function body that was not executed (GH-14154)
-rw-r--r-- | Lib/unittest/test/testmock/testasync.py | 11 |
1 files changed, 2 insertions, 9 deletions
diff --git a/Lib/unittest/test/testmock/testasync.py b/Lib/unittest/test/testmock/testasync.py index fa906e4f71..865cfdc002 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) |