Rehydrate Priority In SetTier (#320)

Co-authored-by: t-iverma <t-iverma@microsoft.com>
Azure · Apr 21, 2022 · 7e62913 · 7e62913
1 parent afc471d
commit 7e62913
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/azblob/url_blob.go b/azblob/url_blob.go
@@ -183,10 +183,10 @@ func (b BlobURL) Undelete(ctx context.Context) (*BlobUndeleteResponse, error) {
 // Note: VersionId is an optional parameter which is part of request URL query params.
 // It can be explicitly set by calling WithVersionID(versionID string) function and hence it not required to pass it here.
 // For detailed information about block blob level tiering see https://docs.microsoft.com/en-us/azure/storage/blobs/storage-blob-storage-tiers.
-func (b BlobURL) SetTier(ctx context.Context, tier AccessTierType, lac LeaseAccessConditions) (*BlobSetTierResponse, error) {
+func (b BlobURL) SetTier(ctx context.Context, tier AccessTierType, lac LeaseAccessConditions, rehydratePriority RehydratePriorityType) (*BlobSetTierResponse, error) {
 	return b.blobClient.SetTier(ctx, tier, nil,
 		nil, // Blob versioning
-		nil, RehydratePriorityNone, nil, lac.pointers(),
+		nil, rehydratePriority, nil, lac.pointers(),
 		nil) // Blob ifTags
 }
 

diff --git a/azblob/zt_url_blob_test.go b/azblob/zt_url_blob_test.go
@@ -1700,7 +1700,7 @@ func (s *aztestsSuite) TestBlobsUndelete(c *chk.C) {
 }
 
 func setAndCheckBlobTier(c *chk.C, containerURL ContainerURL, blobURL BlobURL, tier AccessTierType) {
-	_, err := blobURL.SetTier(ctx, tier, LeaseAccessConditions{})
+	_, err := blobURL.SetTier(ctx, tier, LeaseAccessConditions{}, RehydratePriorityNone)
 	c.Assert(err, chk.IsNil)
 
 	resp, err := blobURL.GetProperties(ctx, BlobAccessConditions{}, ClientProvidedKeyOptions{})
@@ -1762,7 +1762,7 @@ func (s *aztestsSuite) TestBlobTierInferred(c *chk.C) {
 	c.Assert(resp2.Segment.BlobItems[0].Properties.AccessTierInferred, chk.NotNil)
 	c.Assert(resp2.Segment.BlobItems[0].Properties.AccessTier, chk.Not(chk.Equals), "")
 
-	_, err = blobURL.SetTier(ctx, AccessTierP4, LeaseAccessConditions{})
+	_, err = blobURL.SetTier(ctx, AccessTierP4, LeaseAccessConditions{}, RehydratePriorityNone)
 	c.Assert(err, chk.IsNil)
 
 	resp, err = blobURL.GetProperties(ctx, BlobAccessConditions{}, ClientProvidedKeyOptions{})
@@ -1784,9 +1784,9 @@ func (s *aztestsSuite) TestBlobArchiveStatus(c *chk.C) {
 	defer deleteContainer(c, containerURL, false)
 	blobURL, _ := createNewBlockBlob(c, containerURL)
 
-	_, err = blobURL.SetTier(ctx, AccessTierArchive, LeaseAccessConditions{})
+	_, err = blobURL.SetTier(ctx, AccessTierArchive, LeaseAccessConditions{}, RehydratePriorityNone)
 	c.Assert(err, chk.IsNil)
-	_, err = blobURL.SetTier(ctx, AccessTierCool, LeaseAccessConditions{})
+	_, err = blobURL.SetTier(ctx, AccessTierCool, LeaseAccessConditions{}, RehydratePriorityNone)
 	c.Assert(err, chk.IsNil)
 
 	resp, err := blobURL.GetProperties(ctx, BlobAccessConditions{}, ClientProvidedKeyOptions{})
@@ -1803,9 +1803,9 @@ func (s *aztestsSuite) TestBlobArchiveStatus(c *chk.C) {
 
 	blobURL, _ = createNewBlockBlob(c, containerURL)
 
-	_, err = blobURL.SetTier(ctx, AccessTierArchive, LeaseAccessConditions{})
+	_, err = blobURL.SetTier(ctx, AccessTierArchive, LeaseAccessConditions{}, RehydratePriorityNone)
 	c.Assert(err, chk.IsNil)
-	_, err = blobURL.SetTier(ctx, AccessTierHot, LeaseAccessConditions{})
+	_, err = blobURL.SetTier(ctx, AccessTierHot, LeaseAccessConditions{}, RehydratePriorityNone)
 	c.Assert(err, chk.IsNil)
 
 	resp, err = blobURL.GetProperties(ctx, BlobAccessConditions{}, ClientProvidedKeyOptions{})
@@ -1827,7 +1827,7 @@ func (s *aztestsSuite) TestBlobTierInvalidValue(c *chk.C) {
 	defer deleteContainer(c, containerURL, false)
 	blobURL, _ := createNewBlockBlob(c, containerURL)
 
-	_, err = blobURL.SetTier(ctx, AccessTierType("garbage"), LeaseAccessConditions{})
+	_, err = blobURL.SetTier(ctx, AccessTierType("garbage"), LeaseAccessConditions{}, RehydratePriorityNone)
 	validateStorageError(c, err, ServiceCodeInvalidHeaderValue)
 }