Skip to content

Commit e683b81

Browse files
authored
PYTHON-4739 - Use AsyncBulkTestBase in Async TestEncryptedBulkWrite (mongodb#1846)
1 parent f6a418f commit e683b81

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

test/asynchronous/test_encryption.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
import uuid
3030
import warnings
3131
from test.asynchronous import AsyncIntegrationTest, AsyncPyMongoTestCase, async_client_context
32+
from test.asynchronous.test_bulk import AsyncBulkTestBase
3233
from threading import Thread
3334
from typing import Any, Dict, Mapping
3435

@@ -52,7 +53,6 @@
5253
KMIP_CREDS,
5354
LOCAL_MASTER_KEY,
5455
)
55-
from test.test_bulk import BulkTestBase
5656
from test.unified_format import generate_test_classes
5757
from test.utils import (
5858
AllowListEventListener,
@@ -372,7 +372,7 @@ async def target():
372372
await target()
373373

374374

375-
class TestEncryptedBulkWrite(BulkTestBase, AsyncEncryptionIntegrationTest):
375+
class TestEncryptedBulkWrite(AsyncBulkTestBase, AsyncEncryptionIntegrationTest):
376376
async def test_upsert_uuid_standard_encrypt(self):
377377
opts = AutoEncryptionOpts(KMS_PROVIDERS, "keyvault.datakeys")
378378
client = await async_rs_or_single_client(auto_encryption_opts=opts)

test/test_encryption.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
import uuid
3030
import warnings
3131
from test import IntegrationTest, PyMongoTestCase, client_context
32+
from test.test_bulk import BulkTestBase
3233
from threading import Thread
3334
from typing import Any, Dict, Mapping
3435

@@ -52,7 +53,6 @@
5253
KMIP_CREDS,
5354
LOCAL_MASTER_KEY,
5455
)
55-
from test.test_bulk import BulkTestBase
5656
from test.unified_format import generate_test_classes
5757
from test.utils import (
5858
AllowListEventListener,

0 commit comments

Comments
 (0)