DataMovement unify TransferAndVerify Part 3 (#39401)

* block * append * net462 fix * default timeouts * test proxy
Azure · Oct 25, 2023 · a1996d0 · a1996d0
1 parent ef6502f
commit a1996d0
Show file tree

Hide file tree

Showing 3 changed files with 325 additions and 424 deletions.
diff --git a/sdk/storage/Azure.Storage.DataMovement/assets.json b/sdk/storage/Azure.Storage.DataMovement/assets.json
@@ -2,5 +2,5 @@
   "AssetsRepo": "Azure/azure-sdk-assets",
   "AssetsRepoPrefixPath": "net",
   "TagPrefix": "net/storage/Azure.Storage.DataMovement",
-  "Tag": "net/storage/Azure.Storage.DataMovement_0682ba876f"
+  "Tag": "net/storage/Azure.Storage.DataMovement_d8993077de"
 }
diff --git a/sdk/storage/Azure.Storage.DataMovement/tests/Shared/TransferValidator.cs b/sdk/storage/Azure.Storage.DataMovement/tests/Shared/TransferValidator.cs
@@ -38,6 +38,13 @@ public async Task TransferAndVerifyAsync(
             options ??= new DataTransferOptions();
             TestEventsRaised testEventsRaised = new TestEventsRaised(options);
 
+            if (cancellationToken == default)
+            {
+                CancellationTokenSource cts = new();
+                cts.CancelAfter(TimeSpan.FromSeconds(10));
+                cancellationToken = cts.Token;
+            }
+
             DataTransfer transfer = await TransferManager.StartTransferAsync(
                 sourceResource,
                 destinationResource,
@@ -74,6 +81,13 @@ public async Task TransferAndVerifyAsync(
             DataTransferOptions options = default,
             CancellationToken cancellationToken = default)
         {
+            if (cancellationToken == default)
+            {
+                CancellationTokenSource cts = new();
+                cts.CancelAfter(TimeSpan.FromSeconds(10));
+                cancellationToken = cts.Token;
+            }
+
             DataTransfer transfer = await TransferManager.StartTransferAsync(
                sourceResource,
                destinationResource,