Skip to content

Commit

Permalink
Merge pull request #964 from gcheng/testfix
Browse files Browse the repository at this point in the history
validating a customer scenario
  • Loading branch information
Albert Cheng committed Jul 2, 2014
2 parents e8100d7 + 8543e01 commit e5dd2e5
Showing 1 changed file with 84 additions and 27 deletions.
Expand Up @@ -82,32 +82,8 @@ public static void cleanup() {
cleanStorageAccount(storageAccountName);
}

@Test
public void createVirtualMachines() throws Exception {
int random = (int)(Math.random()* 100);
String roleName = testVMPrefix + "vm2";
String computerName = testVMPrefix + "vm2";
String adminuserPassword = testVMPrefix + "!12";
String adminUserName = testVMPrefix;
URI mediaLinkUriValue = new URI("http://"+ storageAccountName + ".blob.core.windows.net/"+storageContainer+ "/" + testVMPrefix +random + ".vhd");
String osVHarddiskName =testVMPrefix + "oshdname" + random;
String operatingSystemName ="Windows";

//required
ArrayList<ConfigurationSet> configlist = new ArrayList<ConfigurationSet>();
ConfigurationSet configset = new ConfigurationSet();
configset.setConfigurationSetType(ConfigurationSetTypes.WINDOWSPROVISIONINGCONFIGURATION);
//required
configset.setComputerName(computerName);
//required
configset.setAdminPassword(adminuserPassword);
//required
configset.setAdminUserName(adminUserName);
configset.setEnableAutomaticUpdates(false);
configlist.add(configset);

//required
String sourceImageName = getOSSourceImage();
private OSVirtualHardDisk createOSVirtualHardDisk(String osVHarddiskName, String operatingSystemName, URI mediaLinkUriValue, String sourceImageName)
{
OSVirtualHardDisk oSVirtualHardDisk = new OSVirtualHardDisk();
//required
oSVirtualHardDisk.setName(osVHarddiskName);
Expand All @@ -117,15 +93,96 @@ public void createVirtualMachines() throws Exception {
oSVirtualHardDisk.setMediaLink(mediaLinkUriValue);
//required
oSVirtualHardDisk.setSourceImageName(sourceImageName);

return oSVirtualHardDisk;
}

private VirtualMachineCreateParameters createVirtualMachineCreateParameter(String roleName, ArrayList<ConfigurationSet> configlist, OSVirtualHardDisk oSVirtualHardDisk, String availabilitySetNameValue) {
VirtualMachineCreateParameters createParameters = new VirtualMachineCreateParameters();
//required
createParameters.setRoleName(roleName);
createParameters.setRoleSize(VirtualMachineRoleSize.MEDIUM);
createParameters.setProvisionGuestAgent(true);
createParameters.setConfigurationSets(configlist);
createParameters.setOSVirtualHardDisk(oSVirtualHardDisk);
createParameters.setAvailabilitySetName(availabilitySetNameValue);
return createParameters;
}

private ArrayList<ConfigurationSet> createConfigList(String computerName,
String adminuserPassword, String adminUserName) {
ArrayList<ConfigurationSet> configlist = new ArrayList<ConfigurationSet>();
ConfigurationSet configset = new ConfigurationSet();
configset.setConfigurationSetType(ConfigurationSetTypes.WINDOWSPROVISIONINGCONFIGURATION);
//required
configset.setComputerName(computerName);
//required
configset.setAdminPassword(adminuserPassword);
//required
configset.setAdminUserName(adminUserName);
configset.setEnableAutomaticUpdates(false);
configlist.add(configset);
return configlist;
}

@Test
public void createVirtualMachineInAvailabilitySetSuccess() throws Exception {
int random = (int)(Math.random()* 100);
String roleName = testVMPrefix + "vm3";
String roleNameSecond = testVMPrefix + "vm4";
String computerName = testVMPrefix + "vm3";
String computerNameSecond = testVMPrefix + "vm4";
String adminuserPassword = testVMPrefix + "!12";
String adminUserName = testVMPrefix;
URI mediaLinkUriValue = new URI("http://"+ storageAccountName + ".blob.core.windows.net/"+storageContainer+ "/" + testVMPrefix +random + ".vhd");
URI mediaLinkUriValueSecond = new URI("http://"+ storageAccountName + ".blob.core.windows.net/"+storageContainer+ "/" + testVMPrefix +random + "2.vhd");
String osVHarddiskName =testVMPrefix + "oshdname" + random;
String osVHarddiskNameSecond = testVMPrefix + "oshdname2" + random;
String operatingSystemName ="Windows";
String availabilitySetNameValue = "azurejava" + random;

//required
ArrayList<ConfigurationSet> configlist = createConfigList(computerName, adminuserPassword, adminUserName);
ArrayList<ConfigurationSet> configlistSecond = createConfigList(computerNameSecond, adminuserPassword, adminUserName);

//required
String sourceImageName = getOSSourceImage();
OSVirtualHardDisk oSVirtualHardDisk = createOSVirtualHardDisk(osVHarddiskName, operatingSystemName, mediaLinkUriValue, sourceImageName);
VirtualMachineCreateParameters createParameters = createVirtualMachineCreateParameter(roleName, configlist, oSVirtualHardDisk, availabilitySetNameValue);

OSVirtualHardDisk oSVirtualHardDiskSecond = createOSVirtualHardDisk(osVHarddiskNameSecond, operatingSystemName, mediaLinkUriValueSecond, sourceImageName);
VirtualMachineCreateParameters createParametersSecond = createVirtualMachineCreateParameter(roleNameSecond, configlistSecond, oSVirtualHardDiskSecond, availabilitySetNameValue);
//Act
OperationResponse operationResponse = computeManagementClient.getVirtualMachinesOperations().create(hostedServiceName, deploymentName, createParameters);
OperationResponse operationResponseSecond = computeManagementClient.getVirtualMachinesOperations().create(hostedServiceName, deploymentName, createParametersSecond);

//Assert
Assert.assertEquals(200, operationResponse.getStatusCode());
Assert.assertNotNull(operationResponse.getRequestId());
Assert.assertEquals(200, operationResponseSecond.getStatusCode());
Assert.assertNotNull(operationResponseSecond.getRequestId());

}


@Test
public void createVirtualMachines() throws Exception {
int random = (int)(Math.random()* 100);
String roleName = testVMPrefix + "vm2";
String computerName = testVMPrefix + "vm2";
String adminuserPassword = testVMPrefix + "!12";
String adminUserName = testVMPrefix;
URI mediaLinkUriValue = new URI("http://"+ storageAccountName + ".blob.core.windows.net/"+storageContainer+ "/" + testVMPrefix +random + ".vhd");
String osVHarddiskName =testVMPrefix + "oshdname" + random;
String operatingSystemName ="Windows";

//required
ArrayList<ConfigurationSet> configlist = createConfigList(computerName, adminuserPassword, adminUserName);

//required
String sourceImageName = getOSSourceImage();
OSVirtualHardDisk oSVirtualHardDisk = createOSVirtualHardDisk(osVHarddiskName, operatingSystemName, mediaLinkUriValue, sourceImageName);
VirtualMachineCreateParameters createParameters = createVirtualMachineCreateParameter(roleName, configlist, oSVirtualHardDisk, null);

//Act
OperationResponse operationResponse = computeManagementClient.getVirtualMachinesOperations().create(hostedServiceName, deploymentName, createParameters);

Expand Down

0 comments on commit e5dd2e5

Please sign in to comment.