Service and Lease Api Async Tests (Azure#37057)

* service and lease tests

* service and lease tests

* more lease tests

* lease consistency

* rest of the lease tests

* recordings, extra tests, and line length

* style

Author: ibrandes

sdk/storage/azure-storage-file-datalake/assets.json

@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "java",
   "TagPrefix": "java/storage/azure-storage-file-datalake",
-  "Tag": "java/storage/azure-storage-file-datalake_8f2462b720"
+  "Tag": "java/storage/azure-storage-file-datalake_b9cf22c9e8"
 }

sdk/storage/azure-storage-file-datalake/src/test/java/com/azure/storage/file/datalake/DataLakeTestBase.java

@@ -263,6 +263,10 @@ protected DataLakeServiceAsyncClient getServiceAsyncClient(TestAccount account)
         return getServiceClientBuilder(account.getCredential(), account.getDataLakeEndpoint()).buildAsyncClient();
     }
 
+    protected DataLakeServiceAsyncClient getServiceAsyncClient(StorageSharedKeyCredential credential, String endpoint) {
+        return getServiceClientBuilder(credential, endpoint).buildAsyncClient();
+    }
+
     /**
      * Some tests require extra configuration for retries when writing.
      * <p>
@@ -553,7 +557,7 @@ protected String setupFileSystemLeaseCondition(DataLakeFileSystemClient fsc, Str
         return Objects.equals(RECEIVED_LEASE_ID, leaseID) ? createLeaseClient(fsc).acquireLease(-1) : leaseID;
     }
 
-    protected String setupFileSystemLeaseCondition(DataLakeFileSystemAsyncClient fsc, String leaseID) {
+    protected String setupFileSystemLeaseAsyncCondition(DataLakeFileSystemAsyncClient fsc, String leaseID) {
         return Objects.equals(RECEIVED_LEASE_ID, leaseID) ? createLeaseAsyncClient(fsc).acquireLease(-1).block() : leaseID;
     }
 
@@ -930,4 +934,6 @@ private static boolean olderThan20201002ServiceVersion() {
     private static boolean olderThan20210410ServiceVersion() {
         return olderThan(DataLakeServiceVersion.V2021_04_10);
     }
+
+
 }

sdk/storage/azure-storage-file-datalake/src/test/java/com/azure/storage/file/datalake/FileSystemAsyncApiTests.java

@@ -288,7 +288,7 @@ public void getPropertiesMin() {
 
     @Test
     public void getPropertiesLease() {
-        String leaseID = setupFileSystemLeaseCondition(dataLakeFileSystemAsyncClient, RECEIVED_LEASE_ID);
+        String leaseID = setupFileSystemLeaseAsyncCondition(dataLakeFileSystemAsyncClient, RECEIVED_LEASE_ID);
 
         assertAsyncResponseStatusCode(dataLakeFileSystemAsyncClient.getPropertiesWithResponse(leaseID), 200);
     }
@@ -376,7 +376,7 @@ public void setMetadataMetadata(String key1, String value1, String key2, String
     @ParameterizedTest
     @MethodSource("setMetadataACSupplier")
     public void setMetadataAC(OffsetDateTime modified, String leaseID) {
-        leaseID = setupFileSystemLeaseCondition(dataLakeFileSystemAsyncClient, leaseID);
+        leaseID = setupFileSystemLeaseAsyncCondition(dataLakeFileSystemAsyncClient, leaseID);
         DataLakeRequestConditions drc = new DataLakeRequestConditions().setLeaseId(leaseID).setIfModifiedSince(modified);
 
         assertAsyncResponseStatusCode(dataLakeFileSystemAsyncClient.setMetadataWithResponse(null, drc), 200);
@@ -465,7 +465,7 @@ public void deleteMin() {
     @MethodSource("modifiedAndLeaseIdSupplier")
     public void deleteAC(OffsetDateTime modified, OffsetDateTime unmodified, String leaseID) {
         DataLakeRequestConditions drc = new DataLakeRequestConditions()
-            .setLeaseId(setupFileSystemLeaseCondition(dataLakeFileSystemAsyncClient, leaseID))
+            .setLeaseId(setupFileSystemLeaseAsyncCondition(dataLakeFileSystemAsyncClient, leaseID))
             .setIfModifiedSince(modified)
             .setIfUnmodifiedSince(unmodified);
 
@@ -581,7 +581,7 @@ public void deleteIfExistsOnFileSystemThatDoesNotExist() {
     @MethodSource("modifiedAndLeaseIdSupplier")
     public void deleteIfExistsAC(OffsetDateTime modified, OffsetDateTime unmodified, String leaseID) {
         DataLakeRequestConditions drc = new DataLakeRequestConditions()
-            .setLeaseId(setupFileSystemLeaseCondition(dataLakeFileSystemAsyncClient, leaseID))
+            .setLeaseId(setupFileSystemLeaseAsyncCondition(dataLakeFileSystemAsyncClient, leaseID))
             .setIfModifiedSince(modified)
             .setIfUnmodifiedSince(unmodified);
 
@@ -945,7 +945,7 @@ public void createFileOptionsWithTimeToExpireRelativeToNow() {
     @Test
     public void createIfNotExistsFileMin() {
         StepVerifier.create(dataLakeFileSystemAsyncClient.createFileIfNotExists(generatePathName())
-            .flatMap(r -> r.exists()))
+            .flatMap(DataLakePathAsyncClient::exists))
             .expectNext(true)
             .verifyComplete();
     }
@@ -1580,7 +1580,7 @@ public void createDirOptionsWithTimeToExpire() {
     @Test
     public void createIfNotExistsDirMin() {
         StepVerifier.create(dataLakeFileSystemAsyncClient.createDirectoryIfNotExists(generatePathName())
-            .flatMap(r -> r.exists()))
+            .flatMap(DataLakePathAsyncClient::exists))
             .expectNext(true)
             .verifyComplete();
     }
@@ -2212,7 +2212,7 @@ public void setAccessPolicyIds() {
     @MethodSource("modifiedAndLeaseIdSupplier")
     public void setAccessPolicyAC(OffsetDateTime modified, OffsetDateTime unmodified, String leaseID) {
         DataLakeRequestConditions cac = new DataLakeRequestConditions()
-            .setLeaseId(setupFileSystemLeaseCondition(dataLakeFileSystemAsyncClient, leaseID))
+            .setLeaseId(setupFileSystemLeaseAsyncCondition(dataLakeFileSystemAsyncClient, leaseID))
             .setIfModifiedSince(modified)
             .setIfUnmodifiedSince(unmodified);
 
@@ -2275,7 +2275,7 @@ public void getAccessPolicy() {
 
     @Test
     public void getAccessPolicyLease() {
-        String leaseID = setupFileSystemLeaseCondition(dataLakeFileSystemAsyncClient, RECEIVED_LEASE_ID);
+        String leaseID = setupFileSystemLeaseAsyncCondition(dataLakeFileSystemAsyncClient, RECEIVED_LEASE_ID);
 
         assertAsyncResponseStatusCode(dataLakeFileSystemAsyncClient.getAccessPolicyWithResponse(leaseID), 200);
     }