/**
 * 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.appslibrary;

import static com.abiquo.api.common.UriTestResolver.resolveTemplateDefinitionURI;
import static com.abiquo.testng.TestConfig.APPS_INTEGRATION_TESTS;
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.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.abiquo.api.resources.AbstractJpaGeneratorIT;
import com.abiquo.model.enumerator.DiskFormatType;
import com.abiquo.server.core.appslibrary.AppsLibrary;
import com.abiquo.server.core.appslibrary.Category;
import com.abiquo.server.core.appslibrary.TemplateDefinition;
import com.abiquo.server.core.appslibrary.TemplateDefinitionDto;
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.User;
import com.abiquo.server.core.infrastructure.Datacenter;

public class TemplateDefinitionResourceIT extends AbstractJpaGeneratorIT
{

    protected Enterprise enterprise;

    protected Datacenter datacenter;

    protected AppsLibrary appsLibrary;

    protected Category category;

    protected TemplateDefinition templateDef;

    private static final String SYSADMIN = "sysadmin";

    @BeforeMethod(groups = {APPS_INTEGRATION_TESTS})
    public void setUpUser()
    {
        enterprise = enterpriseGenerator.createUniqueInstance();
        appsLibrary = appsLibraryGenerator.createUniqueInstance(enterprise);

        datacenter = datacenterGenerator.createUniqueInstance();
        category = categoryGenerator.createUniqueInstance();
        category.setName("category_1");

        Role role = roleGenerator.createInstanceSysAdmin();
        User user = userGenerator.createInstance(enterprise, role, SYSADMIN, SYSADMIN);

        List<Object> entitiesToSetup = new ArrayList<Object>();
        entitiesToSetup.add(enterprise);
        entitiesToSetup.add(appsLibrary);
        entitiesToSetup.add(datacenter);

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

        entitiesToSetup.add(category);

        setup(entitiesToSetup.toArray());
    }

    @AfterMethod(groups = {APPS_INTEGRATION_TESTS})
    public void tearDownUser()
    {
        super.tearDown();
    }

    @Test(groups = {APPS_INTEGRATION_TESTS})
    public void getTemplateDefinition() throws ClientWebException
    {
        templateDef = templateDefGenerator.createInstance(appsLibrary, category);
        templateDef.setDescription("templateDef_1");
        templateDef.setType(DiskFormatType.UNKNOWN);
        setup(templateDef);

        ClientResponse response =
            get(resolveTemplateDefinitionURI(enterprise.getId(), templateDef.getId()), SYSADMIN,
                SYSADMIN, TemplateDefinitionDto.MEDIA_TYPE);

        assertEquals(response.getStatusCode(), 200);

        TemplateDefinitionDto templateDefDto = response.getEntity(TemplateDefinitionDto.class);
        assertNotNull(templateDefDto);
        assertEquals(templateDefDto.getDescription(), "templateDef_1");
        assertEquals(templateDefDto.getDiskFormatType(), "UNKNOWN");
    }

    @Test(groups = {APPS_INTEGRATION_TESTS})
    public void modifyTemplateDefinition() throws ClientWebException
    {
        templateDef = templateDefGenerator.createInstance(appsLibrary, category);
        templateDef.setUrl("http://some.com/url");
        setup(templateDef);

        ClientResponse response =
            get(resolveTemplateDefinitionURI(enterprise.getId(), templateDef.getId()), SYSADMIN,
                SYSADMIN, TemplateDefinitionDto.MEDIA_TYPE);

        assertEquals(response.getStatusCode(), 200);

        TemplateDefinitionDto templateDefDto = response.getEntity(TemplateDefinitionDto.class);
        assertNotNull(templateDefDto);

        // modifications
        templateDefDto.setDescription("new_description");

        response =
            put(resolveTemplateDefinitionURI(enterprise.getId(), templateDefDto.getId()),
                templateDefDto, SYSADMIN, SYSADMIN);

        assertEquals(response.getStatusCode(), 200);
        response =
            get(resolveTemplateDefinitionURI(enterprise.getId(), templateDef.getId()), SYSADMIN,
                SYSADMIN, TemplateDefinitionDto.MEDIA_TYPE);
        TemplateDefinitionDto retrievedPackageDto = response.getEntity(TemplateDefinitionDto.class);
        assertEquals(retrievedPackageDto.getDescription(), "new_description");
    }

    @Test(groups = {APPS_INTEGRATION_TESTS})
    public void deleteTemplateDefinition() throws ClientWebException
    {
        templateDef = templateDefGenerator.createInstance(appsLibrary, category);
        templateDef.setDescription("templateDef_1");
        templateDef.setType(DiskFormatType.UNKNOWN);
        setup(templateDef);

        ClientResponse response =
            delete(resolveTemplateDefinitionURI(enterprise.getId(), templateDef.getId()), SYSADMIN,
                SYSADMIN);

        assertEquals(response.getStatusCode(), 204);

        response =
            get(resolveTemplateDefinitionURI(enterprise.getId(), templateDef.getId()), SYSADMIN,
                SYSADMIN, TemplateDefinitionDto.MEDIA_TYPE);
        assertEquals(response.getStatusCode(), 404);
    }
}