PYTHON-5071 Fix PYTHON-5011 pypy workaround

Author: ShaneHarvey

test/__init__.py

@@ -941,9 +941,17 @@ def fail_point(self, command_args):
         try:
             yield
         finally:
-            client_context.client.admin.command(
-                "configureFailPoint", cmd_on["configureFailPoint"], mode="off"
-            )
+            try:
+                client_context.client.admin.command(
+                    "configureFailPoint", cmd_on["configureFailPoint"], mode="off"
+                )
+            except pymongo.errors.ConnectionFailure:
+                # Workaround PyPy bug described in PYTHON-5011.
+                if not _IS_SYNC and "PyPy" in sys.version:
+                    client_context.client.admin.command(
+                        "configureFailPoint", cmd_on["configureFailPoint"], mode="off"
+                    )
+                raise
 
     @contextmanager
     def fork(

test/asynchronous/__init__.py

@@ -943,9 +943,17 @@ async def fail_point(self, command_args):
         try:
             yield
         finally:
-            await async_client_context.client.admin.command(
-                "configureFailPoint", cmd_on["configureFailPoint"], mode="off"
-            )
+            try:
+                await async_client_context.client.admin.command(
+                    "configureFailPoint", cmd_on["configureFailPoint"], mode="off"
+                )
+            except pymongo.errors.ConnectionFailure:
+                # Workaround PyPy bug described in PYTHON-5011.
+                if not _IS_SYNC and "PyPy" in sys.version:
+                    await async_client_context.client.admin.command(
+                        "configureFailPoint", cmd_on["configureFailPoint"], mode="off"
+                    )
+                raise
 
     @contextmanager
     def fork(