/*
* This file is part of LibrePlan
*
* Copyright (C) 2009-2010 Fundación para o Fomento da Calidade Industrial e
* Desenvolvemento Tecnolóxico de Galicia
* Copyright (C) 2010-2011 Igalia, S.L.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package org.libreplan.business.test.costcategories.daos;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.libreplan.business.BusinessGlobalNames.BUSINESS_SPRING_CONFIG_FILE;
import static org.libreplan.business.test.BusinessGlobalNames.BUSINESS_SPRING_CONFIG_TEST_FILE;
import java.math.BigDecimal;
import java.util.List;
import java.util.UUID;
import org.joda.time.LocalDate;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.libreplan.business.common.exceptions.InstanceNotFoundException;
import org.libreplan.business.common.exceptions.ValidationException;
import org.libreplan.business.costcategories.daos.ICostCategoryDAO;
import org.libreplan.business.costcategories.daos.ITypeOfWorkHoursDAO;
import org.libreplan.business.costcategories.entities.CostCategory;
import org.libreplan.business.costcategories.entities.HourCost;
import org.libreplan.business.costcategories.entities.TypeOfWorkHours;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.transaction.annotation.Transactional;
@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = { BUSINESS_SPRING_CONFIG_FILE,
BUSINESS_SPRING_CONFIG_TEST_FILE })
/**
* Test for {@CostCategoryDAO}
*
* @author Jacobo Aragunde Perez <jaragunde@igalia.com>
*
*/
public class CostCategoryDAOTest {
@Autowired
ICostCategoryDAO costCategoryDAO;
@Autowired
ITypeOfWorkHoursDAO typeOfWorkHoursDAO;
@Test
@Transactional
public void testInSpringContainer() {
assertNotNull(costCategoryDAO);
}
private CostCategory createValidCostCategory() {
CostCategory costCategory = CostCategory.create(UUID.randomUUID().toString());
return costCategory;
}
@Test
@Transactional
public void testSaveCostCategory() {
CostCategory costCategory = createValidCostCategory();
costCategoryDAO.save(costCategory);
assertTrue(costCategory.getId() != null);
}
@Test
@Transactional
public void testRemoveCostCategory() throws InstanceNotFoundException {
CostCategory costCategory = createValidCostCategory();
costCategoryDAO.save(costCategory);
costCategoryDAO.remove(costCategory.getId());
assertFalse(costCategoryDAO.exists(costCategory.getId()));
}
@Test
@Transactional
public void testListCostCategories() {
int previous = costCategoryDAO.list(CostCategory.class).size();
CostCategory costCategory = createValidCostCategory();
costCategoryDAO.save(costCategory);
List<CostCategory> list = costCategoryDAO.list(CostCategory.class);
assertEquals(previous + 1, list.size());
}
@Test
@Transactional
public void testCanAddHourCost() {
CostCategory costCategory = createValidCostCategory();
TypeOfWorkHours type1 = TypeOfWorkHours.create(UUID.randomUUID().toString(),
UUID.randomUUID().toString());
type1.setDefaultPrice(BigDecimal.TEN);
typeOfWorkHoursDAO.save(type1);
HourCost hourCost1 = HourCost.create(BigDecimal.ONE, new LocalDate(2009, 11,1));
hourCost1.setType(type1);
hourCost1.setEndDate(new LocalDate(2009, 11,10));
costCategory.addHourCost(hourCost1);
HourCost hourCost2 = HourCost.create(BigDecimal.ONE, new LocalDate(2009, 11,1));
hourCost2.setType(type1);
hourCost2.setEndDate(new LocalDate(2009, 11,10));
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost2.setInitDate(new LocalDate(2009,10,15));
hourCost2.setEndDate(new LocalDate(2009,11,1));
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost2.setInitDate(new LocalDate(2009,11,10));
hourCost2.setEndDate(new LocalDate(2009,11,10));
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost2.setInitDate(new LocalDate(2009,10,15));
hourCost2.setEndDate(new LocalDate(2009,10,20));
assertTrue(costCategory.canAddHourCost(hourCost2));
TypeOfWorkHours type2 = TypeOfWorkHours.create(UUID.randomUUID().toString(),
UUID.randomUUID().toString());
type2.setDefaultPrice(BigDecimal.TEN);
typeOfWorkHoursDAO.save(type2);
hourCost2.setType(type2);
hourCost2.setInitDate(new LocalDate(2009,10,15));
hourCost2.setEndDate(new LocalDate(2009,11,1));
assertTrue(costCategory.canAddHourCost(hourCost2));
hourCost2.setType(type1);
hourCost2.setInitDate(new LocalDate(2009,10,15));
hourCost2.setEndDate(null);
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost2.setInitDate(new LocalDate(2009,11,9));
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost2.setInitDate(new LocalDate(2009,11,11));
assertTrue(costCategory.canAddHourCost(hourCost2));
hourCost1.setEndDate(null);
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost2.setEndDate(new LocalDate(2009,11,30));
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost1.setEndDate(new LocalDate(2009,11,20));
assertFalse(costCategory.canAddHourCost(hourCost2));
hourCost1.setEndDate(new LocalDate(2009,12,1));
assertTrue(costCategory.canAddHourCost(hourCost2));
}
@Test
@Transactional
public void testListHourCosts() {
CostCategory costCategory = createValidCostCategory();
HourCost hourCost = HourCost.create(BigDecimal.ONE, new LocalDate(2009,11,1));
TypeOfWorkHours type =
TypeOfWorkHours.create(UUID.randomUUID().toString(), UUID.randomUUID().toString());
hourCost.setType(type);
int previous = costCategory.getHourCosts().size();
costCategory.addHourCost(hourCost);
costCategoryDAO.save(costCategory);
assertEquals(previous + 1, costCategory.getHourCosts().size());
costCategory.removeHourCost(hourCost);
costCategoryDAO.save(costCategory);
assertEquals(previous, costCategory.getHourCosts().size());
assertNull(hourCost.getCategory());
}
@Test(expected=ValidationException.class)
@Transactional
public void testHourCostsOverlap() {
CostCategory costCategory = createValidCostCategory();
TypeOfWorkHours type1 = TypeOfWorkHours.create(UUID.randomUUID().toString(),
UUID.randomUUID().toString());
type1.setDefaultPrice(BigDecimal.TEN);
TypeOfWorkHours type2 = TypeOfWorkHours.create(UUID.randomUUID().toString(),
UUID.randomUUID().toString());
type2.setDefaultPrice(BigDecimal.TEN);
//types have to be saved before using them
//otherwise, the overlapping validation will fail
typeOfWorkHoursDAO.save(type1);
typeOfWorkHoursDAO.save(type2);
HourCost hourCost1 = HourCost.create(BigDecimal.ONE, new LocalDate(2009, 11,1));
hourCost1.setType(type1);
hourCost1.setEndDate(new LocalDate(2009, 11,10));
costCategory.addHourCost(hourCost1);
HourCost hourCost2 = HourCost.create(BigDecimal.ONE, new LocalDate(2009, 11,1));
hourCost2.setType(type2);
hourCost2.setEndDate(new LocalDate(2009, 11,10));
costCategory.addHourCost(hourCost2);
//this save is correct
costCategoryDAO.save(costCategory);
hourCost2.setType(type1);
//this save should throw a exception
costCategoryDAO.save(costCategory);
}
}