drop db instead of just keyvault

Author: sleepyStick

test/asynchronous/test_encryption.py

@@ -2435,8 +2435,8 @@ async def asyncSetUp(self):
         await encrypted_client.drop_database("db")
 
         key_doc = json_data("etc", "data", "lookup", "key-doc.json")
-        key_vault = await create_key_vault(encrypted_client.db.keyvault, key_doc)
-        self.addCleanup(key_vault.drop)
+        await create_key_vault(encrypted_client.db.keyvault, key_doc)
+        self.addAsyncCleanup(async_client_context.client.drop_database("db"))
 
         await encrypted_client.db.create_collection(
             "csfle",

test/test_encryption.py

@@ -2419,8 +2419,8 @@ def setUp(self):
         encrypted_client.drop_database("db")
 
         key_doc = json_data("etc", "data", "lookup", "key-doc.json")
-        key_vault = create_key_vault(encrypted_client.db.keyvault, key_doc)
-        self.addCleanup(key_vault.drop)
+        create_key_vault(encrypted_client.db.keyvault, key_doc)
+        self.addCleanup(client_context.client.drop_database("db"))
 
         encrypted_client.db.create_collection(
             "csfle",