catch connection failure

Author: blink1073

test/asynchronous/test_discovery_and_monitoring.py

@@ -28,6 +28,7 @@
 from test.utils_shared import delay
 
 from pymongo.asynchronous.pool import AsyncConnection
+from pymongo.errors import ConnectionFailure
 from pymongo.operations import _Op
 from pymongo.server_selectors import writable_server_selector
 
@@ -491,7 +492,7 @@ async def teardown():
         async def target():
             try:
                 await client.test.test.find_one({"$where": delay(0.1)})
-            except OperationFailure:
+            except ConnectionFailure:
                 pass
 
         # Warm the pool with 10 tasks so there are existing connections.

test/test_discovery_and_monitoring.py

@@ -27,6 +27,7 @@
 from test.utils import flaky
 from test.utils_shared import delay
 
+from pymongo.errors import ConnectionFailure
 from pymongo.operations import _Op
 from pymongo.server_selectors import writable_server_selector
 from pymongo.synchronous.pool import Connection
@@ -485,7 +486,7 @@ def teardown():
         def target():
             try:
                 client.test.test.find_one({"$where": delay(0.1)})
-            except OperationFailure:
+            except ConnectionFailure:
                 pass
 
         # Warm the pool with 10 tasks so there are existing connections.