Test run time optimizations (Azure#4384)

Author: g2vinay

sdk/keyvault/azure-keyvault-keys/src/test/java/com/azure/security/keyvault/keys/KeyAsyncClientTest.java

@@ -337,7 +337,7 @@ public void getDeletedKey() {
                         assertNotNull(deletedKey);
                     }).verifyComplete();
             pollOnKeyDeletion(keyToDeleteAndGet.name());
-            sleep(30000);
+            sleepInRecordMode(30000);
 
             StepVerifier.create(client.getDeletedKey(keyToDeleteAndGet.name()))
                     .assertNext(deletedKeyResponse -> {
@@ -353,7 +353,7 @@ public void getDeletedKey() {
                         assertEquals(HttpResponseStatus.NO_CONTENT.code(), voidResponse.statusCode());
                     }).verifyComplete();
             pollOnKeyPurge(keyToDeleteAndGet.name());
-            sleep(15000);
+            sleepInRecordMode(15000);
         });
     }
 //
@@ -421,9 +421,9 @@ public void listKeyVersions() {
                 client.createKey(key).subscribe(keyResponse -> assertKeyEquals(key, keyResponse.value()));
                 sleepInRecordMode(1000);
             }
-            sleep(30000);
+            sleepInRecordMode(30000);
             client.listKeyVersions(keyName).subscribe(output::add);
-            sleep(30000);
+            sleepInRecordMode(30000);
 
             assertEquals(keyVersions.size(), output.size());
 
@@ -455,9 +455,9 @@ public void listKeys() {
                 client.createKey(key).subscribe(keyResponse -> assertKeyEquals(key, keyResponse.value()));
                 sleepInRecordMode(1000);
             }
-            sleep(30000);
+            sleepInRecordMode(30000);
             client.listKeys().subscribe(output::add);
-            sleep(30000);
+            sleepInRecordMode(30000);
 
             for (KeyBase actualKey : output) {
                 if (keys.containsKey(actualKey.name())) {

sdk/keyvault/azure-keyvault-keys/src/test/java/com/azure/security/keyvault/keys/KeyClientTest.java

@@ -217,7 +217,7 @@ public void restoreKey() {
             pollOnKeyDeletion(keyToBackupAndRestore.name());
             client.purgeDeletedKey(keyToBackupAndRestore.name());
             pollOnKeyPurge(keyToBackupAndRestore.name());
-            sleep(60000);
+            sleepInRecordMode(60000);
             Key restoredKey = client.restoreKey(backupBytes).value();
             assertEquals(keyToBackupAndRestore.name(), restoredKey.name());
             assertEquals(keyToBackupAndRestore.expires(), restoredKey.expires());
@@ -240,7 +240,7 @@ public void listKeys() {
             HashMap<String, KeyCreateOptions> keysToList = keys;
             for (KeyCreateOptions key :  keysToList.values()) {
                 assertKeyEquals(key, client.createKey(key));
-                sleep(5000);
+                sleepInRecordMode(5000);
             }
 
             for (KeyBase actualKey : client.listKeys()) {

sdk/keyvault/azure-keyvault-secrets/src/test/java/com/azure/security/keyvault/secrets/SecretAsyncClientTest.java

@@ -217,7 +217,7 @@ public void getDeletedSecret() {
                         assertNotNull(deletedSecret);
                     }).verifyComplete();
             pollOnSecretDeletion(secretToDeleteAndGet.name());
-            sleep(30000);
+            sleepInRecordMode(30000);
 
             StepVerifier.create(client.getDeletedSecret(secretToDeleteAndGet.name()))
                     .assertNext(deletedSecretResponse -> {
@@ -233,7 +233,7 @@ public void getDeletedSecret() {
                         assertEquals(HttpResponseStatus.NO_CONTENT.code(), voidResponse.statusCode());
                     }).verifyComplete();
             pollOnSecretPurge(secretToDeleteAndGet.name());
-            sleep(10000);
+            sleepInRecordMode(10000);
         });
     }
 
@@ -331,7 +331,7 @@ public void restoreSecret() {
                     }).verifyComplete();
             pollOnSecretPurge(secretToBackupAndRestore.name());
 
-            sleep(60000);
+            sleepInRecordMode(60000);
 
             StepVerifier.create(client.restoreSecret(backup))
                     .assertNext(response -> {
@@ -416,9 +416,9 @@ public void listSecretVersions() {
                 client.setSecret(secret).subscribe(secretResponse -> assertSecretEquals(secret, secretResponse.value()));
                 sleepInRecordMode(1000);
             }
-            sleep(30000);
+            sleepInRecordMode(30000);
             client.listSecretVersions(secretName).subscribe(output::add);
-            sleep(30000);
+            sleepInRecordMode(30000);
 
             assertEquals(secretVersions.size(), output.size());
 
@@ -450,9 +450,9 @@ public void listSecrets() {
                 client.setSecret(secret).subscribe(secretResponse -> assertSecretEquals(secret, secretResponse.value()));
                 sleepInRecordMode(1000);
             }
-            sleep(30000);
+            sleepInRecordMode(30000);
             client.listSecrets().subscribe(output::add);
-            sleep(30000);
+            sleepInRecordMode(30000);
 
             for (SecretBase actualSecret : output) {
                 if (secretsToList.containsKey(actualSecret.name())) {