@Test
public void testPrepareVmDiskOperation() throws Throwable {
com.vmware.photon.controller.api.Operation operation =
com.vmware.photon.controller.api.Operation.DETACH_DISK;
DiskService.State diskState = new DiskService.State();
diskState.name = "test-vm-disk-1";
diskState.projectId = projectId;
diskState.flavorId =
flavorDcpBackend.getEntityByNameAndKind("core-100", PersistentDisk.KIND).getId();
diskState.capacityGb = 64;
diskState.diskType = DiskType.PERSISTENT;
diskState.state = DiskState.ATTACHED;
Operation result = dcpClient.post(DiskServiceFactory.SELF_LINK, diskState);
DiskService.State createdDiskState = result.getBody(DiskService.State.class);
String diskId = ServiceUtils.getIDFromDocumentSelfLink(createdDiskState.documentSelfLink);
List<String> disks = new ArrayList<>();
disks.add(diskId);
TaskEntity task = vmDcpBackend.prepareVmDiskOperation(vmId, disks, operation);
assertThat(task, is(notNullValue()));
assertThat(task.getState(), is(TaskEntity.State.QUEUED));
assertThat(task.getSteps().size(), is(1));
assertThat(task.getSteps().get(0).getOperation(), is(operation));
try {
vmDcpBackend.prepareVmDiskOperation(
vmId, disks, com.vmware.photon.controller.api.Operation.MOCK_OP);
fail("vmDcpBackend.prepareVmOperation with invalid operation should have failed");
} catch (NotImplementedException e) {
// do nothing
}
}
@BeforeMethod
public void setUp() throws Throwable {
commonHostAndClientSetup(basicServiceHost, apiFeDcpRestClient);
commonDataSetup(
tenantDcpBackend,
resourceTicketDcpBackend,
projectDcpBackend,
flavorDcpBackend,
flavorLoader);
vm = new VmService.State();
vm.name = UUID.randomUUID().toString();
flavorEntity = flavorDcpBackend.getEntityByNameAndKind("core-100", Vm.KIND);
vm.flavorId = flavorEntity.getId();
vm.imageId = UUID.randomUUID().toString();
vm.projectId = projectId;
vm.vmState = VmState.CREATING;
vm.affinities = new ArrayList<>();
vm.affinities.add(new LocalitySpec("id1", "kind1"));
Iso iso = new Iso();
iso.setName(UUID.randomUUID().toString());
iso.setSize(-1L);
vm.isos = new ArrayList<>();
vm.isos.add(iso);
vm.metadata = new HashMap<>();
vm.metadata.put("key1", UUID.randomUUID().toString());
vm.networks = new ArrayList<>();
vm.networks.add(UUID.randomUUID().toString());
vm.agent = UUID.randomUUID().toString();
vm.host = UUID.randomUUID().toString();
vm.datastore = UUID.randomUUID().toString();
vm.datastoreName = UUID.randomUUID().toString();
vm.tags = new HashSet<>();
vm.tags.add("namespace1:predicate1=value1");
vm.tags.add("namespace2:predicate2=value2");
Operation result = dcpClient.post(VmServiceFactory.SELF_LINK, vm);
createdVm = result.getBody(VmService.State.class);
vmId = ServiceUtils.getIDFromDocumentSelfLink(createdVm.documentSelfLink);
DeploymentCreateSpec deploymentCreateSpec = new DeploymentCreateSpec();
deploymentCreateSpec.setImageDatastores(Collections.singleton(UUID.randomUUID().toString()));
TaskEntity deploymentTask = deploymentBackend.prepareCreateDeployment(deploymentCreateSpec);
HostCreateSpec hostCreateSpec = new HostCreateSpec();
hostCreateSpec.setAddress(vm.host);
hostCreateSpec.setUsageTags(ImmutableList.of(UsageTag.CLOUD));
hostCreateSpec.setUsername(UUID.randomUUID().toString());
hostCreateSpec.setPassword(UUID.randomUUID().toString());
TaskEntity hostTask =
hostBackend.prepareHostCreate(hostCreateSpec, deploymentTask.getEntityId());
hostId = hostTask.getEntityId();
}
private static void commonHostAndClientTeardown() throws Throwable {
if (dcpClient != null) {
dcpClient.stop();
dcpClient = null;
}
if (host != null) {
host.destroy();
host = null;
}
}
@Test
public void testDetachIso() throws Throwable {
VmService.State vmState = new VmService.State();
Iso iso = new Iso();
iso.setName(UUID.randomUUID().toString());
vmState.isos = new ArrayList<>();
vmState.isos.add(iso);
dcpClient.patch(VmServiceFactory.SELF_LINK + "/" + vm.getId(), vmState);
assertThat(vmDcpBackend.isosAttached(vm).isEmpty(), is(false));
vmDcpBackend.detachIso(vm);
assertThat(vmDcpBackend.isosAttached(vm).isEmpty(), is(true));
}
private static void commonVmAndImageSetup(
VmDcpBackend vmDcpBackend, NetworkDcpBackend networkDcpBackend) throws Throwable {
AttachedDiskCreateSpec disk1 =
new AttachedDiskCreateSpecBuilder().name("disk1").flavor("core-100").bootDisk(true).build();
AttachedDiskCreateSpec disk2 =
new AttachedDiskCreateSpecBuilder().name("disk2").flavor("core-200").capacityGb(10).build();
List<LocalitySpec> affinities = new ArrayList<>();
affinities.add(new LocalitySpec("disk-id1", "disk"));
affinities.add(new LocalitySpec("disk-id2", "disk"));
ImageService.State imageServiceState = new ImageService.State();
imageServiceState.name = "image-1";
imageServiceState.state = ImageState.READY;
imageServiceState.size = 1024L * 1024L;
imageServiceState.replicationType = ImageReplicationType.EAGER;
imageServiceState.imageSettings = new ArrayList<>();
ImageService.State.ImageSetting imageSetting = new ImageService.State.ImageSetting();
imageSetting.name = "n1";
imageSetting.defaultValue = "v1";
imageServiceState.imageSettings.add(imageSetting);
imageSetting = new ImageService.State.ImageSetting();
imageSetting.name = "n2";
imageSetting.defaultValue = "v2";
imageServiceState.imageSettings.add(imageSetting);
Operation result = dcpClient.post(ImageServiceFactory.SELF_LINK, imageServiceState);
createdImageState = result.getBody(ImageService.State.class);
imageId = ServiceUtils.getIDFromDocumentSelfLink(createdImageState.documentSelfLink);
vmCreateSpec = new VmCreateSpec();
vmCreateSpec.setName("test-vm");
vmCreateSpec.setFlavor("core-100");
vmCreateSpec.setSourceImageId(imageId);
vmCreateSpec.setAttachedDisks(ImmutableList.of(disk1, disk2));
vmCreateSpec.setAffinities(affinities);
vmCreateSpec.setTags(ImmutableSet.of("value1", "value2"));
List<String> networks = new ArrayList<>();
List<String> portGroups = new ArrayList<>();
portGroups.add("p1");
NetworkCreateSpec networkCreateSpec = new NetworkCreateSpec();
networkCreateSpec.setName("n1");
networkCreateSpec.setPortGroups(portGroups);
TaskEntity networkTask = networkDcpBackend.createNetwork(networkCreateSpec);
networks.add(networkTask.getEntityId());
portGroups = new ArrayList<>();
portGroups.add("p2");
networkCreateSpec.setName("n2");
networkCreateSpec.setPortGroups(portGroups);
networkTask = networkDcpBackend.createNetwork(networkCreateSpec);
networks.add(networkTask.getEntityId());
vmCreateSpec.setNetworks(networks);
createdVmTaskEntity = vmDcpBackend.prepareVmCreate(projectId, vmCreateSpec);
}