@Test public void testUpdateState() throws Throwable { assertThat(vm.getState(), is(not(VmState.ERROR))); vmDcpBackend.updateState(vm, VmState.ERROR); assertThat(vm.getState(), is(VmState.ERROR)); vm = vmDcpBackend.findById(vmId); assertThat(vm.getState(), is(VmState.ERROR)); String agent = UUID.randomUUID().toString(); String agentIp = UUID.randomUUID().toString(); String datastoreId = UUID.randomUUID().toString(); String datastoreName = UUID.randomUUID().toString(); vmDcpBackend.updateState(vm, VmState.STARTED, agent, agentIp, datastoreId, datastoreName); assertThat(vm.getState(), is(VmState.STARTED)); assertThat(vm.getAgent(), is(agent)); assertThat(vm.getHost(), is(agentIp)); assertThat(vm.getDatastore(), is(datastoreId)); assertThat(vm.getDatastoreName(), is(datastoreName)); vm = vmDcpBackend.findById(vmId); assertThat(vm.getState(), is(VmState.STARTED)); assertThat(vm.getAgent(), is(agent)); assertThat(vm.getHost(), is(agentIp)); assertThat(vm.getDatastore(), is(datastoreId)); assertThat(vm.getDatastoreName(), is(datastoreName)); }
@Test public void testAddtag() throws Throwable { vm = vmDcpBackend.findById(vm.getId()); int originalSize = vm.getTags().size(); Tag tag = new Tag("namespace:predicate=value"); vmDcpBackend.addTag(vm.getId(), tag); vm = vmDcpBackend.findById(vm.getId()); assertThat(vm.getTags().size(), is(originalSize + 1)); List<String> tagValues = vm.getTags().stream().map(t -> t.getValue()).collect(Collectors.toList()); assertTrue(tagValues.contains(tag.getValue())); }
@Test public void testFindById() throws Throwable { VmEntity foundVmEntity = vmDcpBackend.findById(vmId); assertThat(foundVmEntity.getName(), is(vm.name)); assertThat(foundVmEntity.getFlavorId(), is(vm.flavorId)); assertThat(foundVmEntity.getImageId(), is(vm.imageId)); assertThat(foundVmEntity.getProjectId(), is(vm.projectId)); assertThat(foundVmEntity.getState(), is(vm.vmState)); assertThat( foundVmEntity.getAffinities().get(0).getResourceId(), is(vm.affinities.get(0).getId())); assertThat( foundVmEntity.getAffinities().get(0).getKind(), is(vm.affinities.get(0).getKind())); assertThat(foundVmEntity.getIsos().get(0).getName(), is(vm.isos.get(0).getName())); assertThat(foundVmEntity.getIsos().get(0).getSize(), is(vm.isos.get(0).getSize())); assertThat(foundVmEntity.getMetadata().get("key1"), is(vm.metadata.get("key1"))); assertThat(foundVmEntity.getNetworks().get(0), is(vm.networks.get(0))); assertThat(vm.tags.contains(foundVmEntity.getTags().iterator().next().getValue()), is(true)); assertThat(foundVmEntity.getAgent(), is(vm.agent)); assertThat(foundVmEntity.getHost(), is(vm.host)); assertThat(foundVmEntity.getDatastore(), is(vm.datastore)); assertThat(foundVmEntity.getDatastoreName(), is(vm.datastoreName)); }
@Test public void testAddTagWhenOriginalTagIsNull() throws Throwable { // overwriting the vmCreateSpec tags to empty from non-empty vmCreateSpec.setTags(null); createdVmTaskEntity = vmDcpBackend.prepareVmCreate(projectId, vmCreateSpec); vmId = createdVmTaskEntity.getEntityId(); entityLockDcpBackend.clearTaskLocks(createdVmTaskEntity); vm = vmDcpBackend.findById(vmId); int originalSize = vm.getTags().size(); Tag tag = new Tag("namespace:predicate=value"); vmDcpBackend.addTag(vm.getId(), tag); vm = vmDcpBackend.findById(vm.getId()); assertThat(vm.getTags().size(), is(1)); List<String> tagValues = vm.getTags().stream().map(t -> t.getValue()).collect(Collectors.toList()); assertTrue(tagValues.contains(tag.getValue())); }
@Test public void testFindByIdWithNonExistingId() throws Throwable { String id = UUID.randomUUID().toString(); try { vmDcpBackend.findById(id); fail("vmDcpBackend.findById for a non existing id should have failed"); } catch (VmNotFoundException e) { assertThat(e.getMessage(), containsString(id)); } }
@Test public void testPrepareVmGetMksTicket() throws Throwable { vmDcpBackend.updateState(vmDcpBackend.findById(vmId), VmState.STARTED); TaskEntity task = vmDcpBackend.prepareVmGetMksTicket(vmId); assertThat(task, is(notNullValue())); assertThat(task.getState(), is(TaskEntity.State.QUEUED)); assertThat(task.getSteps().size(), is(1)); assertThat( task.getSteps().get(0).getOperation(), is(com.vmware.photon.controller.api.Operation.GET_MKS_TICKET)); }
@Test public void testAddIso() throws Throwable { IsoEntity isoEntity = new IsoEntity(); isoEntity.setId("iso-id"); isoEntity.setName("iso-name"); isoEntity.setSize(100L); vmDcpBackend.addIso(isoEntity, vm); VmEntity updatedVm = vmDcpBackend.findById(vmId); assertThat(updatedVm, CoreMatchers.notNullValue()); assertThat(updatedVm.getIsos().size(), is(1)); assertThat(updatedVm.getIsos().get(0), is(isoEntity)); }
@Test public void testPrepareVmCreate() throws Throwable { String vmId = createdVmTaskEntity.getEntityId(); assertThat(createdVmTaskEntity.getSteps().size(), is(2)); assertThat( createdVmTaskEntity.getSteps().get(0).getOperation(), is(com.vmware.photon.controller.api.Operation.RESERVE_RESOURCE)); assertThat( createdVmTaskEntity .getSteps() .get(0) .getTransientResourceEntities(ProjectEntity.KIND) .size(), is(1)); assertThat( createdVmTaskEntity .getSteps() .get(0) .getTransientResourceEntities(ProjectEntity.KIND) .get(0) .getId(), is(projectId)); assertThat( createdVmTaskEntity.getSteps().get(1).getOperation(), is(com.vmware.photon.controller.api.Operation.CREATE_VM)); assertThat(createdVmTaskEntity.getToBeLockedEntityIds().size(), is(1)); assertThat(createdVmTaskEntity.getToBeLockedEntityIds().get(0), is(vmId)); VmEntity vm = vmDcpBackend.findById(vmId); assertThat(vm, is(notNullValue())); assertThat(getUsage("vm.cost"), is(1.0)); assertThat(vm.getImageId(), is(imageId)); assertThat(vm.getAffinities().get(0).getResourceId(), is("disk-id1")); assertThat(vm.getAffinities().get(0).getKind(), is("disk")); assertThat(vm.getAffinities().get(1).getResourceId(), is("disk-id2")); assertThat(vm.getAffinities().get(1).getKind(), is("disk")); Set<TagEntity> tags = vm.getTags(); assertThat(tags.size(), is(2)); TagEntity tag1 = new TagEntity(); tag1.setValue("value1"); TagEntity tag2 = new TagEntity(); tag2.setValue("value2"); assertTrue(tags.contains(tag1)); assertTrue(tags.contains(tag2)); assertThat(vmCreateSpec.getNetworks().equals(vm.getNetworks()), is(true)); }
@BeforeMethod public void setUp() throws Throwable { commonHostAndClientSetup(basicServiceHost, apiFeDcpRestClient); commonDataSetup( tenantDcpBackend, resourceTicketDcpBackend, projectDcpBackend, flavorDcpBackend, flavorLoader); commonVmAndImageSetup(vmDcpBackend, networkDcpBackend); vmId = createdVmTaskEntity.getEntityId(); entityLockDcpBackend.clearTaskLocks(createdVmTaskEntity); vm = vmDcpBackend.findById(vmId); }
@Test public void testPrepareSetMetadata() throws Throwable { Map<String, String> metadata = new HashMap<>(); metadata.put("key", "value"); TaskEntity task = vmDcpBackend.prepareSetMetadata(vmId, metadata); assertThat(task, is(notNullValue())); assertThat(task.getState(), is(TaskEntity.State.COMPLETED)); assertThat(task.getSteps().size(), is(0)); // check that metadata was saved VmEntity updatedVm = vmDcpBackend.findById(vmId); assertThat(updatedVm, notNullValue()); assertThat(updatedVm.getMetadata(), is(metadata)); // make sure that no other fields have changed updatedVm.setMetadata(this.vm.getMetadata()); assertThat(this.vm, is(updatedVm)); }