[Key Vault] Update tests which disable soft-delete

sdk/keyvault/azure-keyvault-certificates/tests/test_certificates_client.py

@@ -8,6 +8,7 @@
 import time
 import pytest
 
+from azure.core.exceptions import ResourceExistsError
 from azure_devtools.scenario_tests import RecordingProcessor, RequestUrlNormalizer
 
 from azure.keyvault.certificates import (
@@ -480,7 +481,7 @@ def test_get_pending_certificate_signing_request(self, client, **kwargs):
         self.assertEqual(client.get_certificate_operation(certificate_name=cert_name).csr, pending_version_csr)
 
     @ResourceGroupPreparer(random_name_enabled=True)
-    @KeyVaultPreparer(enable_soft_delete=False)
+    @KeyVaultPreparer()
     @KeyVaultClientPreparer()
     def test_backup_restore(self, client, **kwargs):
         policy = CertificatePolicy.get_default()
@@ -496,8 +497,12 @@ def test_backup_restore(self, client, **kwargs):
         # delete the certificate
         client.begin_delete_certificate(certificate_name=cert_name).wait()
 
+        # purge the certificate
+        client.purge_deleted_certificate(certificate_name=cert_name)
+
         # restore certificate
-        restored_certificate = client.restore_certificate_backup(backup=certificate_backup)
+        restore_function = functools.partial(client.restore_certificate_backup, certificate_backup)
+        restored_certificate = self._poll_until_no_exception(restore_function, ResourceExistsError)
         self._validate_certificate_bundle(cert=restored_certificate, cert_name=cert_name, cert_policy=policy)
 
     @ResourceGroupPreparer(random_name_enabled=True)

sdk/keyvault/azure-keyvault-certificates/tests/test_certificates_client_async.py

@@ -7,6 +7,7 @@
 import logging
 import json
 
+from azure.core.exceptions import ResourceExistsError
 from azure_devtools.scenario_tests import RecordingProcessor
 from azure.keyvault.certificates import (
     AdministratorContact,
@@ -492,7 +493,7 @@ async def test_get_pending_certificate_signing_request(self, client, **kwargs):
         self.assertEqual((await client.get_certificate_operation(certificate_name=cert_name)).csr, pending_version_csr)
 
     @ResourceGroupPreparer(random_name_enabled=True)
-    @KeyVaultPreparer(enable_soft_delete=False)
+    @KeyVaultPreparer()
     @KeyVaultClientPreparer()
     async def test_backup_restore(self, client, **kwargs):
         cert_name = self.get_resource_name("cert")
@@ -508,8 +509,14 @@ async def test_backup_restore(self, client, **kwargs):
         # delete the certificate
         await client.delete_certificate(certificate_name=cert_name)
 
+        # purge the certificate
+        await client.purge_deleted_certificate(certificate_name=cert_name)
+
         # restore certificate
-        restored_certificate = await client.restore_certificate_backup(backup=certificate_backup)
+        await self._poll_until_no_exception(
+            client.restore_certificate_backup, certificate_backup, expected_exception=ResourceExistsError
+        )
+        restored_certificate = await client.get_certificate(certificate_name=cert_name)
         self._validate_certificate_bundle(cert=restored_certificate, cert_name=cert_name, cert_policy=policy)
 
     @ResourceGroupPreparer(random_name_enabled=True)

sdk/keyvault/azure-keyvault-certificates/tests/test_examples_certificates.py

@@ -4,6 +4,7 @@
 # ------------------------------------
 from __future__ import print_function
 import functools
+import time
 
 from azure.keyvault.certificates import (
     CertificateClient,
@@ -198,7 +199,7 @@ def test_example_certificate_list_operations(self, client, **kwargs):
         # [END list_deleted_certificates]
 
     @ResourceGroupPreparer(random_name_enabled=True)
-    @KeyVaultPreparer(enable_soft_delete=False)
+    @KeyVaultPreparer()
     @KeyVaultClientPreparer()
     def test_example_certificate_backup_restore(self, client, **kwargs):
         certificate_client = client
@@ -230,6 +231,10 @@ def test_example_certificate_backup_restore(self, client, **kwargs):
         # [END backup_certificate]
 
         certificate_client.begin_delete_certificate(certificate_name=cert_name).wait()
+        certificate_client.purge_deleted_certificate(certificate_name=cert_name)
+
+        if self.is_live:
+            time.sleep(15)
 
         # [START restore_certificate]
 

sdk/keyvault/azure-keyvault-certificates/tests/test_examples_certificates_async.py

@@ -3,6 +3,7 @@
 # Licensed under the MIT License.
 # ------------------------------------
 import functools
+import time
 
 from azure.keyvault.certificates import CertificatePolicy, CertificateContentType, WellKnownIssuerNames
 from azure.keyvault.certificates.aio import CertificateClient
@@ -184,7 +185,7 @@ async def test_example_certificate_list_operations(self, client, **kwargs):
         # [END list_deleted_certificates]
 
     @ResourceGroupPreparer(random_name_enabled=True)
-    @KeyVaultPreparer(enable_soft_delete=False)
+    @KeyVaultPreparer()
     @KeyVaultClientPreparer()
     async def test_example_certificate_backup_restore(self, client, **kwargs):
         certificate_client = client
@@ -220,6 +221,10 @@ async def test_example_certificate_backup_restore(self, client, **kwargs):
         # [END backup_certificate]
 
         await certificate_client.delete_certificate(certificate_name=cert_name)
+        await certificate_client.purge_deleted_certificate(certificate_name=cert_name)
+
+        if self.is_live:
+            time.sleep(15)
 
         # [START restore_certificate]