Skip to content

Commit 7750635

Browse files
author
Maddie Clayton
committed
add more compute cmdlets and tests
1 parent a19b781 commit 7750635

File tree

7 files changed

+17
-8
lines changed

7 files changed

+17
-8
lines changed

src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/ImageTests.ps1

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,9 @@ function Test-Image
103103
$imageConfig = Remove-AzureRmImageDataDisk -Image $imageConfig -Lun 3;
104104
Assert-AreEqual 2 $imageConfig.StorageProfile.DataDisks.Count;
105105

106-
$createdImage = New-AzureRmImage -Image $imageConfig -ImageName $imageName -ResourceGroupName $rgname
106+
$job = New-AzureRmImage -Image $imageConfig -ImageName $imageName -ResourceGroupName $rgname -AsJob
107+
$job | Wait-Job
108+
$createdImage = $job | Receive-Job
107109

108110
# Verify Image properties
109111
Assert-NotNull $createdImage.Id;

src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/VirtualMachineRunCommandTests.ps1

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -109,8 +109,6 @@ function Test-VirtualMachineSetRunCommand
109109

110110
$job = Invoke-AzureRmVMRunCommand -ResourceGroupName $rgname -Name $vmname -CommandId $commandId -ScriptPath $path -Parameter $param -AsJob
111111
$job | Wait-Job
112-
$result = $job | Receive-Job
113-
$result_output = $result | Out-String;
114112

115113
# Remove All VMs
116114
Get-AzureRmVM -ResourceGroupName $rgname | Remove-AzureRmVM -Force;

src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/VirtualMachineScaleSetTests.ps1

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,9 @@ function Test-VirtualMachineScaleSet-Common($IsManaged)
266266
{
267267
for ($j = 0; $j -lt 2; $j++)
268268
{
269-
$st = Set-AzureRmVmssVM -ReimageAll -ResourceGroupName $rgname -VMScaleSetName $vmssName -InstanceId $j
269+
$job = Set-AzureRmVmssVM -ReimageAll -ResourceGroupName $rgname -VMScaleSetName $vmssName -InstanceId $j -AsJob
270+
$job | Wait-Job
271+
$st = $job | Receive-Job
270272
}
271273
}
272274

src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/VirtualMachineTests.ps1

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2987,7 +2987,8 @@ function Test-VirtualMachineManagedDiskConversion
29872987
Stop-AzureRmVM -ResourceGroupName $rgname -Name $vmname -Force
29882988

29892989
# Convert VM to managed disks
2990-
ConvertTo-AzureRmVMManagedDisk -ResourceGroupName $rgname -VMName $vmname
2990+
$job = ConvertTo-AzureRmVMManagedDisk -ResourceGroupName $rgname -VMName $vmname -AsJob
2991+
$job | Wait-Job
29912992

29922993
$vm2 = Get-AzureRmVM -Name $vmname -ResourceGroupName $rgname;
29932994

src/ResourceManager/Compute/Commands.Compute/Generated/Image/ImageCreateOrUpdateMethod.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ protected PSArgument[] CreateImageCreateOrUpdateParameters()
116116
[OutputType(typeof(PSImage))]
117117
public partial class NewAzureRmImage : ComputeAutomationBaseCmdlet
118118
{
119-
protected override void ProcessRecord()
119+
public override void ExecuteCmdlet()
120120
{
121121
ExecuteClientAction(() =>
122122
{

src/ResourceManager/Compute/Commands.Compute/Generated/VirtualMachine/VirtualMachineConvertToManagedDisksMethod.cs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ protected PSArgument[] CreateVirtualMachineConvertToManagedDisksParameters()
101101
[OutputType(typeof(PSOperationStatusResponse))]
102102
public partial class ConvertToAzureRmVMManagedDisk : ComputeAutomationBaseCmdlet
103103
{
104-
protected override void ProcessRecord()
104+
public override void ExecuteCmdlet()
105105
{
106106
ExecuteClientAction(() =>
107107
{
@@ -137,5 +137,8 @@ protected override void ProcessRecord()
137137
[Alias("Name")]
138138
[AllowNull]
139139
public string VMName { get; set; }
140+
141+
[Parameter(Mandatory = false, HelpMessage = "Run cmdlet in the background")]
142+
public SwitchParameter AsJob { get; set; }
140143
}
141144
}

src/ResourceManager/Compute/Commands.Compute/Generated/VirtualMachineScaleSetVM/VirtualMachineScaleSetVMReimageMethod.cs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ protected PSArgument[] CreateVirtualMachineScaleSetVMReimageParameters()
115115
[OutputType(typeof(PSOperationStatusResponse))]
116116
public partial class SetAzureRmVmssVM : ComputeAutomationBaseCmdlet
117117
{
118-
protected override void ProcessRecord()
118+
public override void ExecuteCmdlet()
119119
{
120120
ExecuteClientAction(() =>
121121
{
@@ -202,5 +202,8 @@ protected override void ProcessRecord()
202202
Mandatory = true)]
203203
[AllowNull]
204204
public SwitchParameter ReimageAll { get; set; }
205+
206+
[Parameter(Mandatory = false, HelpMessage = "Run cmdlet in the background")]
207+
public SwitchParameter AsJob { get; set; }
205208
}
206209
}

0 commit comments

Comments
 (0)