Automation tests (Azure#35545)

Author: mcgallan

sdk/automation/Azure.ResourceManager.Automation/tests/AutomationManagementTestBase.cs

@@ -13,6 +13,8 @@ namespace Azure.ResourceManager.Automation.Tests
     public class AutomationManagementTestBase : ManagementRecordedTestBase<AutomationManagementTestEnvironment>
     {
         protected ArmClient Client { get; private set; }
+        protected AzureLocation DefaultLocation => TestEnvironment.Location;
+        protected SubscriptionResource DefaultSubscription { get; private set; }
 
         protected AutomationManagementTestBase(bool isAsync, RecordedTestMode mode)
         : base(isAsync, mode)
@@ -25,17 +27,26 @@ protected AutomationManagementTestBase(bool isAsync)
         }
 
         [SetUp]
-        public void CreateCommonClient()
+        public async Task CreateCommonClient()
         {
             Client = GetArmClient();
+            DefaultSubscription = await Client.GetDefaultSubscriptionAsync().ConfigureAwait(false);
         }
 
-        protected async Task<ResourceGroupResource> CreateResourceGroup(SubscriptionResource subscription, string rgNamePrefix, AzureLocation location)
+        protected async Task<ResourceGroupResource> CreateResourceGroupAsync()
         {
-            string rgName = Recording.GenerateAssetName(rgNamePrefix);
-            ResourceGroupData input = new ResourceGroupData(location);
-            var lro = await subscription.GetResourceGroups().CreateOrUpdateAsync(WaitUntil.Completed, rgName, input);
-            return lro.Value;
+            var resourceGroupName = Recording.GenerateAssetName("testRG-");
+            var rgOp = await DefaultSubscription.GetResourceGroups().CreateOrUpdateAsync(
+                WaitUntil.Completed,
+                resourceGroupName,
+                new ResourceGroupData(DefaultLocation)
+                {
+                    Tags =
+                    {
+                        { "test", "env" }
+                    }
+                });
+            return rgOp.Value;
         }
     }
 }

sdk/automation/Azure.ResourceManager.Automation/tests/Helpers/ResourceDataHelpers.cs

@@ -0,0 +1,223 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System.Collections.Generic;
+using Azure.ResourceManager.Automation.Models;
+using Azure.ResourceManager.Models;
+using Azure.ResourceManager.Resources.Models;
+using NUnit.Framework;
+using Azure.Core;
+using System;
+using System.Text;
+using Azure.ResourceManager.Resources;
+using System.Threading.Tasks;
+using NUnit.Framework.Internal;
+using System.IO;
+
+namespace Azure.ResourceManager.Automation.Tests.Helpers
+{
+    public static class ResourceDataHelpers
+    {
+        public static IDictionary<string, string> ReplaceWith(this IDictionary<string, string> dest, IDictionary<string, string> src)
+        {
+            dest.Clear();
+            foreach (var kv in src)
+            {
+                dest.Add(kv);
+            }
+
+            return dest;
+        }
+
+        public static void AssertResource(ResourceData r1, ResourceData r2)
+        {
+            Assert.AreEqual(r1.Name, r2.Name);
+            Assert.AreEqual(r1.Id, r2.Id);
+            Assert.AreEqual(r1.ResourceType, r2.ResourceType);
+        }
+
+        #region Credential
+        public static void AssertCredential(AutomationCredentialData data1, AutomationCredentialData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.UserName, data2.UserName);
+            Assert.AreEqual(data1.Description, data2.Description);
+        }
+
+        public static AutomationCredentialCreateOrUpdateContent GetCredentialData(string name)
+        {
+            var data = new AutomationCredentialCreateOrUpdateContent(name, "userName1", "pwd1")
+            {
+                Description = "description of credential"
+            };
+            return data;
+        }
+        #endregion
+
+        #region DscConfigurationData
+        public static void AssertDscConfiguration(DscConfigurationData data1, DscConfigurationData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.Location, data2.Location);
+            Assert.AreEqual(data1.Description, data2.Description);
+            Assert.AreEqual(data1.State, data2.State);
+            Assert.AreEqual(data1.JobCount, data2.JobCount);
+        }
+
+        public static DscConfigurationCreateOrUpdateContent GetDscConfigurationData(string name)
+        {
+            var data = new DscConfigurationCreateOrUpdateContent(new AutomationContentSource()
+            {
+                Hash = new AutomationContentHash("sha256", "A9E5DB56BA21513F61E0B3868816FDC6D4DF5131F5617D7FF0D769674BD5072F"),
+                Value = "Configuration "+name+@" { Node SampleConfiguration.localhost { WindowsFeature IIS { Name = ""Web - Server""; Ensure = ""Present""; }}}",
+                SourceType = AutomationContentSourceType.EmbeddedContent,
+                Version = "1.0.0"
+            })
+            {
+                Description= "new sample configuration test",
+                Location = AzureLocation.EastUS,
+            };
+            return data;
+        }
+        #endregion
+
+        #region DscNodeConfigurationData
+        public static void AssertDscNodeConfiguration(DscNodeConfigurationData data1, DscNodeConfigurationData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.ConfigurationName, data2.ConfigurationName);
+            Assert.AreEqual(data1.Source, data2.Source);
+            Assert.AreEqual(data1.NodeCount, data2.NodeCount);
+        }
+
+        public static DscNodeConfigurationCreateOrUpdateContent GetDscNodeConfigurationData(string dscconfigurationName)
+        {
+            var data = new DscNodeConfigurationCreateOrUpdateContent()
+            {
+                Name = "SampleConfiguration.localhost",
+                Source = new AutomationContentSource()
+                {
+                    Hash = new AutomationContentHash("sha256", "6DE256A57F01BFA29B88696D5E77A383D6E61484C7686E8DB955FA10ACE9FFE5"),
+                    Value = @"instance of MSFT_RoleResource as $MSFT_RoleResource1ref { ResourceID = ""[WindowsFeature]IIS""; Ensure = ""Present""; SourceInfo = ""::3::32::WindowsFeature""; Name = ""Web-Server""; ModuleName = ""PsDesiredStateConfiguration""; ModuleVersion = ""1.0""; ConfigurationName = ""SampleConfiguration""; }; instance of OMI_ConfigurationDocument { Version=""2.0.0""; MinimumCompatibleVersion = ""1.0.0""; CompatibleVersionAdditionalProperties= {""Omi_BaseResource:ConfigurationName""}; Author=""vameru""; GenerationDate=""03/30/2017 13:40:25""; GenerationHost=""VAMERU-BACKEND""; Name=""SampleConfiguration""; };",
+                    SourceType = AutomationContentSourceType.EmbeddedContent,
+                    Version = "1.0"
+                },
+                ConfigurationName = "SampleConfiguration",
+                IsIncrementNodeConfigurationBuildRequired = false
+            };
+            return data;
+        }
+        #endregion
+
+        #region Runbook
+
+        public static AutomationRunbookCreateOrUpdateContent GetRunbookData()
+        {
+            var data = new AutomationRunbookCreateOrUpdateContent(AutomationRunbookType.Script)
+            {
+                Location = AzureLocation.EastUS
+            };
+            return data;
+        }
+
+        public static void AssertRunbook(AutomationRunbookData data1, AutomationRunbookData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.Location, data2.Location);
+            Assert.AreEqual(data1.State, data2.State);
+            Assert.AreEqual(data1.Description, data2.Description);
+            Assert.AreEqual(data1.JobCount, data2.JobCount);
+        }
+        #endregion
+
+        #region SourceControl
+        public static AutomationSourceControlCreateOrUpdateContent GetSourceControlData()
+        {
+            var data = new AutomationSourceControlCreateOrUpdateContent()
+            {
+                RepoUri = new Uri("https://dev.azure.com/vinkumar0563/_git/VinKumar-AzureAutomation"),
+                Branch = "sdktest",
+                FolderPath = "/Runbooks/PowershellScripts",
+                IsAutoSyncEnabled = false,
+                IsAutoPublishRunbookEnabled = true,
+                SourceType = SourceControlSourceType.VsoGit,
+                SecurityToken = new SourceControlSecurityTokenProperties()
+                {
+                    AccessToken = "stringfortoken",
+                    TokenType = SourceControlTokenType.PersonalAccessToken
+                },
+                Description = "test creating a Source Control",
+            };
+            return data;
+        }
+
+        public static void AssertSourceControl(AutomationSourceControlData data1, AutomationSourceControlData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.Description, data2.Description);
+            Assert.AreEqual(data1.Branch, data2.Branch);
+            Assert.AreEqual(data1.RepoUri, data2.RepoUri);
+            Assert.AreEqual(data1.FolderPath, data2.FolderPath);
+        }
+        #endregion
+
+        #region Variable
+        public static AutomationVariableCreateOrUpdateContent GetVariableData(string name)
+        {
+            var data = new AutomationVariableCreateOrUpdateContent(name)
+            {
+                Value = "10",
+                IsEncrypted = false,
+            };
+            return data;
+        }
+
+        public static void AssertVariable(AutomationVariableData data1, AutomationVariableData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.Value, data2.Value);
+            Assert.AreEqual(data1.IsEncrypted, data2.IsEncrypted);
+            Assert.AreEqual(data1.CreatedOn, data2.CreatedOn);
+            Assert.AreEqual(data1.LastModifiedOn, data2.LastModifiedOn);
+            Assert.AreEqual(data1.Description, data2.Description);
+        }
+        #endregion
+
+        #region Schedule
+        public static AutomationScheduleCreateOrUpdateContent GetScheduleData(string name)
+        {
+            var data = new AutomationScheduleCreateOrUpdateContent(name, DateTime.Today.AddDays(1), AutomationScheduleFrequency.Hour)
+            {
+                Interval = BinaryData.FromString("1"),
+            };
+            return data;
+        }
+
+        public static void AssertSchedule(AutomationScheduleData data1, AutomationScheduleData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.Frequency, data2.Frequency);
+            Assert.AreEqual(data1.Description, data2.Description);
+        }
+        #endregion
+
+        #region Account
+        public static AutomationAccountCreateOrUpdateContent GetAccountData()
+        {
+            var data = new AutomationAccountCreateOrUpdateContent()
+            {
+                Sku = new AutomationSku("Basic"),
+                Location = AzureLocation.EastUS
+            };
+            return data;
+        }
+
+        public static void AssertAccount(AutomationAccountData data1, AutomationAccountData data2)
+        {
+            AssertResource(data1, data2);
+            Assert.AreEqual(data1.Location, data2.Location);
+            Assert.AreEqual(data1.Description, data2.Description);
+        }
+        #endregion
+    }
+}