[Storage] delete/set_immutability_policy for BlobClient

sdk/storage/azure_storage_blob/src/clients/blob_client.rs

@@ -8,16 +8,19 @@ use crate::{
     generated::clients::PageBlobClient as GeneratedPageBlobClient,
     generated::models::{
         BlobClientAcquireLeaseResult, BlobClientBreakLeaseResult, BlobClientChangeLeaseResult,
-        BlobClientDownloadResult, BlobClientGetAccountInfoResult, BlobClientGetPropertiesResult,
-        BlobClientReleaseLeaseResult, BlobClientRenewLeaseResult, BlockBlobClientUploadResult,
+        BlobClientDeleteImmutabilityPolicyResult, BlobClientDownloadResult,
+        BlobClientGetAccountInfoResult, BlobClientGetPropertiesResult,
+        BlobClientReleaseLeaseResult, BlobClientRenewLeaseResult,
+        BlobClientSetImmutabilityPolicyResult, BlockBlobClientUploadResult,
     },
     models::{
         AccessTier, BlobClientAcquireLeaseOptions, BlobClientBreakLeaseOptions,
-        BlobClientChangeLeaseOptions, BlobClientDeleteOptions, BlobClientDownloadOptions,
-        BlobClientGetAccountInfoOptions, BlobClientGetPropertiesOptions, BlobClientGetTagsOptions,
-        BlobClientReleaseLeaseOptions, BlobClientRenewLeaseOptions, BlobClientSetMetadataOptions,
-        BlobClientSetPropertiesOptions, BlobClientSetTagsOptions, BlobClientSetTierOptions,
-        BlobTags, BlockBlobClientUploadOptions, StorageErrorCode,
+        BlobClientChangeLeaseOptions, BlobClientDeleteImmutabilityPolicyOptions,
+        BlobClientDeleteOptions, BlobClientDownloadOptions, BlobClientGetAccountInfoOptions,
+        BlobClientGetPropertiesOptions, BlobClientGetTagsOptions, BlobClientReleaseLeaseOptions,
+        BlobClientRenewLeaseOptions, BlobClientSetImmutabilityPolicyOptions,
+        BlobClientSetMetadataOptions, BlobClientSetPropertiesOptions, BlobClientSetTagsOptions,
+        BlobClientSetTierOptions, BlobTags, BlockBlobClientUploadOptions, StorageErrorCode,
     },
     pipeline::StorageHeadersPolicy,
     AppendBlobClient, BlobClientOptions, BlockBlobClient, PageBlobClient,
@@ -436,4 +439,28 @@ impl BlobClient {
             Err(e) => Err(e),
         }
     }
+
+    /// Sets the immutability policy on the blob.
+    ///
+    /// # Arguments
+    ///
+    /// * `options` - Optional configuration for the request.
+    pub async fn set_immutability_policy(
+        &self,
+        options: Option<BlobClientSetImmutabilityPolicyOptions<'_>>,
+    ) -> Result<Response<BlobClientSetImmutabilityPolicyResult, NoFormat>> {
+        self.client.set_immutability_policy(options).await
+    }
+
+    /// Deletes the immutability policy on the blob.
+    ///
+    /// # Arguments
+    ///
+    /// * `options` - Optional configuration for the request.
+    pub async fn delete_immutability_policy(
+        &self,
+        options: Option<BlobClientDeleteImmutabilityPolicyOptions<'_>>,
+    ) -> Result<Response<BlobClientDeleteImmutabilityPolicyResult, NoFormat>> {
+        self.client.delete_immutability_policy(options).await
+    }
 }

sdk/storage/azure_storage_blob/src/models/extensions.rs

@@ -2,8 +2,9 @@
 // Licensed under the MIT License.
 
 use crate::models::{
-    AppendBlobClientCreateOptions, BlobTag, BlobTags, BlockBlobClientUploadBlobFromUrlOptions,
-    BlockBlobClientUploadOptions, PageBlobClientCreateOptions,
+    AppendBlobClientCreateOptions, BlobImmutabilityPolicyMode, BlobTag, BlobTags,
+    BlockBlobClientUploadBlobFromUrlOptions, BlockBlobClientUploadOptions, ImmutabilityPolicyMode,
+    PageBlobClientCreateOptions,
 };
 use std::collections::HashMap;
 
@@ -108,3 +109,13 @@ impl From<HashMap<String, String>> for BlobTags {
         }
     }
 }
+
+// Converts a `ImmutabilityPolicyMode` into an `BlobImmutabilityPolicyMode`.
+impl From<ImmutabilityPolicyMode> for BlobImmutabilityPolicyMode {
+    fn from(mode: ImmutabilityPolicyMode) -> Self {
+        match mode {
+            ImmutabilityPolicyMode::Locked => BlobImmutabilityPolicyMode::Locked,
+            ImmutabilityPolicyMode::Unlocked => BlobImmutabilityPolicyMode::Unlocked,
+        }
+    }
+}

sdk/storage/azure_storage_blob/src/models/mod.rs

@@ -16,15 +16,16 @@ pub use crate::generated::models::{
     BlobClientBreakLeaseResultHeaders, BlobClientChangeLeaseOptions, BlobClientChangeLeaseResult,
     BlobClientChangeLeaseResultHeaders, BlobClientCopyFromUrlResult,
     BlobClientCopyFromUrlResultHeaders, BlobClientCreateSnapshotResult,
-    BlobClientCreateSnapshotResultHeaders, BlobClientDeleteImmutabilityPolicyResult,
-    BlobClientDeleteImmutabilityPolicyResultHeaders, BlobClientDeleteOptions,
-    BlobClientDownloadOptions, BlobClientDownloadResult, BlobClientDownloadResultHeaders,
-    BlobClientGetAccountInfoOptions, BlobClientGetAccountInfoResult,
-    BlobClientGetAccountInfoResultHeaders, BlobClientGetPropertiesOptions,
-    BlobClientGetPropertiesResult, BlobClientGetPropertiesResultHeaders, BlobClientGetTagsOptions,
-    BlobClientReleaseLeaseOptions, BlobClientReleaseLeaseResult,
-    BlobClientReleaseLeaseResultHeaders, BlobClientRenewLeaseOptions, BlobClientRenewLeaseResult,
-    BlobClientRenewLeaseResultHeaders, BlobClientSetExpiryResult, BlobClientSetExpiryResultHeaders,
+    BlobClientCreateSnapshotResultHeaders, BlobClientDeleteImmutabilityPolicyOptions,
+    BlobClientDeleteImmutabilityPolicyResult, BlobClientDeleteImmutabilityPolicyResultHeaders,
+    BlobClientDeleteOptions, BlobClientDownloadOptions, BlobClientDownloadResult,
+    BlobClientDownloadResultHeaders, BlobClientGetAccountInfoOptions,
+    BlobClientGetAccountInfoResult, BlobClientGetAccountInfoResultHeaders,
+    BlobClientGetPropertiesOptions, BlobClientGetPropertiesResult,
+    BlobClientGetPropertiesResultHeaders, BlobClientGetTagsOptions, BlobClientReleaseLeaseOptions,
+    BlobClientReleaseLeaseResult, BlobClientReleaseLeaseResultHeaders, BlobClientRenewLeaseOptions,
+    BlobClientRenewLeaseResult, BlobClientRenewLeaseResultHeaders, BlobClientSetExpiryResult,
+    BlobClientSetExpiryResultHeaders, BlobClientSetImmutabilityPolicyOptions,
     BlobClientSetImmutabilityPolicyResult, BlobClientSetImmutabilityPolicyResultHeaders,
     BlobClientSetLegalHoldResult, BlobClientSetLegalHoldResultHeaders,
     BlobClientSetMetadataOptions, BlobClientSetPropertiesOptions, BlobClientSetTagsOptions,

sdk/storage/azure_storage_blob/tests/blob_client.rs

@@ -3,6 +3,7 @@
 
 use azure_core::{
     http::{ClientOptions, RequestContent, StatusCode},
+    time::OffsetDateTime,
     Bytes,
 };
 use azure_core_test::{recorded, Matcher, TestContext};
@@ -12,8 +13,10 @@ use azure_storage_blob::{
         BlobClientChangeLeaseResultHeaders, BlobClientDownloadOptions,
         BlobClientDownloadResultHeaders, BlobClientGetAccountInfoResultHeaders,
         BlobClientGetPropertiesOptions, BlobClientGetPropertiesResultHeaders,
+        BlobClientSetImmutabilityPolicyOptions, BlobClientSetImmutabilityPolicyResultHeaders,
         BlobClientSetMetadataOptions, BlobClientSetPropertiesOptions, BlobClientSetTierOptions,
-        BlockBlobClientUploadOptions, LeaseState,
+        BlobImmutabilityPolicyMode, BlockBlobClientUploadOptions, ImmutabilityPolicyMode,
+        LeaseState,
     },
     BlobClient, BlobClientOptions, BlobContainerClient, BlobContainerClientOptions,
 };
@@ -644,3 +647,46 @@ async fn test_encoding_edge_cases(ctx: TestContext) -> Result<(), Box<dyn Error>
 
     Ok(())
 }
+
+#[recorded::test]
+async fn test_immutability_policy(ctx: TestContext) -> Result<(), Box<dyn Error>> {
+    //TODO: Point to versioning account, add datetimes to recording variables
+    // Recording Setup
+    let recording = ctx.recording();
+    let container_client = get_container_client(recording, false).await?;
+    let blob_client = container_client.blob_client(&get_blob_name(recording));
+    container_client.create_container(None).await?;
+    create_test_blob(&blob_client, None, None).await?;
+    let test_expiry_time = Some(OffsetDateTime::now_utc() + Duration::from_secs(300));
+    let test_immutability_policy_mode = ImmutabilityPolicyMode::Unlocked;
+
+    // Set Immutability Policy
+    let immutability_policy_options = BlobClientSetImmutabilityPolicyOptions {
+        immutability_policy_expiry: test_expiry_time.clone(),
+        immutability_policy_mode: Some(test_immutability_policy_mode),
+        ..Default::default()
+    };
+    let response = blob_client
+        .set_immutability_policy(Some(immutability_policy_options))
+        .await?;
+
+    // Assert
+    let mode = response.immutability_policy_mode()?;
+    let expires_on = response.immutability_policy_expires_on()?;
+
+    let response_mode: BlobImmutabilityPolicyMode = test_immutability_policy_mode.into();
+    assert_eq!(response_mode, mode.unwrap());
+    assert_eq!(test_expiry_time, expires_on);
+
+    // Delete Immutability Policy
+    blob_client.delete_immutability_policy(None).await?;
+    let response = blob_client.get_properties(None).await?;
+
+    // Assert
+    let mode = response.immutability_policy_mode()?;
+    let expires_on = response.immutability_policy_expires_on()?;
+    assert!(mode.is_none());
+    assert!(expires_on.is_none());
+
+    Ok(())
+}

sdk/storage/azure_storage_blob/tsp-location.yaml

@@ -1,4 +1,4 @@
 directory: specification/storage/Microsoft.BlobStorage
-commit: 6b5a0fdadca03abe0b8e91a68a62cd1e639ed55e
+commit: 64da9cf708dfbae0ff9ca2cde6829b123d9966bb
 repo: Azure/azure-rest-api-specs
 additionalDirectories: