address teardown failures in data lake tests

Author: blink1073

test/__init__.py

@@ -1189,6 +1189,7 @@ def setUp(self) -> None:
         if client_context.serverless and not getattr(self, "RUN_ON_SERVERLESS", False):
             raise SkipTest("this test does not support serverless")
         self.client = client_context.client
+        self.addCleanup(self.client.close)
         self.db = self.client.pymongo_test
         if client_context.auth_enabled:
             self.credentials = {"username": db_user, "password": db_pwd}

test/asynchronous/__init__.py

@@ -1207,6 +1207,7 @@ async def asyncSetUp(self) -> None:
         if async_client_context.serverless and not getattr(self, "RUN_ON_SERVERLESS", False):
             raise SkipTest("this test does not support serverless")
         self.client = async_client_context.client
+        self.addAsyncCleanup(self.client.close)
         self.db = self.client.pymongo_test
         if async_client_context.auth_enabled:
             self.credentials = {"username": db_user, "password": db_pwd}