Skip to content

Commit 1ab8165

Browse files
huangpfhuangpf
committed
update
1 parent 3e3a508 commit 1ab8165

File tree

4 files changed

+71
-88
lines changed

4 files changed

+71
-88
lines changed

‎src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/VirtualMachineExtensionTests.ps1

Lines changed: 15 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -95,13 +95,12 @@ function Test-VirtualMachineExtension
9595

9696
$p.StorageProfile.OSDisk = $null;
9797
$p = Set-AzureVMOperatingSystem -VM $p -Windows -ComputerName $computerName -Credential $cred -ProvisionVMAgent;
98-
$p = Set-AzureVMSourceImage -VM $p -Name $img <#-DestinationVhdsContainer $vhdContainer#>;
98+
$p = Set-AzureVMSourceImage -VM $p -Name $img;
9999

100100
Assert-AreEqual $p.OSProfile.AdminUsername $user;
101101
Assert-AreEqual $p.OSProfile.ComputerName $computerName;
102102
Assert-AreEqual $p.OSProfile.AdminPassword $password;
103103
Assert-AreEqual $p.OSProfile.WindowsConfiguration.ProvisionVMAgent $true;
104-
# Assert-AreEqual $p.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
105104
Assert-AreEqual $p.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
106105

107106
# Virtual Machine
@@ -169,7 +168,6 @@ function Test-VirtualMachineExtension
169168
Assert-AreEqual $vm1.Name $vmname;
170169
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces.Count 1;
171170
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces[0].ReferenceUri $nicId;
172-
# Assert-AreEqual $vm1.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
173171
Assert-AreEqual $vm1.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
174172
Assert-AreEqual $vm1.OSProfile.AdminUsername $user;
175173
Assert-AreEqual $vm1.OSProfile.ComputerName $computerName;
@@ -275,13 +273,12 @@ function Test-VirtualMachineCustomScriptExtension
275273

276274
$p.StorageProfile.OSDisk = $null;
277275
$p = Set-AzureVMOperatingSystem -VM $p -Windows -ComputerName $computerName -Credential $cred -ProvisionVMAgent;
278-
$p = Set-AzureVMSourceImage -VM $p -Name $img #-DestinationVhdsContainer $vhdContainer;
276+
$p = Set-AzureVMSourceImage -VM $p -Name $img;
279277

280278
Assert-AreEqual $p.OSProfile.AdminUsername $user;
281279
Assert-AreEqual $p.OSProfile.ComputerName $computerName;
282280
Assert-AreEqual $p.OSProfile.AdminPassword $password;
283281
Assert-AreEqual $p.OSProfile.WindowsConfiguration.ProvisionVMAgent $true;
284-
#Assert-AreEqual $p.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
285282
Assert-AreEqual $p.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
286283

287284
# Virtual Machine
@@ -291,12 +288,12 @@ function Test-VirtualMachineCustomScriptExtension
291288
# Virtual Machine Extension
292289
$extname = 'csetest';
293290
$extver = '1.1';
294-
$publisher = 'Microsoft.Compute';
291+
$publisher = 'Microsoft.Compute';
295292
$exttype = 'CustomScriptExtension';
296-
$fileToExecute = "a.exe";
297-
$containderName = "script"
293+
$fileToExecute = "a.exe";
294+
$containderName = "script"
298295

299-
# Set custom script extension
296+
# Set custom script extension
300297
Set-AzureVMCustomScriptExtension -ResourceGroupName $rgname -Location $loc -VMName $vmname -Name $extname -TypeHandlerVersion $extver -StorageAccountName $stoname -StorageAccountKey $stokey -FileName $fileToExecute -ContainerName $containderName;
301298

302299
# Get VM Extension
@@ -306,8 +303,8 @@ function Test-VirtualMachineCustomScriptExtension
306303
Assert-AreEqual $ext.Publisher $publisher;
307304
Assert-AreEqual $ext.ExtensionType $exttype;
308305
Assert-AreEqual $ext.TypeHandlerVersion $extver;
309-
Assert-AreEqual $ext.CommandToExecute "powershell -ExecutionPolicy Unrestricted -file " + $fileToExecute + " ";
310-
Assert-True $ext.Uri[0].Contains($stoname + ".blob.core.windows.net/" + $containderName + "/" + $fileToExecute);
306+
Assert-AreEqual $ext.CommandToExecute "powershell -ExecutionPolicy Unrestricted -file " + $fileToExecute + " ";
307+
Assert-True $ext.Uri[0].Contains($stoname + ".blob.core.windows.net/" + $containderName + "/" + $fileToExecute);
311308
Assert-NotNull $ext.ProvisioningState;
312309

313310
$ext = Get-AzureVMCustomScriptExtension -ResourceGroupName $rgname -VMName $vmname -Name $extname -Status;
@@ -316,8 +313,8 @@ function Test-VirtualMachineCustomScriptExtension
316313
Assert-AreEqual $ext.Publisher $publisher;
317314
Assert-AreEqual $ext.ExtensionType $exttype;
318315
Assert-AreEqual $ext.TypeHandlerVersion $extver;
319-
Assert-AreEqual $ext.CommandToExecute "powershell -ExecutionPolicy Unrestricted -file " + $fileToExecute + " ";
320-
Assert-True $ext.Uri[0].Contains($stoname + ".blob.core.windows.net/" + $containderName + "/" + $fileToExecute);
316+
Assert-AreEqual $ext.CommandToExecute "powershell -ExecutionPolicy Unrestricted -file " + $fileToExecute + " ";
317+
Assert-True $ext.Uri[0].Contains($stoname + ".blob.core.windows.net/" + $containderName + "/" + $fileToExecute);
321318
Assert-NotNull $ext.ProvisioningState;
322319
Assert-NotNull $ext.Statuses;
323320

@@ -326,7 +323,6 @@ function Test-VirtualMachineCustomScriptExtension
326323
Assert-AreEqual $vm1.Name $vmname;
327324
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces.Count 1;
328325
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces[0].ReferenceUri $nicId;
329-
# Assert-AreEqual $vm1.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
330326
Assert-AreEqual $vm1.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
331327
Assert-AreEqual $vm1.OSProfile.AdminUsername $user;
332328
Assert-AreEqual $vm1.OSProfile.ComputerName $computerName;
@@ -434,13 +430,12 @@ function Test-VirtualMachineAccessExtension
434430

435431
$p.StorageProfile.OSDisk = $null;
436432
$p = Set-AzureVMOperatingSystem -VM $p -Windows -ComputerName $computerName -Credential $cred -ProvisionVMAgent;
437-
$p = Set-AzureVMSourceImage -VM $p -Name $img #-DestinationVhdsContainer $vhdContainer;
433+
$p = Set-AzureVMSourceImage -VM $p -Name $img;
438434

439435
Assert-AreEqual $p.OSProfile.AdminUsername $user;
440436
Assert-AreEqual $p.OSProfile.ComputerName $computerName;
441437
Assert-AreEqual $p.OSProfile.AdminPassword $password;
442438
Assert-AreEqual $p.OSProfile.WindowsConfiguration.ProvisionVMAgent $true;
443-
#Assert-AreEqual $p.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
444439
Assert-AreEqual $p.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
445440

446441
# Virtual Machine
@@ -450,13 +445,13 @@ function Test-VirtualMachineAccessExtension
450445
# Virtual Machine Extension
451446
$extname = 'csetest';
452447
$extver = '2.0';
453-
$user2 = "Bar12";
448+
$user2 = "Bar12";
454449
$password2 = 'FoO@123' + $rgname;
455450

456-
# Set custom script extension
451+
# Set custom script extension
457452
Set-AzureVMAccessExtension -ResourceGroupName $rgname -Location $loc -VMName $vmname -Name $extname -TypeHandlerVersion $extver -UserName $user2 -Password $password2;
458453

459-
$publisher = 'Microsoft.Compute';
454+
$publisher = 'Microsoft.Compute';
460455
$exttype = 'VMAccessAgent';
461456

462457
# Get VM Extension
@@ -466,7 +461,7 @@ function Test-VirtualMachineAccessExtension
466461
Assert-AreEqual $ext.Publisher $publisher;
467462
Assert-AreEqual $ext.ExtensionType $exttype;
468463
Assert-AreEqual $ext.TypeHandlerVersion $extver;
469-
Assert-AreEqual $ext.UserName $user2;
464+
Assert-AreEqual $ext.UserName $user2;
470465
Assert-NotNull $ext.ProvisioningState;
471466

472467
$ext = Get-AzureVMAccessExtension -ResourceGroupName $rgname -VMName $vmname -Name $extname -Status;
@@ -483,7 +478,6 @@ function Test-VirtualMachineAccessExtension
483478
Assert-AreEqual $vm1.Name $vmname;
484479
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces.Count 1;
485480
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces[0].ReferenceUri $nicId;
486-
# Assert-AreEqual $vm1.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
487481
Assert-AreEqual $vm1.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
488482
Assert-AreEqual $vm1.OSProfile.AdminUsername $user;
489483
Assert-AreEqual $vm1.OSProfile.ComputerName $computerName;

‎src/ResourceManager/Compute/Commands.Compute.Test/ScenarioTests/VirtualMachineNetworkInterfaceTests.ps1

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -94,12 +94,11 @@ function Test-SingleNetworkInterface
9494

9595
$p.StorageProfile.OSDisk = $null;
9696
$p = Set-AzureVMOperatingSystem -VM $p -Windows -ComputerName $computerName -Credential $cred;
97-
$p = Set-AzureVMSourceImage -VM $p -Name $img # -DestinationVhdsContainer $vhdContainer;
97+
$p = Set-AzureVMSourceImage -VM $p -Name $img;
9898

9999
Assert-AreEqual $p.OSProfile.AdminUsername $user;
100100
Assert-AreEqual $p.OSProfile.ComputerName $computerName;
101101
Assert-AreEqual $p.OSProfile.AdminPassword $password;
102-
# Assert-AreEqual $p.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
103102
Assert-AreEqual $p.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
104103

105104
# Virtual Machine
@@ -111,7 +110,6 @@ function Test-SingleNetworkInterface
111110
Assert-AreEqual $vm1.Name $vmname;
112111
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces.Count 1;
113112
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces[0].ReferenceUri $nicId;
114-
# Assert-AreEqual $vm1.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
115113
Assert-AreEqual $vm1.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
116114
Assert-AreEqual $vm1.OSProfile.AdminUsername $user;
117115
Assert-AreEqual $vm1.OSProfile.ComputerName $computerName;
@@ -202,7 +200,7 @@ function Test-SingleNetworkInterfaceDnsSettings
202200

203201
$p.StorageProfile.OSDisk = $null;
204202
$p = Set-AzureVMOperatingSystem -VM $p -Windows -ComputerName $computerName -Credential $cred;
205-
$p = Set-AzureVMSourceImage -VM $p -Name $img # -DestinationVhdsContainer $vhdContainer;
203+
$p = Set-AzureVMSourceImage -VM $p -Name $img;
206204

207205
# Virtual Machine
208206
# TODO: Still need to do retry for New-AzureVM for SA, even it's returned in Get-.
@@ -313,12 +311,11 @@ function Test-MultipleNetworkInterface
313311

314312
$p.StorageProfile.OSDisk = $null;
315313
$p = Set-AzureVMOperatingSystem -VM $p -Windows -ComputerName $computerName -Credential $cred;
316-
$p = Set-AzureVMSourceImage -VM $p -Name $img # -DestinationVhdsContainer $vhdContainer;
314+
$p = Set-AzureVMSourceImage -VM $p -Name $img;
317315

318316
Assert-AreEqual $p.OSProfile.AdminUsername $user;
319317
Assert-AreEqual $p.OSProfile.ComputerName $computerName;
320318
Assert-AreEqual $p.OSProfile.AdminPassword $password;
321-
# Assert-AreEqual $p.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
322319
Assert-AreEqual $p.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
323320

324321
# Virtual Machine
@@ -331,7 +328,6 @@ function Test-MultipleNetworkInterface
331328
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces.Count 2;
332329
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces[0].ReferenceUri $nic1.Id;
333330
Assert-AreEqual $vm1.NetworkProfile.NetworkInterfaces[1].ReferenceUri $nic2.Id;
334-
# Assert-AreEqual $vm1.StorageProfile.DestinationVhdsContainer.ToString() $vhdContainer;
335331
Assert-AreEqual $vm1.StorageProfile.SourceImage.ReferenceUri ('/' + (Get-AzureSubscription -Current).SubscriptionId + '/services/images/' + $img);
336332
Assert-AreEqual $vm1.OSProfile.AdminUsername $user;
337333
Assert-AreEqual $vm1.OSProfile.ComputerName $computerName;

0 commit comments

Comments
 (0)