From 47b225702869a34e81b827036d4d9d1e77d1a532 Mon Sep 17 00:00:00 2001 From: Shruti Sridhar <77828382+shruti-sridhar@users.noreply.github.com> Date: Mon, 12 Aug 2024 17:59:44 -0700 Subject: [PATCH] PYTHON-4641 Fix failure in async version of client bulk CSOT test (#1790) --- test/asynchronous/test_client_bulk_write.py | 3 ++- test/test_client_bulk_write.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/test/asynchronous/test_client_bulk_write.py b/test/asynchronous/test_client_bulk_write.py index 7ce989bbe..4fe4fce2d 100644 --- a/test/asynchronous/test_client_bulk_write.py +++ b/test/asynchronous/test_client_bulk_write.py @@ -523,7 +523,7 @@ class TestClientBulkWriteCRUD(AsyncIntegrationTest): # https://github.com/mongodb/specifications/blob/master/source/client-side-operations-timeout/tests/README.md#11-multi-batch-bulkwrites -class TestClientBulkWriteTimeout(AsyncIntegrationTest): +class TestClientBulkWriteCSOT(AsyncIntegrationTest): async def asyncSetUp(self): self.max_write_batch_size = await async_client_context.max_write_batch_size self.max_bson_object_size = await async_client_context.max_bson_size @@ -567,6 +567,7 @@ class TestClientBulkWriteTimeout(AsyncIntegrationTest): w="majority", ) self.addAsyncCleanup(client.aclose) + await client.admin.command("ping") # Init the client first. with self.assertRaises(ClientBulkWriteException) as context: await client.bulk_write(models=models) self.assertIsInstance(context.exception.error, NetworkTimeout) diff --git a/test/test_client_bulk_write.py b/test/test_client_bulk_write.py index 639990880..8f6aad0cf 100644 --- a/test/test_client_bulk_write.py +++ b/test/test_client_bulk_write.py @@ -523,7 +523,7 @@ class TestClientBulkWriteCRUD(IntegrationTest): # https://github.com/mongodb/specifications/blob/master/source/client-side-operations-timeout/tests/README.md#11-multi-batch-bulkwrites -class TestClientBulkWriteTimeout(IntegrationTest): +class TestClientBulkWriteCSOT(IntegrationTest): def setUp(self): self.max_write_batch_size = client_context.max_write_batch_size self.max_bson_object_size = client_context.max_bson_size @@ -567,6 +567,7 @@ class TestClientBulkWriteTimeout(IntegrationTest): w="majority", ) self.addCleanup(client.close) + client.admin.command("ping") # Init the client first. with self.assertRaises(ClientBulkWriteException) as context: client.bulk_write(models=models) self.assertIsInstance(context.exception.error, NetworkTimeout)