add storage tests (#25532)

Azure · Jan 29, 2022 · 460ec74 · 460ec74
1 parent b122d32
commit 460ec74
Show file tree

Hide file tree

Showing 12 changed files with 27,576 additions and 0 deletions.
diff --git a/...Manager.Storage/tests/SessionRecords/BlobContainerTests/BlobContainerEncryptionScope.json b/...Manager.Storage/tests/SessionRecords/BlobContainerTests/BlobContainerEncryptionScope.json
diff --git a/...er.Storage/tests/SessionRecords/BlobContainerTests/BlobContainerEncryptionScopeAsync.json b/...er.Storage/tests/SessionRecords/BlobContainerTests/BlobContainerEncryptionScopeAsync.json
diff --git a/...ure.ResourceManager.Storage/tests/SessionRecords/BlobContainerTests/GetBlobContainer.json b/...ure.ResourceManager.Storage/tests/SessionRecords/BlobContainerTests/GetBlobContainer.json
diff --git a/...esourceManager.Storage/tests/SessionRecords/BlobContainerTests/GetBlobContainerAsync.json b/...esourceManager.Storage/tests/SessionRecords/BlobContainerTests/GetBlobContainerAsync.json
diff --git a/...urceManager.Storage/tests/SessionRecords/BlobContainerTests/UpdateImmutabilityPolicy.json b/...urceManager.Storage/tests/SessionRecords/BlobContainerTests/UpdateImmutabilityPolicy.json
diff --git a/...anager.Storage/tests/SessionRecords/BlobContainerTests/UpdateImmutabilityPolicyAsync.json b/...anager.Storage/tests/SessionRecords/BlobContainerTests/UpdateImmutabilityPolicyAsync.json
diff --git a/...Manager.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountGetOperations.json b/...Manager.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountGetOperations.json
diff --git a/...er.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountGetOperationsAsync.json b/...er.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountGetOperationsAsync.json
diff --git a/...rceManager.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountOperations.json b/...rceManager.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountOperations.json
diff --git a/...nager.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountOperationsAsync.json b/...nager.Storage/tests/SessionRecords/StorageAccountTests/StorageAccountOperationsAsync.json
diff --git a/sdk/storage/Azure.ResourceManager.Storage/tests/Tests/BlobContainerTests.cs b/sdk/storage/Azure.ResourceManager.Storage/tests/Tests/BlobContainerTests.cs
@@ -78,6 +78,63 @@ public async Task CreateDeleteBlobContainer()
             Assert.IsFalse(await _blobContainerCollection.ExistsAsync(containerName));
         }
 
+        [Test]
+        [RecordedTest]
+        public async Task GetBlobContainer()
+        {
+            string containerName = Recording.GenerateAssetName("testblob");
+            BlobContainer container = (await _blobContainerCollection.CreateOrUpdateAsync(containerName, new BlobContainerData())).Value;
+
+            Assert.IsEmpty(container.Data.Metadata);
+            Assert.Null(container.Data.PublicAccess);
+
+            LegalHold legalHoldModel = new LegalHold(new List<string> { "tag1", "tag2", "tag3" });
+            LegalHold legalHold = await container.SetLegalHoldAsync(legalHoldModel);
+            Assert.IsTrue(legalHold.HasLegalHold);
+            Assert.AreEqual(new List<string> { "tag1", "tag2", "tag3" }, legalHold.Tags);
+
+            ImmutabilityPolicy immutabilityPolicyModel = new ImmutabilityPolicy() { ImmutabilityPeriodSinceCreationInDays = 3 };
+            ImmutabilityPolicy immutabilityPolicy = await container.CreateOrUpdateImmutabilityPolicyAsync(parameters: immutabilityPolicyModel);
+            Assert.NotNull(immutabilityPolicy.Id);
+            Assert.NotNull(immutabilityPolicy.Type);
+            Assert.NotNull(immutabilityPolicy.Name);
+            Assert.AreEqual(3, immutabilityPolicy.ImmutabilityPeriodSinceCreationInDays);
+            Assert.AreEqual(ImmutabilityPolicyState.Unlocked, immutabilityPolicy.State);
+
+            immutabilityPolicy = await container.LockImmutabilityPolicyAsync(immutabilityPolicy.Etag);
+            Assert.NotNull(immutabilityPolicy.Id);
+            Assert.NotNull(immutabilityPolicy.Type);
+            Assert.NotNull(immutabilityPolicy.Name);
+            Assert.AreEqual(3, immutabilityPolicy.ImmutabilityPeriodSinceCreationInDays);
+            Assert.AreEqual(ImmutabilityPolicyState.Locked, immutabilityPolicy.State);
+
+            immutabilityPolicyModel = new ImmutabilityPolicy() { ImmutabilityPeriodSinceCreationInDays = 100 };
+            immutabilityPolicy = await container.ExtendImmutabilityPolicyAsync(immutabilityPolicy.Etag, parameters: immutabilityPolicyModel);
+            Assert.NotNull(immutabilityPolicy.Id);
+            Assert.NotNull(immutabilityPolicy.Type);
+            Assert.NotNull(immutabilityPolicy.Name);
+            Assert.AreEqual(100, immutabilityPolicy.ImmutabilityPeriodSinceCreationInDays);
+            Assert.AreEqual(ImmutabilityPolicyState.Locked, immutabilityPolicy.State);
+
+            container = await container.GetAsync();
+            Assert.IsEmpty(container.Data.Metadata);
+            Assert.AreEqual(PublicAccess.None, container.Data.PublicAccess);
+            Assert.AreEqual(3, container.Data.ImmutabilityPolicy.UpdateHistory.Count);
+            Assert.AreEqual(ImmutabilityPolicyUpdateType.Put, container.Data.ImmutabilityPolicy.UpdateHistory[0].Update);
+            Assert.AreEqual(ImmutabilityPolicyUpdateType.Lock, container.Data.ImmutabilityPolicy.UpdateHistory[1].Update);
+            Assert.AreEqual(ImmutabilityPolicyUpdateType.Extend, container.Data.ImmutabilityPolicy.UpdateHistory[2].Update);
+            Assert.IsTrue(container.Data.LegalHold.HasLegalHold);
+            Assert.AreEqual(3, container.Data.LegalHold.Tags.Count);
+            Assert.AreEqual("tag1", container.Data.LegalHold.Tags[0].Tag);
+            Assert.AreEqual("tag2", container.Data.LegalHold.Tags[1].Tag);
+            Assert.AreEqual("tag3", container.Data.LegalHold.Tags[2].Tag);
+
+            legalHold = await container.ClearLegalHoldAsync(legalHold);
+            Assert.IsFalse(legalHold.HasLegalHold);
+
+            await container.DeleteAsync();
+        }
+
         [Test]
         [RecordedTest]
         public async Task GetAllBlobContainers()
@@ -158,6 +215,49 @@ public async Task CreateDeleteImmutabilityPolicy()
             Assert.AreEqual(0, immutabilityPolicyModel.ImmutabilityPeriodSinceCreationInDays);
         }
 
+        [Test]
+        [RecordedTest]
+        public async Task UpdateImmutabilityPolicy()
+        {
+            // create a blob container
+            string containerName = Recording.GenerateAssetName("testblob");
+            BlobContainerData data = new BlobContainerData();
+            BlobContainer container = (await _blobContainerCollection.CreateOrUpdateAsync(containerName, new BlobContainerData())).Value;
+
+            //create immutability policy
+            ImmutabilityPolicy immutabilityPolicyModel = new ImmutabilityPolicy() { ImmutabilityPeriodSinceCreationInDays = 3 };
+            ImmutabilityPolicy immutabilityPolicy = await container.CreateOrUpdateImmutabilityPolicyAsync(parameters: immutabilityPolicyModel);
+
+            //validate
+            Assert.NotNull(immutabilityPolicy.Id);
+            Assert.NotNull(immutabilityPolicy.Type);
+            Assert.NotNull(immutabilityPolicy.Name);
+            Assert.AreEqual(3, immutabilityPolicy.ImmutabilityPeriodSinceCreationInDays);
+            Assert.AreEqual(ImmutabilityPolicyState.Unlocked, immutabilityPolicy.State);
+
+            //update immutability policy
+            immutabilityPolicyModel = new ImmutabilityPolicy()
+            {
+                ImmutabilityPeriodSinceCreationInDays = 5,
+                AllowProtectedAppendWrites = true
+            };
+            immutabilityPolicy = await container.CreateOrUpdateImmutabilityPolicyAsync(ifMatch: immutabilityPolicy.Etag, parameters: immutabilityPolicyModel);
+            Assert.NotNull(immutabilityPolicy.Id);
+            Assert.NotNull(immutabilityPolicy.Type);
+            Assert.NotNull(immutabilityPolicy.Name);
+            Assert.AreEqual(5, immutabilityPolicy.ImmutabilityPeriodSinceCreationInDays);
+            Assert.AreEqual(ImmutabilityPolicyState.Unlocked, immutabilityPolicy.State);
+            Assert.IsTrue(immutabilityPolicy.AllowProtectedAppendWrites);
+
+            immutabilityPolicy = await container.GetImmutabilityPolicyAsync(ifMatch: immutabilityPolicy.Etag);
+            Assert.NotNull(immutabilityPolicy.Id);
+            Assert.NotNull(immutabilityPolicy.Type);
+            Assert.NotNull(immutabilityPolicy.Name);
+            Assert.AreEqual(5, immutabilityPolicy.ImmutabilityPeriodSinceCreationInDays);
+            Assert.AreEqual(ImmutabilityPolicyState.Unlocked, immutabilityPolicy.State);
+            Assert.IsTrue(immutabilityPolicy.AllowProtectedAppendWrites);
+        }
+
         [Test]
         [RecordedTest]
         public async Task LockImmutabilityPolicy()
@@ -623,5 +723,32 @@ public async Task BlobContainersVLW()
             Assert.IsTrue(container2.Data.ImmutableStorageWithVersioning.Enabled);
             Assert.AreEqual("Completed", container2.Data.ImmutableStorageWithVersioning.MigrationState);
         }
+
+        [Test]
+        [RecordedTest]
+        public async Task BlobContainerEncryptionScope()
+        {
+            //create encryption scope
+            string scopeName1 = "testscope1";
+            string scopeName2 = "testscope2";
+            EncryptionScopeData data = new EncryptionScopeData()
+            {
+                Source = EncryptionScopeSource.MicrosoftStorage,
+                State = EncryptionScopeState.Enabled
+            };
+            await _storageAccount.GetEncryptionScopes().CreateOrUpdateAsync(scopeName1, data);
+            await _storageAccount.GetEncryptionScopes().CreateOrUpdateAsync(scopeName2, data);
+
+            //create container
+            string containerName = Recording.GenerateAssetName("container");
+            BlobContainer blobContainer = (await _blobContainerCollection.CreateOrUpdateAsync(containerName, new BlobContainerData() { DefaultEncryptionScope = scopeName1, DenyEncryptionScopeOverride = false })).Value;
+            Assert.AreEqual(scopeName1, blobContainer.Data.DefaultEncryptionScope);
+            Assert.False(blobContainer.Data.DenyEncryptionScopeOverride.Value);
+
+            //Update container not support Encryption scope
+            BlobContainer blobContainer2 = (await _blobContainerCollection.CreateOrUpdateAsync(containerName, new BlobContainerData() { DefaultEncryptionScope = scopeName2, DenyEncryptionScopeOverride = true })).Value;
+            Assert.AreEqual(scopeName2, blobContainer2.Data.DefaultEncryptionScope);
+            Assert.True(blobContainer2.Data.DenyEncryptionScopeOverride.Value);
+        }
     }
 }
diff --git a/sdk/storage/Azure.ResourceManager.Storage/tests/Tests/StorageAccountTests.cs b/sdk/storage/Azure.ResourceManager.Storage/tests/Tests/StorageAccountTests.cs
@@ -39,6 +39,59 @@ public async Task ClearStorageAccounts()
             }
         }
 
+        [Test]
+        [RecordedTest]
+        public async Task StorageAccountGetOperations()
+        {
+            RestApiCollection operationCollection = Client.GetRestApis("Microsoft.Storage");
+            List<RestApi> apiList = await operationCollection.GetAllAsync().ToEnumerableAsync();
+            Assert.IsTrue(apiList.Count() > 1);
+        }
+
+        [Test]
+        [RecordedTest]
+        public async Task StorageAccountOperations()
+        {
+            RestApiCollection operationCollection = Client.GetRestApis("Microsoft.Storage");
+            List<RestApi> apiList = await operationCollection.GetAllAsync().ToEnumerableAsync();
+            bool exist1 = false;
+            bool exist2 = false;
+            foreach (RestApi restApi in apiList)
+            {
+                if (CheckRestApi(restApi, "Microsoft.Storage/storageAccounts/write", "Microsoft Storage", "Storage Accounts", "Create/Update Storage Account"))
+                {
+                    exist1 = true;
+                }
+                if (CheckRestApi(restApi, "Microsoft.Storage/storageAccounts/delete", "Microsoft Storage", "Storage Accounts", "Delete Storage Account"))
+                {
+                    exist2 = true;
+                }
+            }
+            Assert.IsTrue(exist1);
+            Assert.IsTrue(exist2);
+        }
+
+        public bool CheckRestApi(RestApi restApi, string name, string provider, string resource, string operation)
+        {
+            if (restApi.Name != name)
+            {
+                return false;
+            }
+            if (restApi.Operation != operation)
+            {
+                return false;
+            }
+            if (restApi.Provider != provider)
+            {
+                return false;
+            }
+            if (restApi.Resource != resource)
+            {
+                return false;
+            }
+            return true;
+        }
+
         [Test]
         [RecordedTest]
         public async Task ListSku()