VirtualDatacenterRepTest.java

/**
 * Abiquo community edition
 * cloud management application for hybrid clouds
 * Copyright (C) 2008-2010 - Abiquo Holdings S.L.
 *
 * This application is free software; you can redistribute it and/or
 * modify it under the terms of the GNU LESSER GENERAL PUBLIC
 * LICENSE as published by the Free Software Foundation under
 * version 3 of the License
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * LESSER GENERAL PUBLIC LICENSE v.3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
 * Boston, MA 02111-1307, USA.
 */

package com.abiquo.server.core.cloud;

import java.lang.reflect.InvocationTargetException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import javax.persistence.EntityManager;

import org.testng.Assert;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.abiquo.model.enumerator.RemoteServiceType;
import com.abiquo.server.core.common.persistence.DefaultJpaDataAccessTestBase;
import com.abiquo.server.core.infrastructure.RemoteService;
import com.abiquo.server.core.infrastructure.RemoteServiceGenerator;
import com.abiquo.server.core.infrastructure.network.IpPoolManagement;
import com.abiquo.server.core.infrastructure.network.IpPoolManagementGenerator;
import com.abiquo.server.core.infrastructure.network.VLANNetwork;
import com.abiquo.server.core.infrastructure.network.VLANNetworkGenerator;
import com.softwarementors.bzngine.engines.jpa.EntityManagerHelper;

public class VirtualDatacenterRepTest extends DefaultJpaDataAccessTestBase
{
    private VirtualDatacenterGenerator virtualDatacenterGenerator;

    private VLANNetworkGenerator vlanGenerator;

    private IpPoolManagementGenerator ipGenerator;

    private RemoteServiceGenerator remoteServiceGenerator;

    private VirtualDatacenterGenerator eg()
    {
        return this.virtualDatacenterGenerator;
    }

    private VLANNetworkGenerator vlanEg()
    {
        return vlanGenerator;
    }

    @Override
    @BeforeMethod
    public void methodSetUp()
    {
        super.methodSetUp();
        this.virtualDatacenterGenerator = new VirtualDatacenterGenerator(getSeed());
        this.vlanGenerator = new VLANNetworkGenerator(getSeed());
        this.ipGenerator = new IpPoolManagementGenerator(getSeed());
        this.remoteServiceGenerator = new RemoteServiceGenerator(getSeed());
    }

    @Test
    public void test_findVirtualDatacenterById()
    {
        List<Object> additionalEntities = new ArrayList<Object>();

        VirtualDatacenter vdc = eg().createUniqueInstance();
        eg().addAuxiliaryEntitiesToPersist(vdc, additionalEntities);

        persistAll(ds(), additionalEntities, vdc);

        VirtualDatacenterRep rep =
            new VirtualDatacenterRep(ds().createEntityManagerAndBeginRollbackTransaction());
        Assert.assertNotNull(rep.findById(vdc.getId()));
    }

    @Test
    public void test_findAllDatacenters() throws IllegalAccessException, InvocationTargetException,
        NoSuchMethodException
    {
        List<Object> additionalEntities = new ArrayList<Object>();

        VirtualDatacenter vdc1 = eg().createUniqueInstance();
        eg().addAuxiliaryEntitiesToPersist(vdc1, additionalEntities);

        VirtualDatacenter vdc2 = eg().createUniqueInstance();
        eg().addAuxiliaryEntitiesToPersist(vdc2, additionalEntities);

        persistAll(ds(), additionalEntities, vdc1, vdc2);

        VirtualDatacenterRep rep =
            new VirtualDatacenterRep(ds().createEntityManagerAndBeginRollbackTransaction());

        List<VirtualDatacenter> result = (List<VirtualDatacenter>) rep.findAll();
        assertNotNull(result);
        assertEquals(result.size(), 2);
    }

    @Test
    public void testFindVLANNetworkById()
    {
        List<Object> additionalEntities = new ArrayList<Object>();

        VLANNetwork lan = vlanEg().createUniqueInstance();
        vlanEg().addAuxiliaryEntitiesToPersist(lan, additionalEntities);

        persistAll(ds(), additionalEntities, lan);

        VirtualDatacenterRep rep =
            new VirtualDatacenterRep(ds().createEntityManagerAndBeginRollbackTransaction());
        Assert.assertNotNull(rep.findVlanById(lan.getId()));
    }

    @Test
    public void testFindAllVLANNetworks()
    {
        List<Object> additionalEntities = new ArrayList<Object>();

        VLANNetwork n1 = vlanEg().createUniqueInstance();
        vlanEg().addAuxiliaryEntitiesToPersist(n1, additionalEntities);
        VLANNetwork n2 = vlanEg().createUniqueInstance();
        vlanEg().addAuxiliaryEntitiesToPersist(n2, additionalEntities);

        persistAll(ds(), additionalEntities, n1, n2);

        VirtualDatacenterRep rep =
            new VirtualDatacenterRep(ds().createEntityManagerAndBeginRollbackTransaction());
        Collection<VLANNetwork> lans = rep.findAllVlans();
        Assert.assertNotNull(lans);
        Assert.assertEquals(lans.size(), 2);
    }

    @Test
    public void testFindAllVLANNetworksByVirtualDatacenter()
    {
        List<Object> additionalEntities = new ArrayList<Object>();

        VirtualDatacenter vdc1 = virtualDatacenterGenerator.createUniqueInstance();
        this.virtualDatacenterGenerator.addAuxiliaryEntitiesToPersist(vdc1, additionalEntities);

        VLANNetwork n1_1 = vlanEg().createUniqueInstance();
        vlanEg().addAuxiliaryEntitiesToPersist(n1_1, additionalEntities);
        VLANNetwork n1_2 = vlanEg().createUniqueInstance();
        vlanEg().addAuxiliaryEntitiesToPersist(n1_2, additionalEntities);

        n1_1.setNetwork(vdc1.getNetwork());
        n1_2.setNetwork(vdc1.getNetwork());

        VirtualDatacenter vdc2 = virtualDatacenterGenerator.createUniqueInstance();
        this.virtualDatacenterGenerator.addAuxiliaryEntitiesToPersist(vdc2, additionalEntities);

        VLANNetwork n2_1 = vlanEg().createUniqueInstance();
        vlanEg().addAuxiliaryEntitiesToPersist(n2_1, additionalEntities);

        n2_1.setNetwork(vdc2.getNetwork());

        persistAll(ds(), additionalEntities, vdc1, vdc2, n1_1, n1_2, n2_1);

        VirtualDatacenterRep rep =
            new VirtualDatacenterRep(ds().createEntityManagerAndBeginRollbackTransaction());
        Collection<VLANNetwork> networks = rep.findVlansByVirtualDatacener(vdc1);
        Assert.assertFalse(networks.isEmpty());
        Assert.assertEquals(networks.size(), 2);
    }

    @Test
    public void testFindZeroVLANNetworksByDatacenter()
    {
        List<Object> additionalEntities = new ArrayList<Object>();

        VirtualDatacenter vdc1 = virtualDatacenterGenerator.createUniqueInstance();
        this.virtualDatacenterGenerator.addAuxiliaryEntitiesToPersist(vdc1, additionalEntities);

        VirtualDatacenter vdc2 = virtualDatacenterGenerator.createUniqueInstance();
        this.virtualDatacenterGenerator.addAuxiliaryEntitiesToPersist(vdc2, additionalEntities);

        VLANNetwork n2_1 = vlanEg().createUniqueInstance();
        vlanEg().addAuxiliaryEntitiesToPersist(n2_1, additionalEntities);

        n2_1.setNetwork(vdc2.getNetwork());

        persistAll(ds(), additionalEntities, vdc1, vdc2, n2_1);

        VirtualDatacenterRep rep =
            new VirtualDatacenterRep(ds().createEntityManagerAndBeginRollbackTransaction());
        Collection<VLANNetwork> networks = rep.findVlansByVirtualDatacener(vdc1);
        Assert.assertTrue(networks.isEmpty());
    }

    @Test
    public void deleteVirtualDatacenter()
    {
        VirtualDatacenter vdc = virtualDatacenterGenerator.createUniqueInstance();
        IpPoolManagement ip = ipGenerator.createInstance(vdc, vdc.getNetwork());
        RemoteService rs =
            remoteServiceGenerator.createInstance(RemoteServiceType.DHCP_SERVICE,
                vdc.getDatacenter());

        List<Object> additionalEntities = new ArrayList<Object>();
        ipGenerator.addAuxiliaryEntitiesToPersist(ip, additionalEntities);

        persistAll(ds(), additionalEntities, ip);

        EntityManager em = ds().createEntityManagerAndBeginReadWriteTransaction();

        VirtualDatacenterRep rep = new VirtualDatacenterRep(em);
        VirtualDatacenter vdc1 = rep.findById(vdc.getId());

        try
        {
            rep.delete(vdc1);
            EntityManagerHelper.commitAndClose(em);
        }
        catch (Exception e)
        {
            e.printStackTrace();
            fail();
        }
    }
}