RoleResourceIT.java example

Explorer
abiquo-master
/**
 * 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.api.resources;

import static com.abiquo.api.common.Assert.assertLinkExist;
import static com.abiquo.api.common.UriTestResolver.resolveEnterpriseURI;
import static com.abiquo.api.common.UriTestResolver.resolveRoleActionGetPrivilegesURI;
import static com.abiquo.api.common.UriTestResolver.resolveRoleURI;
import static org.testng.Assert.assertEquals;
import static org.testng.Assert.assertNotNull;

import java.util.ArrayList;
import java.util.List;

import org.apache.wink.client.ClientResponse;
import org.apache.wink.client.ClientWebException;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.abiquo.model.transport.LinksDto;
import com.abiquo.server.core.enterprise.Enterprise;
import com.abiquo.server.core.enterprise.Privilege;
import com.abiquo.server.core.enterprise.Role;
import com.abiquo.server.core.enterprise.RoleDto;
import com.abiquo.server.core.enterprise.User;

public class RoleResourceIT extends AbstractJpaGeneratorIT
{
    @BeforeMethod
    public void setupSysadmin()
    {
        Enterprise e = enterpriseGenerator.createUniqueInstance();
        Role r = roleGenerator.createInstanceSysAdmin();
        User u = userGenerator.createInstance(e, r, "sysadmin", "sysadmin");

        List<Object> entitiesToSetup = new ArrayList<Object>();
        entitiesToSetup.add(e);
        for (Privilege p : r.getPrivileges())
        {
            entitiesToSetup.add(p);
        }
        entitiesToSetup.add(r);
        entitiesToSetup.add(u);

        setup(entitiesToSetup.toArray());
    }

    @Test
    public void getRoleDoesntExist() throws ClientWebException
    {
        Role role = roleGenerator.createUniqueInstance();

        List<Object> entitiesToSetup = new ArrayList<Object>();

        for (Privilege p : role.getPrivileges())
        {
            entitiesToSetup.add(p);
        }
        entitiesToSetup.add(role);

        setup(entitiesToSetup.toArray());

        ClientResponse response =
            get(resolveRoleURI(12345), "sysadmin", "sysadmin", RoleDto.MEDIA_TYPE);
        assertEquals(response.getStatusCode(), 404);
    }

    @Test
    public void getRole() throws Exception
    {
        Role role = roleGenerator.createUniqueInstance();
        List<Object> entitiesToSetup = new ArrayList<Object>();

        for (Privilege p : role.getPrivileges())
        {
            entitiesToSetup.add(p);
        }
        entitiesToSetup.add(role);

        setup(entitiesToSetup.toArray());

        ClientResponse response =
            get(resolveRoleURI(role.getId()), "sysadmin", "sysadmin",
                RoleDto.MEDIA_TYPE);

        RoleDto dto = response.getEntity(RoleDto.class);

        assertNotNull(dto);
    }

    @Test
    public void roleContainCorrectLinks() throws ClientWebException
    {
        Role role = roleGenerator.createUniqueInstance();
        Enterprise ent = enterpriseGenerator.createUniqueInstance();
        role.setEnterprise(ent);

        List<Object> entitiesToSetup = new ArrayList<Object>();

        for (Privilege p : role.getPrivileges())
        {
            entitiesToSetup.add(p);
        }
        entitiesToSetup.add(ent);
        entitiesToSetup.add(role);

        setup(entitiesToSetup.toArray());

        String href = resolveRoleURI(role.getId());
        String enterpriseUri = resolveEnterpriseURI(role.getEnterprise().getId());
        String privilegesUri = resolveRoleActionGetPrivilegesURI(role.getId());

        RoleDto dto =
            get(href, "sysadmin", "sysadmin", RoleDto.MEDIA_TYPE).getEntity(
                RoleDto.class);

        assertNotNull(dto.getLinks());

        assertLinkExist(dto, href, "edit");
        assertLinkExist(dto, enterpriseUri, "enterprise");
        assertLinkExist(dto, privilegesUri, "privileges");
    }

}