|
| 1 | +using System; |
| 2 | +using System.Collections.Generic; |
| 3 | +using System.Linq; |
| 4 | +using Microsoft.WindowsAzure.Commands.ServiceManagement.IaaS.Extensions; |
| 5 | +using Microsoft.WindowsAzure.Commands.ServiceManagement.Model; |
| 6 | +using Xunit; |
| 7 | + |
| 8 | +namespace Microsoft.WindowsAzure.Commands.ServiceManagement.Extensions.Test.DSC.UnitTests |
| 9 | +{ |
| 10 | + using NSM = Management.Compute.Models; |
| 11 | + |
| 12 | + public class GetAzureVmDscExtensionStatusUnitTest |
| 13 | + { |
| 14 | + private readonly GetAzureVmDscExtensionStatusCommand getAzureVmDscExtensionStatusCmdlet; |
| 15 | + private const string ServiceName = "dsc-service"; |
| 16 | + |
| 17 | + public GetAzureVmDscExtensionStatusUnitTest() |
| 18 | + { |
| 19 | + getAzureVmDscExtensionStatusCmdlet = new GetAzureVmDscExtensionStatusCommand(); |
| 20 | + } |
| 21 | + |
| 22 | + [Fact] |
| 23 | + public void TestGetService() |
| 24 | + { |
| 25 | + //when service name is passed as argument in the cmdlet |
| 26 | + getAzureVmDscExtensionStatusCmdlet.GetService(ServiceName, null); |
| 27 | + Assert.Equal(ServiceName, getAzureVmDscExtensionStatusCmdlet.Service); |
| 28 | + |
| 29 | + //when vm object is passed as argument in the cmdlet |
| 30 | + getAzureVmDscExtensionStatusCmdlet.GetService("", GetAzureVM(ServiceName, ServiceName)); |
| 31 | + Assert.Equal(ServiceName, getAzureVmDscExtensionStatusCmdlet.Service); |
| 32 | + } |
| 33 | + |
| 34 | + [Fact] |
| 35 | + public void TestGetVirtualMachineDscStatusContextListWithServiceName() |
| 36 | + { |
| 37 | + getAzureVmDscExtensionStatusCmdlet.Service = ServiceName; |
| 38 | + |
| 39 | + // service has multiple vm's |
| 40 | + var roles = new List<NSM.Role> {CreateRole("dscmachine01"), CreateRole("dscmachine02")}; |
| 41 | + var roleInstances = new List<NSM.RoleInstance> |
| 42 | + { |
| 43 | + CreateRoleInstance("dscmachine01"), |
| 44 | + CreateRoleInstance("dscmachine02") |
| 45 | + }; |
| 46 | + |
| 47 | + var deploymentResponse = CreateDeploymentGetResponse(ServiceName, roles, roleInstances); |
| 48 | + var dscExtensionStatusContexts = |
| 49 | + getAzureVmDscExtensionStatusCmdlet |
| 50 | + .GetVirtualMachineDscStatusContextList(deploymentResponse); |
| 51 | + Assert.Null(getAzureVmDscExtensionStatusCmdlet.Name); |
| 52 | + Assert.Null(getAzureVmDscExtensionStatusCmdlet.VmName); |
| 53 | + Assert.NotNull(dscExtensionStatusContexts); |
| 54 | + Assert.Equal(dscExtensionStatusContexts.Count, 2); |
| 55 | + |
| 56 | + } |
| 57 | + |
| 58 | + [Fact] |
| 59 | + public void TestGetVirtualMachineDscStatusContextListWithServiceNameAndVmName() |
| 60 | + { |
| 61 | + getAzureVmDscExtensionStatusCmdlet.Service = ServiceName; |
| 62 | + getAzureVmDscExtensionStatusCmdlet.Name = "dscmachine01"; |
| 63 | + |
| 64 | + // service has multiple vm's |
| 65 | + var roles = new List<NSM.Role> {CreateRole("dscmachine01"), CreateRole("dscmachine02")}; |
| 66 | + var roleInstances = new List<NSM.RoleInstance> |
| 67 | + { |
| 68 | + CreateRoleInstance("dscmachine01"), |
| 69 | + CreateRoleInstance("dscmachine02") |
| 70 | + }; |
| 71 | + |
| 72 | + var deploymentResponse = CreateDeploymentGetResponse(ServiceName, roles, roleInstances); |
| 73 | + var dscExtensionStatusContexts = |
| 74 | + getAzureVmDscExtensionStatusCmdlet |
| 75 | + .GetVirtualMachineDscStatusContextList(deploymentResponse); |
| 76 | + Assert.NotNull(getAzureVmDscExtensionStatusCmdlet.Name); |
| 77 | + Assert.Null(getAzureVmDscExtensionStatusCmdlet.VmName); |
| 78 | + Assert.NotNull(dscExtensionStatusContexts); |
| 79 | + Assert.Equal(dscExtensionStatusContexts.Count, 1); |
| 80 | + |
| 81 | + } |
| 82 | + |
| 83 | + [Fact] |
| 84 | + public void TestGetVirtualMachineDscStatusContextListWithServiceNameAndIncorrectVmName() |
| 85 | + { |
| 86 | + getAzureVmDscExtensionStatusCmdlet.Service = ServiceName; |
| 87 | + getAzureVmDscExtensionStatusCmdlet.Name = "some-blah"; |
| 88 | + |
| 89 | + // service has multiple vm's |
| 90 | + var roles = new List<NSM.Role> {CreateRole("dscmachine01"), CreateRole("dscmachine02")}; |
| 91 | + var roleInstances = new List<NSM.RoleInstance> |
| 92 | + { |
| 93 | + CreateRoleInstance("dscmachine01"), |
| 94 | + CreateRoleInstance("dscmachine02") |
| 95 | + }; |
| 96 | + |
| 97 | + var deploymentResponse = CreateDeploymentGetResponse(ServiceName, roles, roleInstances); |
| 98 | + var dscExtensionStatusContexts = |
| 99 | + getAzureVmDscExtensionStatusCmdlet |
| 100 | + .GetVirtualMachineDscStatusContextList(deploymentResponse); |
| 101 | + Assert.NotNull(getAzureVmDscExtensionStatusCmdlet.Name); |
| 102 | + Assert.Null(getAzureVmDscExtensionStatusCmdlet.VmName); |
| 103 | + Assert.Equal(dscExtensionStatusContexts.Count, 0); |
| 104 | + |
| 105 | + } |
| 106 | + |
| 107 | + [Fact] |
| 108 | + public void TestGetVirtualMachineDscStatusContextListWithVm() |
| 109 | + { |
| 110 | + getAzureVmDscExtensionStatusCmdlet.Service = ServiceName; |
| 111 | + getAzureVmDscExtensionStatusCmdlet.VmName = "dscmachine02"; |
| 112 | + |
| 113 | + // service has multiple vm's |
| 114 | + var roles = new List<NSM.Role> {CreateRole("dscmachine02")}; |
| 115 | + var roleInstances = new List<NSM.RoleInstance> {CreateRoleInstance("dscmachine02")}; |
| 116 | + |
| 117 | + var deploymentResponse = CreateDeploymentGetResponse(ServiceName, roles, roleInstances); |
| 118 | + var dscExtensionStatusContexts = |
| 119 | + getAzureVmDscExtensionStatusCmdlet |
| 120 | + .GetVirtualMachineDscStatusContextList(deploymentResponse); |
| 121 | + Assert.Null(getAzureVmDscExtensionStatusCmdlet.Name); |
| 122 | + Assert.NotNull(getAzureVmDscExtensionStatusCmdlet.VmName); |
| 123 | + Assert.Equal(dscExtensionStatusContexts.Count, 1); |
| 124 | + } |
| 125 | + |
| 126 | + [Fact] |
| 127 | + public void TestCreateDscStatusContext() |
| 128 | + { |
| 129 | + getAzureVmDscExtensionStatusCmdlet.Service = ServiceName; |
| 130 | + |
| 131 | + var roles = new List<NSM.Role> {CreateRole("dscmachine02")}; |
| 132 | + var roleInstances = new List<NSM.RoleInstance> {CreateRoleInstance("dscmachine02")}; |
| 133 | + var context = |
| 134 | + getAzureVmDscExtensionStatusCmdlet.CreateDscStatusContext( |
| 135 | + roles[0], roleInstances[0]); |
| 136 | + Assert.NotNull(context); |
| 137 | + Assert.Equal(context.Name, "dscmachine02"); |
| 138 | + Assert.Equal(context.StatusCode, 1); |
| 139 | + Assert.Equal(context.ServiceName, ServiceName); |
| 140 | + Assert.Equal(context.Status, "Success"); |
| 141 | + Assert.Equal(context.StatusMessage, "Dsc Configuration was applied successful"); |
| 142 | + Assert.Equal(context.DscConfigurationLog.Count(), GetFormattedMessage().Count()); |
| 143 | + } |
| 144 | + |
| 145 | + private IPersistentVM GetAzureVM(String roleName, String serviceName) |
| 146 | + { |
| 147 | + var vm = new PersistentVM {RoleName = roleName}; |
| 148 | + var vmContext = new PersistentVMRoleContext |
| 149 | + { |
| 150 | + DeploymentName = roleName, |
| 151 | + Name = roleName, |
| 152 | + ServiceName = serviceName, |
| 153 | + VM = vm |
| 154 | + }; |
| 155 | + |
| 156 | + return vmContext; |
| 157 | + } |
| 158 | + |
| 159 | + private NSM.DeploymentGetResponse CreateDeploymentGetResponse(string serviceName, IList<NSM.Role> roles, |
| 160 | + IList<NSM.RoleInstance> roleInstances) |
| 161 | + { |
| 162 | + var response = new NSM.DeploymentGetResponse |
| 163 | + { |
| 164 | + Name = serviceName, |
| 165 | + Configuration = "config", |
| 166 | + Status = Management.Compute.Models.DeploymentStatus.Starting, |
| 167 | + PersistentVMDowntime = new NSM.PersistentVMDowntime |
| 168 | + { |
| 169 | + EndTime = DateTime.Now, |
| 170 | + StartTime = DateTime.Now, |
| 171 | + Status = "", |
| 172 | + }, |
| 173 | + LastModifiedTime = DateTime.Now, |
| 174 | + Roles = roles, |
| 175 | + RoleInstances = roleInstances |
| 176 | + }; |
| 177 | + |
| 178 | + return response; |
| 179 | + } |
| 180 | + |
| 181 | + private NSM.RoleInstance CreateRoleInstance(String roleName) |
| 182 | + { |
| 183 | + var roleInstance = new NSM.RoleInstance |
| 184 | + { |
| 185 | + RoleName = roleName, |
| 186 | + ResourceExtensionStatusList = CreateResourceExtensionStatus() |
| 187 | + }; |
| 188 | + return roleInstance; |
| 189 | + } |
| 190 | + |
| 191 | + private NSM.Role CreateRole(String roleName) |
| 192 | + { |
| 193 | + var role = new NSM.Role |
| 194 | + { |
| 195 | + RoleName = roleName |
| 196 | + }; |
| 197 | + |
| 198 | + return role; |
| 199 | + } |
| 200 | + |
| 201 | + private List<NSM.ResourceExtensionStatus> CreateResourceExtensionStatus() |
| 202 | + { |
| 203 | + var resourceExtensionStatusList = new List<NSM.ResourceExtensionStatus>(); |
| 204 | + |
| 205 | + var resourceBgiExtensionStatus = new NSM.ResourceExtensionStatus |
| 206 | + { |
| 207 | + HandlerName = "BGIInfo" |
| 208 | + }; |
| 209 | + var resourceDscExtensionStatus = new NSM.ResourceExtensionStatus |
| 210 | + { |
| 211 | + HandlerName = "Microsoft.Powershell.DSC", |
| 212 | + ExtensionSettingStatus = CreateExtensionSettingStatus() |
| 213 | + }; |
| 214 | + |
| 215 | + resourceExtensionStatusList.Add(resourceBgiExtensionStatus); |
| 216 | + resourceExtensionStatusList.Add(resourceDscExtensionStatus); |
| 217 | + |
| 218 | + return resourceExtensionStatusList; |
| 219 | + } |
| 220 | + |
| 221 | + private NSM.ResourceExtensionConfigurationStatus CreateExtensionSettingStatus() |
| 222 | + { |
| 223 | + var extensionSettingStatus = new NSM.ResourceExtensionConfigurationStatus |
| 224 | + { |
| 225 | + Code = 1, |
| 226 | + Status = "Success", |
| 227 | + FormattedMessage = new NSM.GuestAgentFormattedMessage |
| 228 | + { |
| 229 | + Message = "Dsc Configuration was applied successful" |
| 230 | + }, |
| 231 | + Timestamp = new DateTime(), |
| 232 | + SubStatusList = CreateResourceExtensionSubStatus(1, CreateGuestAgentFormattedMessage()) |
| 233 | + }; |
| 234 | + |
| 235 | + return extensionSettingStatus; |
| 236 | + } |
| 237 | + |
| 238 | + private List<NSM.ResourceExtensionSubStatus> CreateResourceExtensionSubStatus(int code, |
| 239 | + NSM.GuestAgentFormattedMessage formattedMessage) |
| 240 | + { |
| 241 | + var resourceExtensionSubStatusList = new List<NSM.ResourceExtensionSubStatus>(); |
| 242 | + var resourceExtensionSubStatus = new NSM.ResourceExtensionSubStatus |
| 243 | + { |
| 244 | + Code = code, |
| 245 | + FormattedMessage = formattedMessage, |
| 246 | + Status = "Success", |
| 247 | + Name = "DSC Status" |
| 248 | + }; |
| 249 | + resourceExtensionSubStatusList.Add(resourceExtensionSubStatus); |
| 250 | + return resourceExtensionSubStatusList; |
| 251 | + } |
| 252 | + |
| 253 | + private NSM.GuestAgentFormattedMessage CreateGuestAgentFormattedMessage() |
| 254 | + { |
| 255 | + var formattedMessage = new NSM.GuestAgentFormattedMessage |
| 256 | + { |
| 257 | + Message = |
| 258 | + "[ESPARMAR-2012R2]:LCM:[Start Set]\r\n[ESPARMAR-2012R2]:LCM:[Start Resource] " + |
| 259 | + "[[WindowsFeature]IIS]\r\n[ESPARMAR-2012R2]:LCM:[Start Test] [[WindowsFeature]IIS]\r\n[ESPARMAR-2012R2]" |
| 260 | + }; |
| 261 | + return formattedMessage; |
| 262 | + } |
| 263 | + |
| 264 | + private IEnumerable<string> GetFormattedMessage() |
| 265 | + { |
| 266 | + const string message = "[ESPARMAR-2012R2]:LCM:[Start Set]\r\n[ESPARMAR-2012R2]:LCM:[Start Resource] " + |
| 267 | + "[[WindowsFeature]IIS]\r\n[ESPARMAR-2012R2]:LCM:[Start Test] [[WindowsFeature]IIS]\r\n[ESPARMAR-2012R2]"; |
| 268 | + |
| 269 | + return message.Split(new[] {"\r\n", "\n"}, StringSplitOptions.None); |
| 270 | + } |
| 271 | + } |
| 272 | +} |
0 commit comments