@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); }