Feature/storage/set blob tier snapshot version

sdk/storage/Azure.Storage.Blobs.Batch/tests/BlobBatchClientTests.cs

@@ -569,6 +569,51 @@ public async Task Batch_Double_Dispose_Response_Still_Available()
             scenario.AssertStatus(202, responses);
             await scenario.AssertDeleted(blobs);
         }
+
+        [Test]
+        [ServiceVersion(Min = BlobClientOptions.ServiceVersion.V2019_12_12)]
+        public async Task SetBlobAccessTier_Snapshot()
+        {
+            await using TestScenario scenario = Scenario();
+            BlockBlobClient[] blobs = await scenario.CreateBlockBlobsAsync(1);
+            Response<BlobSnapshotInfo> blobSnapshotResponse = await blobs[0].CreateSnapshotAsync();
+            blobs[0] = blobs[0].WithSnapshot(blobSnapshotResponse.Value.Snapshot);
+
+            BlobBatchClient client = scenario.GetBlobBatchClient();
+            using BlobBatch batch = client.CreateBatch();
+            Response[] responses = new Response[]
+            {
+                batch.SetBlobAccessTier(blobs[0].Uri, AccessTier.Cool),
+            };
+            Response response = await client.SubmitBatchAsync(batch);
+
+            scenario.AssertStatus(202, response);
+            scenario.AssertStatus(200, responses);
+            await scenario.AssertTiers(AccessTier.Cool, blobs);
+        }
+
+        [Test]
+        [ServiceVersion(Min = BlobClientOptions.ServiceVersion.V2019_12_12)]
+        public async Task SetBlobAccessTier_Version()
+        {
+            await using TestScenario scenario = Scenario();
+            BlockBlobClient[] blobs = await scenario.CreateBlockBlobsAsync(1);
+            Response<BlobInfo> setMetadataResponse = await blobs[0].SetMetadataAsync(BuildMetadata());
+
+            blobs[0] = blobs[0].WithVersion(setMetadataResponse.Value.VersionId);
+
+            BlobBatchClient client = scenario.GetBlobBatchClient();
+            using BlobBatch batch = client.CreateBatch();
+            Response[] responses = new Response[]
+            {
+                batch.SetBlobAccessTier(blobs[0].Uri, AccessTier.Cool),
+            };
+            Response response = await client.SubmitBatchAsync(batch);
+
+            scenario.AssertStatus(202, response);
+            scenario.AssertStatus(200, responses);
+            await scenario.AssertTiers(AccessTier.Cool, blobs);
+        }
         #endregion SetBlobAccessTier
 
         #region Scenario helper
@@ -609,9 +654,23 @@ public async Task<BlobClient[]> CreateBlobsAsync(BlobContainerClient container,
                 return blobs;
             }
 
+            public async Task<BlockBlobClient[]> CreateBlockBlobsAsync(BlobContainerClient container, int count)
+            {
+                BlockBlobClient[] blobs = new BlockBlobClient[count];
+                for (int i = 0; i < count; i++)
+                {
+                    blobs[i] = _test.InstrumentClient(container.GetBlockBlobClient("blob" + (++_blobId)));
+                    await blobs[i].UploadAsync(new MemoryStream(_test.GetRandomBuffer(Constants.KB)));
+                }
+                return blobs;
+            }
+
             public async Task<BlobClient[]> CreateBlobsAsync(int count) =>
                 await CreateBlobsAsync(await CreateContainerAsync(), count);
 
+            public async Task<BlockBlobClient[]> CreateBlockBlobsAsync(int count) =>
+                await CreateBlockBlobsAsync(await CreateContainerAsync(), count);
+
             public async Task<Uri[]> CreateBlobUrisAsync(BlobContainerClient container, int count) =>
                 (await CreateBlobsAsync(container, count)).Select(b => b.Uri).ToArray();
 
@@ -661,9 +720,24 @@ public async Task AssertTiers(AccessTier tier, BlobClient blob)
                 }
             }
 
+            public async Task AssertTiers(AccessTier tier, BlockBlobClient blob)
+            {
+                try
+                {
+                    BlobProperties properties = await blob.GetPropertiesAsync();
+                    Assert.AreEqual(tier.ToString(), properties.AccessTier.ToString());
+                }
+                catch (RequestFailedException)
+                {
+                }
+            }
+
             public Task AssertTiers(AccessTier tier, BlobClient[] blobs) =>
                 Task.WhenAll(blobs.Select(b => AssertTiers(tier, b)));
 
+            public Task AssertTiers(AccessTier tier, BlockBlobClient[] blobs) =>
+                Task.WhenAll(blobs.Select(b => AssertTiers(tier, b)));
+
             public void AssertStatus(int status, params Response[] responses) =>
                 Assert.IsTrue(responses.All(r => r.Status == status));