Address review

Author: NoahStapp

test/asynchronous/test_client.py

@@ -1907,7 +1907,6 @@ async def test_direct_connection(self):
             AsyncMongoClient(["host1", "host2"], directConnection=True)
 
     @unittest.skipIf("PyPy" in sys.version, "PYTHON-2927 fails often on PyPy")
-    @skipIf(os.environ.get("DEBUG_LOG"), "Enabling debug logs breaks this test")
     async def test_continuous_network_errors(self):
         def server_description_count():
             i = 0

test/asynchronous/test_discovery_and_monitoring.py

@@ -433,12 +433,15 @@ async def mock_close(self, reason):
                 await listener.async_wait_for_event(monitoring.ServerHeartbeatFailedEvent, 1)
             # Wait until all idle connections are closed to simulate real-world conditions
             await listener.async_wait_for_event(monitoring.ConnectionClosedEvent, 10)
+            # Wait for one more find to complete after the pool has been reset, then shutdown the task
+            n = len(latencies)
+            await async_wait_until(lambda: len(latencies) >= n + 1, "run one more find")
             should_exit.append(True)
+            await task.join()
             # No operation latency should not significantly exceed close_delay
             self.assertLessEqual(max(latencies), close_delay * 5.0)
         finally:
             AsyncConnection.close_conn = original_close
-            await task.join()
 
 
 class TestServerMonitoringMode(AsyncIntegrationTest):

test/test_client.py

@@ -1864,7 +1864,6 @@ def test_direct_connection(self):
             MongoClient(["host1", "host2"], directConnection=True)
 
     @unittest.skipIf("PyPy" in sys.version, "PYTHON-2927 fails often on PyPy")
-    @skipIf(os.environ.get("DEBUG_LOG"), "Enabling debug logs breaks this test")
     def test_continuous_network_errors(self):
         def server_description_count():
             i = 0

test/test_discovery_and_monitoring.py

@@ -431,12 +431,15 @@ def mock_close(self, reason):
                 listener.wait_for_event(monitoring.ServerHeartbeatFailedEvent, 1)
             # Wait until all idle connections are closed to simulate real-world conditions
             listener.wait_for_event(monitoring.ConnectionClosedEvent, 10)
+            # Wait for one more find to complete after the pool has been reset, then shutdown the task
+            n = len(latencies)
+            wait_until(lambda: len(latencies) >= n + 1, "run one more find")
             should_exit.append(True)
+            task.join()
             # No operation latency should not significantly exceed close_delay
             self.assertLessEqual(max(latencies), close_delay * 5.0)
         finally:
             Connection.close_conn = original_close
-            task.join()
 
 
 class TestServerMonitoringMode(IntegrationTest):