5771 - GetGenericResourceOperations in ArmClient

sdk/resourcemanager/Azure.ResourceManager.Core/src/Extensions/ARMClientExtensions.cs

@@ -0,0 +1,51 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System;
+using System.Collections.Generic;
+
+namespace Azure.ResourceManager.Core
+{
+    /// <summary>
+    /// A class of extensions for the ARM Client.
+    /// </summary>
+    public static class ArmClientExtensions
+    {
+        /// <summary>
+        /// Get the operations for a list of specific resources.
+        /// </summary>
+        /// <param name="client"></param>
+        /// <param name="ids">A list of the IDs of the resources to retrieve.</param>
+        /// <returns></returns>
+        public static IList<GenericResourceOperations> GetGenericResourceOperations(this ArmClient client, IEnumerable<string> ids)
+        {
+            if (ids == null)
+            {
+                throw new ArgumentNullException(nameof(ids));
+            }
+
+            IList<GenericResourceOperations> genericRespirceOperations = new List<GenericResourceOperations>();
+            foreach (string id in ids)
+            {
+                genericRespirceOperations.Add(new GenericResourceOperations(client.DefaultSubscription, id));
+            }
+            return genericRespirceOperations;
+        }
+
+        /// <summary>
+        /// Get the operations for an specific resource.
+        /// </summary>
+        /// <param name="client"></param>
+        /// <param name="id">The ID of the resource to retrieve.</param>
+        /// <returns></returns>
+        public static GenericResourceOperations GetGenericResourceOperations(this ArmClient client, string id)
+        {
+            if (id == null)
+            {
+                throw new ArgumentNullException(nameof(id));
+            }
+
+            return new GenericResourceOperations(client.DefaultSubscription, id);
+        }
+    }
+}

sdk/resourcemanager/Azure.ResourceManager.Core/tests/Scenario/ArmClientTests.cs

@@ -0,0 +1,134 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using NUnit.Framework;
+using System.Threading.Tasks;
+using Azure.Core.TestFramework;
+
+namespace Azure.ResourceManager.Core.Tests.Scenario
+{
+    class ArmClientTests : ResourceManagerTestBase
+    {
+        private string _rgName;
+        private readonly string _location = "southcentralus";
+
+        public ArmClientTests(bool isAsync)
+            : base(isAsync)//, RecordedTestMode.Record)
+        {
+        }
+
+        [OneTimeSetUp]
+        public async Task LocalOneTimeSetup()
+        {
+            _rgName = SessionRecording.GenerateAssetName("testRg-");
+            var subscription = await GlobalClient.GetSubscriptions().TryGetAsync(SessionEnvironment.SubscriptionId);
+            _ = subscription.GetResourceGroups().Construct(_location).StartCreateOrUpdateAsync(_rgName).ConfigureAwait(false).GetAwaiter().GetResult().Value;
+            StopSessionRecording();
+        }
+
+        [TestCase]
+        public void GetGenericOperationsTests()
+        {
+            var ids = new List<string>()
+            {
+                $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/foo-1/",
+                $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/foo-2/",
+                $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/foo-3/",
+                $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/foo-4/"
+            };
+
+            var genericResourceOperationsList = Client.GetGenericResourceOperations(ids);
+
+            foreach(GenericResourceOperations operations in genericResourceOperationsList)
+            {
+                Assert.AreEqual(operations.Id, ids[0]);
+                ids.RemoveAt(0);
+            }
+        }
+
+        [TestCase]
+        public void GetGenericResourceOperationsSingleIDTests()
+        {
+            string id = $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/foo-1/";
+            Assert.AreEqual(Client.GetGenericResourceOperations(id).Id, id);
+        }
+
+        [TestCase]
+        [RecordedTest]
+        public async Task GetGenericResourceOperationsWithSingleValidResource()
+        {
+            string id = $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/{_rgName}/";
+            var genericResourceOperations = Client.GetGenericResourceOperations(id);
+            var genericResource = await genericResourceOperations.GetAsync();
+            Assert.AreEqual(200, genericResource.GetRawResponse().Status);
+        }
+
+        [TestCase]
+        [RecordedTest]
+        public void GetGenericResourceOperationsWithSingleInvalidResource()
+        {
+            string id = $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/foo-1/";
+            var genericResourceOperations = Client.GetGenericResourceOperations(id);
+            RequestFailedException exception = Assert.ThrowsAsync<RequestFailedException>(async () => await genericResourceOperations.GetAsync());
+            Assert.AreEqual(404, exception.Status);
+        }
+
+        [TestCase]
+        [RecordedTest]
+        public async Task GetGenericOperationsWithListOfValidResource()
+        {
+            var ids = new List<string>()
+            {
+                $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/{_rgName}/"
+            };
+
+            var genericResourceOperationsList = Client.GetGenericResourceOperations(ids);
+
+            foreach (GenericResourceOperations operations in genericResourceOperationsList)
+            {
+                var genericResource = await operations.GetAsync();
+                Assert.AreEqual(200, genericResource.GetRawResponse().Status);
+            }
+        }
+
+        [TestCase]
+        [RecordedTest]
+        public void GetGenericOperationsWithListOfInvalidResource()
+        {
+            var ids = new List<string>()
+            {
+                $"/subscriptions/{TestEnvironment.SubscriptionId}/resourceGroups/non-existent/"
+            };
+
+            var genericResourceOperationsList = Client.GetGenericResourceOperations(ids);
+
+            foreach (GenericResourceOperations operations in genericResourceOperationsList)
+            {
+                RequestFailedException exception = Assert.ThrowsAsync<RequestFailedException>(async () => await operations.GetAsync());
+                Assert.AreEqual(404, exception.Status);
+            }
+        }
+
+        [TestCase]
+        public void GetGenericResourceOperationWithNullSetOfIds()
+        {
+            string[] x = null;
+            Assert.Throws<ArgumentNullException>(() => { Client.GetGenericResourceOperations(x); });
+        }
+
+        [TestCase]
+        public void GetGenericResourceOperationWithNullId()
+        {
+                string x = null;
+                Assert.Throws<ArgumentNullException>(() => { Client.GetGenericResourceOperations(x); });
+        }
+
+        [TestCase]
+        public void GetGenericResourceOperationEmptyTest()
+        {
+            var ids = new List<string>();
+            Assert.AreEqual(new List<string>(), Client.GetGenericResourceOperations(ids));
+        }
+    }
+}