Fixed bug

Author: grizzlytheodore

src/Compute/Compute.Test/ScenarioTests/ComputeTestCommon.ps1

@@ -791,5 +791,8 @@ function Get-AvailableSku
 
     $res = get-azcomputeresourcesku $location | where-object ResourceType -match $resourceType
     $res = $res.where({$_.restrictions.count -eq 0})
+    if ($resourceType -match "virtualmachine"){
+        $res = $res.where({$_.Name -notmatch "Standard_B"})
+    }
     return $res[0].Name
 }

src/Compute/Compute.Test/ScenarioTests/VirtualMachinePatchTests.ps1

@@ -30,21 +30,42 @@ function Test-InvokeAzVmAssessPatch
         New-AzResourceGroup -Name $rgname -Location $loc -Force;
 
         # VM Profile & Hardware
-        $vmsize = 'Standard_DS2_v2';
+        $vmsize = Get-AvailableSku $loc "virtualMachine"
         $vmname = 'vm' + $rgname;
 
+        $p = New-AzVMConfig -VMName $vmname -VMSize $vmsize -Priority 'Low' -MaxPrice 0.1538;
+
+        # NRP
+        $subnet = New-AzVirtualNetworkSubnetConfig -Name ('subnet' + $rgname) -AddressPrefix "10.0.0.0/24";
+        $vnet = New-AzVirtualNetwork -Force -Name ('vnet' + $rgname) -ResourceGroupName $rgname -Location $loc -AddressPrefix "10.0.0.0/16" -Subnet $subnet;
+        $vnet = Get-AzVirtualNetwork -Name ('vnet' + $rgname) -ResourceGroupName $rgname;
+        $subnetId = $vnet.Subnets[0].Id;
+        $nic = New-AzNetworkInterface -Force -Name ('nic' + $rgname) -ResourceGroupName $rgname -Location $loc -SubnetId $subnetId
+        $nic = Get-AzNetworkInterface -Name ('nic' + $rgname) -ResourceGroupName $rgname;
+        $nicId = $nic.Id;
+
+        $p = Add-AzVMNetworkInterface -VM $p -Id $nicId;
+
         # OS & Image
-        $user = "Foo12";
+        $user = "Foo2";
         $password = $PLACEHOLDER;
         $securePassword = ConvertTo-SecureString $password -AsPlainText -Force;
         $cred = New-Object System.Management.Automation.PSCredential ($user, $securePassword);
+        $computerName = 'test';
+
+        $p = Set-AzVMOperatingSystem -VM $p -Windows -ComputerName $computerName -Credential $cred;
+
+        $imgRef = Get-DefaultCRPImage -loc $loc;
+        $p = ($imgRef | Set-AzVMSourceImage -VM $p);
+
+        # Create a Virtual Machine
+        New-AzVM -ResourceGroupName $rgname -Location $loc -VM $p;
 
-        New-AzVm -resourcegroupname $rgname -location $loc -name $vmName -credential $cred -size $vmsize
         $patchResult = invoke-azvmassesspatch -resourcegroupname $rgname -vmname $vmname
 
-        Assert.NotNull($patchResult);
-        Assert.Equal("Succeeded", $patchResult.Status);
-        Assert.NotNull($patchResult.StartDateTime);
+        Assert-NotNull $patchResult;
+        Assert-AreEqual "Succeeded" $patchResult.Status;
+        Assert-NotNull $patchResult.StartDateTime;
 
     }
     finally

src/Compute/Compute.Test/SessionRecords/Microsoft.Azure.Commands.Compute.Test.ScenarioTests.VirtualMachinePatchTests/TestInvokeAzVmAssessPatch.json

@@ -103,16 +103,7 @@ public override void ExecuteCmdlet()
                     }
 
 
-                    //extension call instead??
-                    //var result = VirtualMachinesClient.AssessPatches(resourceGroupName, vmName);
-                    //var result = VirtualMachinesClient.AssessPatchesWithHttpMessagesAsync(resourceGroupName, vmName).GetAwaiter().GetResult();
-                    //var result = VirtualMachinesClient.AssessPatchesWithHttpMessagesAsync(resourceGroupName, vmName);
                     var result = VirtualMachinesClient.AssessPatches(resourceGroupName, vmName);
-
-                    //var psObject = new PSRunCommandResult();
-                    //var psObject = new PSAssessPatchResult();
-                    //var psObject = ComputeAutoMapperProfile.Mapper.Map<PSAzureOperationResponse>(result);
-                    //ComputeAutomationAutoMapperProfile.Mapper.Map<VirtualMachineAssessPatchesResult, PSAzureOperationResponse>(result, psObject);
                     var psObject = new PSVirtualMachineAssessPatchesResult();
                     ComputeAutomationAutoMapperProfile.Mapper.Map<VirtualMachineAssessPatchesResult, PSVirtualMachineAssessPatchesResult>(result, psObject);
                     WriteObject(psObject);

src/Compute/Compute.Test/SessionRecords/Microsoft.Azure.Commands.Compute.Test.SecenarioTests.VirtualMachinePatchTests/TestInvokeAzVmAssessPatch.json

@@ -23,13 +23,13 @@ namespace Microsoft.Azure.Commands.Compute.Automation.Models
 {
     public class PSVirtualMachineAssessPatchesResult
     {
-        public string Status { get; }
-        public string AssessmentActivityId { get; }
-        public bool? RebootPending { get; }
-        public int? CriticalAndSecurityPatchCount { get; }
-        public int? OtherPatchCount { get; }
-        public DateTime? StartDateTime { get; }
-        public IList<VirtualMachineSoftwarePatchProperties> Patches { get; }
-        public ApiError Error { get; }
+        public string Status { get; set; }
+        public string AssessmentActivityId { get; set; }
+        public bool? RebootPending { get; set; }
+        public int? CriticalAndSecurityPatchCount { get; set; }
+        public int? OtherPatchCount { get; set; }
+        public DateTime? StartDateTime { get; set; }
+        public IList<VirtualMachineSoftwarePatchProperties> Patches { get; set; }
+        public ApiError Error { get; set; }
     }
 }