mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

PYTHON-4739 - Use AsyncBulkTestBase in Async TestEncryptedBulkWrite (#1846)

Browse Source
This commit is contained in:
Noah Stapp 2024-09-09 11:05:04 -04:00 committed by GitHub
parent f6a418f590
commit e683b81bf4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/asynchronous/test_encryption.py
View File

@ -29,6 +29,7 @@ import traceback
import uuid
import warnings
from test.asynchronous import AsyncIntegrationTest, AsyncPyMongoTestCase, async_client_context
from test.asynchronous.test_bulk import AsyncBulkTestBase
from threading import Thread
from typing import Any, Dict, Mapping
@ -52,7 +53,6 @@ from test.helpers import (
KMIP_CREDS,
LOCAL_MASTER_KEY,
)
from test.test_bulk import BulkTestBase
from test.unified_format import generate_test_classes
from test.utils import (
AllowListEventListener,
@ -372,7 +372,7 @@ class TestClientSimple(AsyncEncryptionIntegrationTest):
await target()
class TestEncryptedBulkWrite(BulkTestBase, AsyncEncryptionIntegrationTest):
class TestEncryptedBulkWrite(AsyncBulkTestBase, AsyncEncryptionIntegrationTest):
async def test_upsert_uuid_standard_encrypt(self):
opts = AutoEncryptionOpts(KMS_PROVIDERS, "keyvault.datakeys")
client = await async_rs_or_single_client(auto_encryption_opts=opts)

2
test/test_encryption.py
View File

@ -29,6 +29,7 @@ import traceback
import uuid
import warnings
from test import IntegrationTest, PyMongoTestCase, client_context
from test.test_bulk import BulkTestBase
from threading import Thread
from typing import Any, Dict, Mapping
@ -52,7 +53,6 @@ from test.helpers import (
KMIP_CREDS,
LOCAL_MASTER_KEY,
)
from test.test_bulk import BulkTestBase
from test.unified_format import generate_test_classes
from test.utils import (
AllowListEventListener,