fix flaky test setup

Author: blink1073

test/asynchronous/test_encryption.py

@@ -3262,6 +3262,7 @@ async def test_kms_retry(self):
 class TestAutomaticDecryptionKeys(AsyncEncryptionIntegrationTest):
     @async_client_context.require_no_standalone
     @async_client_context.require_version_min(7, 0, -1)
+    @flaky  # PYTHON-4982
     async def asyncSetUp(self):
         await super().asyncSetUp()
         self.key1_document = json_data("etc", "data", "keys", "key1-document.json")
@@ -3276,8 +3277,6 @@ async def asyncSetUp(self):
             OPTS,
         )
 
-    # PYTHON-4982
-    @flaky
     async def test_01_simple_create(self):
         coll, _ = await self.client_encryption.create_encrypted_collection(
             database=self.db,

test/test_encryption.py

@@ -3244,6 +3244,7 @@ def test_kms_retry(self):
 class TestAutomaticDecryptionKeys(EncryptionIntegrationTest):
     @client_context.require_no_standalone
     @client_context.require_version_min(7, 0, -1)
+    @flaky  # PYTHON-4982
     def setUp(self):
         super().setUp()
         self.key1_document = json_data("etc", "data", "keys", "key1-document.json")
@@ -3258,8 +3259,6 @@ def setUp(self):
             OPTS,
         )
 
-    # PYTHON-4982
-    @flaky
     def test_01_simple_create(self):
         coll, _ = self.client_encryption.create_encrypted_collection(
             database=self.db,