/**
   * Static method that converts the persistence {@link VirtualMachine} object to transfer {@link
   * VirtualMachineDto} object. It also adds REST self-discover {@link RESTLink}s
   *
   * @param vm Input persistence object.
   * @param restBuilder Context-injected {@link RESTLink} builder to create the links.
   * @return the corresponding transfer object.
   * @throws Exception Serialization unhandled exception.
   */
  public static VirtualMachineDto createTransferObject(
      final Integer datacenterId,
      final Integer rackId,
      final Integer machineId,
      final VirtualMachine vm,
      final VirtualAppliance vapp,
      final IRESTBuilder restBuilder)
      throws Exception {
    // TODO: Try with enterprise and user != null
    VirtualMachineDto vmDto =
        ModelTransformer.transportFromPersistence(VirtualMachineDto.class, vm);
    if (vm.getIdType() == 0) {
      vmDto.setIdType(com.abiquo.server.core.cloud.VirtualMachine.NOT_MANAGED);
    } else {
      vmDto.setIdType(com.abiquo.server.core.cloud.VirtualMachine.MANAGED);
    }
    Integer enterpriseId = null;
    Integer userId = null;

    if (vm.getEnterprise() != null) {
      enterpriseId = vm.getEnterprise().getId();
    }
    if (vm.getUser() != null) {
      userId = vm.getUser().getId();
    }
    vmDto.setLinks(
        restBuilder.buildVirtualMachineAdminLinks(
            datacenterId,
            rackId,
            machineId,
            enterpriseId,
            userId,
            vm.getHypervisor().getType(),
            vapp,
            vm.getId()));

    final VirtualMachineTemplate vmtemplate = vm.getVirtualMachineTemplate();
    if (vmtemplate != null) {
      if (vmtemplate.getRepository() != null) {
        vmDto.addLink(
            restBuilder.buildVirtualMachineTemplateLink(
                vmtemplate.getEnterprise().getId(),
                vmtemplate.getRepository().getDatacenter().getId(),
                vmtemplate.getId()));
      } else {
        // imported virtual machines
        vmDto.addLink(
            restBuilder.buildVirtualMachineTemplateLink(
                vmtemplate.getEnterprise().getId(),
                vm.getHypervisor().getMachine().getRack().getDatacenter().getId(),
                vmtemplate.getId()));
      }
    }
    return vmDto;
  }
Exemplo n.º 2
0
  @Test
  public void testDeleteMachineWithVirtualMachinesDeployed() {
    Hypervisor hypervisor = hypervisorGenerator.createUniqueInstance();
    Datacenter datacenter = hypervisor.getMachine().getDatacenter();
    VirtualDatacenter vdc = vdcGenerator.createInstance(datacenter);
    RemoteService rm =
        remoteServiceGenerator.createInstance(RemoteServiceType.VIRTUAL_SYSTEM_MONITOR, datacenter);

    VirtualImage image = virtualImageGenerator.createInstance(vdc.getEnterprise());
    VirtualAppliance vapp = virtualApplianceGenerator.createInstance(vdc);
    VirtualMachine vm =
        vmGenerator.createInstance(image, vdc.getEnterprise(), hypervisor, "vm_test");
    vm.setState(VirtualMachineState.RUNNING);
    vm.setIdType(VirtualMachine.MANAGED);

    NodeVirtualImage node = new NodeVirtualImage("node_test", vapp, image, vm);

    hypervisor.getMachine().setHypervisor(hypervisor);

    List<Object> entitiesToPersist = new ArrayList<Object>();
    entitiesToPersist.add(vdc.getEnterprise());
    entitiesToPersist.add(datacenter);
    entitiesToPersist.add(rm);
    entitiesToPersist.add(hypervisor.getMachine().getRack());
    entitiesToPersist.add(hypervisor.getMachine());
    entitiesToPersist.add(hypervisor);
    entitiesToPersist.add(vdc);
    entitiesToPersist.add(image);
    entitiesToPersist.add(vapp);
    for (Privilege p : vm.getUser().getRole().getPrivileges()) {
      entitiesToPersist.add(p);
    }
    entitiesToPersist.add(vm.getUser().getRole());
    entitiesToPersist.add(vm.getUser());
    entitiesToPersist.add(vm);
    entitiesToPersist.add(node);

    setup(entitiesToPersist.toArray());

    int machineId = hypervisor.getMachine().getId();

    EntityManager em = getEntityManager();
    EntityManagerHelper.beginReadWriteTransaction(em);

    MachineService service = new MachineService(em);
    service.setVsm(new VsmServiceStubMock()); // Must use the mocked VSM
    service.removeMachine(machineId);

    EntityManagerHelper.commit(em);

    EntityManagerHelper.beginRollbackTransaction(em);
    service = new MachineService(em);

    try {
      Machine m = service.getMachine(machineId);
    } catch (NotFoundException e) {
      org.testng.Assert.assertEquals(
          e.getErrors().iterator().next().getMessage(), "The requested machine does not exist");
    }

    VirtualMachineService vmService = new VirtualMachineService(em);

    VirtualMachine virtualMachine =
        vmService.getVirtualMachine(vdc.getId(), vapp.getId(), vm.getId());
    org.testng.Assert.assertNull(virtualMachine.getHypervisor());
    org.testng.Assert.assertNull(virtualMachine.getDatastore());
    org.testng.Assert.assertEquals(virtualMachine.getState(), VirtualMachineState.NOT_DEPLOYED);
  }