Fixed large file live test to add policy per call instead of per retry (

#17593) Co-authored-by: gapra <gapra@microsoft.com>
Azure · Nov 16, 2020 · 9215dcc · 9215dcc
1 parent c97aee7
commit 9215dcc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/sdk/storage/azure-storage-blob/src/test/java/com/azure/storage/blob/LargeBlobTest.groovy b/sdk/storage/azure-storage-blob/src/test/java/com/azure/storage/blob/LargeBlobTest.groovy
@@ -6,6 +6,7 @@ package com.azure.storage.blob
 import com.azure.core.http.HttpMethod
 import com.azure.core.http.HttpPipelineCallContext
 import com.azure.core.http.HttpPipelineNextPolicy
+import com.azure.core.http.HttpPipelinePosition
 import com.azure.core.http.HttpRequest
 import com.azure.core.http.HttpResponse
 import com.azure.core.http.policy.HttpPipelinePolicy
@@ -383,5 +384,9 @@ class LargeBlobTest extends APISpec {
                 request.getUrl().getPath().endsWith(blobName) &&
                 request.getUrl().getQuery() == null
         }
+
+        HttpPipelinePosition getPipelinePosition() {
+            return HttpPipelinePosition.PER_CALL;
+        }
     }
 }
diff --git a/...-storage-file-datalake/src/test/java/com/azure/storage/file/datalake/LargeFileTest.groovy b/...-storage-file-datalake/src/test/java/com/azure/storage/file/datalake/LargeFileTest.groovy
@@ -2,6 +2,7 @@ package com.azure.storage.file.datalake
 
 import com.azure.core.http.HttpPipelineCallContext
 import com.azure.core.http.HttpPipelineNextPolicy
+import com.azure.core.http.HttpPipelinePosition
 import com.azure.core.http.HttpRequest
 import com.azure.core.http.HttpResponse
 import com.azure.core.http.policy.HttpPipelinePolicy
@@ -273,5 +274,9 @@ class LargeFileTest extends APISpec{
         private boolean isFlushRequest(HttpRequest request) {
             return request.url.getQuery() != null && request.url.getQuery().contains("action=flush")
         }
+
+        HttpPipelinePosition getPipelinePosition() {
+            return HttpPipelinePosition.PER_CALL;
+        }
     }
 }