fix (#39417)

Azure · Mar 28, 2024 · 688298d · 688298d
1 parent 695f020
commit 688298d
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/...torage-file-datalake/src/test/java/com/azure/storage/file/datalake/FileAsyncApiTests.java b/...torage-file-datalake/src/test/java/com/azure/storage/file/datalake/FileAsyncApiTests.java
@@ -2640,8 +2640,9 @@ public void appendDataLeaseAutoRenew() {
                 DataLakeFileAppendOptions appendOptions = new DataLakeFileAppendOptions()
                     .setLeaseAction(LeaseAction.AUTO_RENEW)
                     .setLeaseId(r.getT3());
-                return Mono.zip(r.getT2().appendWithResponse(DATA.getDefaultBinaryData(), 0, appendOptions), r.getT2().getProperties());
-            });
+                return Mono.zip(r.getT2().appendWithResponse(DATA.getDefaultBinaryData(), 0, appendOptions),
+                    Mono.just(r.getT2()));
+            }).flatMap(tuple -> Mono.zip(Mono.just(tuple.getT1()), tuple.getT2().getProperties()));
 
         StepVerifier.create(response)
             .assertNext(r -> {
@@ -2668,9 +2669,11 @@ public void appendDataLeaseRelease() {
                 Mono<Response<Void>> response1 = leaseClient.acquireLease(15)
                     .then(r.appendWithResponse(DATA.getDefaultBinaryData(), 0, appendOptions));
 
-                Mono<PathProperties> response2 = r.getProperties();
-
-                return Mono.zip(response1, response2);
+                return Mono.zip(response1, Mono.just(r));
+            })
+            .flatMap(tuple -> {
+                Mono<PathProperties> response2 = tuple.getT2().getProperties();
+                return Mono.zip(Mono.just(tuple.getT1()), response2);
             });
 
         StepVerifier.create(response)