/*
* 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.orders.daos;
import static junit.framework.Assert.assertNotNull;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertThat;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
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.Arrays;
import java.util.Collections;
import java.util.Date;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.SortedSet;
import java.util.UUID;
import javax.annotation.Resource;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.libreplan.business.IDataBootstrap;
import org.libreplan.business.advance.bootstrap.PredefinedAdvancedTypes;
import org.libreplan.business.advance.entities.AdvanceMeasurement;
import org.libreplan.business.advance.entities.AdvanceType;
import org.libreplan.business.advance.entities.DirectAdvanceAssignment;
import org.libreplan.business.advance.exceptions.DuplicateAdvanceAssignmentForOrderElementException;
import org.libreplan.business.advance.exceptions.DuplicateValueTrueReportGlobalAdvanceException;
import org.libreplan.business.calendars.daos.IBaseCalendarDAO;
import org.libreplan.business.calendars.entities.BaseCalendar;
import org.libreplan.business.common.exceptions.InstanceNotFoundException;
import org.libreplan.business.common.exceptions.ValidationException;
import org.libreplan.business.labels.daos.ILabelDAO;
import org.libreplan.business.labels.daos.ILabelTypeDAO;
import org.libreplan.business.labels.entities.Label;
import org.libreplan.business.labels.entities.LabelType;
import org.libreplan.business.orders.daos.IOrderElementDAO;
import org.libreplan.business.orders.entities.Order;
import org.libreplan.business.orders.entities.OrderElement;
import org.libreplan.business.orders.entities.OrderLine;
import org.libreplan.business.orders.entities.OrderLineGroup;
import org.libreplan.business.orders.entities.SumChargedEffort;
import org.libreplan.business.qualityforms.daos.IQualityFormDAO;
import org.libreplan.business.qualityforms.entities.QualityForm;
import org.libreplan.business.qualityforms.entities.TaskQualityForm;
import org.libreplan.business.requirements.entities.DirectCriterionRequirement;
import org.libreplan.business.resources.daos.ICriterionDAO;
import org.libreplan.business.resources.daos.ICriterionTypeDAO;
import org.libreplan.business.resources.entities.Criterion;
import org.libreplan.business.resources.entities.CriterionType;
import org.libreplan.business.scenarios.IScenarioManager;
import org.libreplan.business.scenarios.bootstrap.IScenariosBootstrap;
import org.libreplan.business.scenarios.entities.OrderVersion;
import org.libreplan.business.test.calendars.entities.BaseCalendarTest;
import org.libreplan.business.test.orders.entities.OrderElementTest;
import org.libreplan.business.test.planner.daos.ResourceAllocationDAOTest;
import org.libreplan.business.workingday.EffortDuration;
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 })
/*
* @author Diego Pino García <dpino@igalia.com>
*/
public class OrderElementDAOTest {
@Resource
private IDataBootstrap defaultAdvanceTypesBootstrapListener;
@Before
public void loadRequiredaData() {
defaultAdvanceTypesBootstrapListener.loadRequiredData();
}
@Autowired
IQualityFormDAO qualityFormDAO;
@Autowired
private IOrderElementDAO orderElementDAO;
@Autowired
private IBaseCalendarDAO calendarDAO;
@Autowired
private IScenariosBootstrap scenariosBootstrap;
@Autowired
private IScenarioManager scenarioManager;
@Autowired
private ILabelDAO labelDAO;
@Autowired
private ILabelTypeDAO labelTypeDAO;
@Autowired
private ICriterionDAO criterionDAO;
@Autowired
private ICriterionTypeDAO criterionTypeDAO;
@Before
public void loadRequiredData() {
scenariosBootstrap.loadRequiredData();
}
@Test
@Transactional
public void testInSpringContainer() {
assertNotNull(orderElementDAO);
}
private OrderLine createValidOrderLine() {
String unique = UUID.randomUUID().toString();
return createValidOrderLine(unique, unique);
}
private OrderLine createValidOrderLine(String name, String code) {
Order order = createValidOrder();
OrderLine orderLine = createStandAloneLine(name, code);
order.add(orderLine);
return orderLine;
}
private OrderLine createStandAloneLine() {
String uniqueCode = UUID.randomUUID().toString();
return createStandAloneLine(uniqueCode, uniqueCode);
}
private OrderLine createStandAloneLine(String name, String code) {
OrderLine orderLine = OrderLine.create();
orderLine.setName(name);
orderLine.setCode(code);
return orderLine;
}
private OrderLineGroup createValidOrderLineGroup() {
String unique = UUID.randomUUID().toString();
OrderLineGroup result = createValidOrderLineGroup(unique, unique);
return result;
}
private OrderLineGroup createValidOrderLineGroup(String name, String code) {
Order order = createValidOrder();
OrderLineGroup orderLineGroup = OrderLineGroup.create();
orderLineGroup.setName(name);
orderLineGroup.setCode(code);
order.add(orderLineGroup);
OrderLine line = OrderLine.createOrderLineWithUnfixedPercentage(10);
orderLineGroup.add(line);
line.setName(UUID.randomUUID().toString());
line.setCode(UUID.randomUUID().toString());
return orderLineGroup;
}
private Order createValidOrder() {
Order order = Order.create();
order.setName(UUID.randomUUID().toString());
order.setCode(UUID.randomUUID().toString());
order.setInitDate(new Date());
BaseCalendar basicCalendar = BaseCalendarTest.createBasicCalendar();
calendarDAO.save(basicCalendar);
order.setCalendar(basicCalendar);
OrderVersion orderVersion = ResourceAllocationDAOTest
.setupVersionUsing(scenarioManager, order);
orderElementDAO.save(order);
orderElementDAO.flush();
order.useSchedulingDataFor(orderVersion);
return order;
}
@Test
@Transactional
public void testSaveOrderLine() {
OrderLine orderLine = createValidOrderLine();
orderElementDAO.save(orderLine);
assertTrue(orderElementDAO.exists(orderLine.getId()));
}
@Test
@Transactional
public void testFindUniqueByCode() throws InstanceNotFoundException {
OrderLine orderLine = createValidOrderLine();
orderElementDAO.save(orderLine);
orderLine.setCode(((Long) orderLine.getId()).toString());
orderElementDAO.save(orderLine);
OrderLine found = (OrderLine) orderElementDAO
.findUniqueByCode(orderLine
.getCode());
assertTrue(found != null && found.getCode().equals(orderLine.getCode()));
}
@Test
@Transactional
public void testFindUniqueByCodeAndOrderLineGroup()
throws InstanceNotFoundException {
// Create OrderLineGroupLine
OrderLineGroup orderLineGroup = createValidOrderLineGroup();
orderElementDAO.save(orderLineGroup.getOrder());
orderElementDAO.flush();
orderLineGroup.setCode(((Long) orderLineGroup.getId()).toString());
orderElementDAO.save(orderLineGroup.getOrder());
// Create OrderLineGroup
OrderLine orderLine = createStandAloneLine();
orderElementDAO.save(orderLine);
orderLine.setCode(((Long) orderLine.getId()).toString());
orderLineGroup.add(orderLine);
OrderLine found = (OrderLine) orderElementDAO
.findUniqueByCodeAndParent(
orderLineGroup, orderLine.getCode());
assertTrue(found != null && found.getCode().equals(orderLine.getCode()));
}
@Test
@Transactional
public void testFindByCodeInRoot() throws InstanceNotFoundException {
// Create OrderLineGroupLine
Order order = createValidOrder();
List<OrderElement> list = orderElementDAO.findByCodeAndParent(null,
order.getCode());
assertFalse(list.isEmpty());
}
@Test
@Transactional
public void testSaveOrderLineWithAdvanceAssignments()
throws DuplicateValueTrueReportGlobalAdvanceException,
DuplicateAdvanceAssignmentForOrderElementException,
InstanceNotFoundException {
OrderLine orderLine = createValidOrderLine();
OrderElementTest.addAvanceAssignmentWithMeasurement(orderLine,
PredefinedAdvancedTypes.UNITS.getType(), new BigDecimal(1000),
new BigDecimal(400), true);
orderElementDAO.save(orderLine);
orderElementDAO.flush();
assertTrue(orderElementDAO.exists(orderLine.getId()));
OrderLine found = (OrderLine) orderElementDAO.find(orderLine.getId());
Set<DirectAdvanceAssignment> directAdvanceAssignments = found.getDirectAdvanceAssignments();
assertThat(directAdvanceAssignments.size(), equalTo(1));
SortedSet<AdvanceMeasurement> advanceMeasurements = directAdvanceAssignments
.iterator().next().getAdvanceMeasurements();
assertThat(advanceMeasurements.size(), equalTo(1));
assertThat(advanceMeasurements.iterator().next().getValue(),
equalTo(new BigDecimal(400)));
}
@Test
@Transactional
public void testSaveOrderLineGroupWithAdvanceAssignments()
throws DuplicateValueTrueReportGlobalAdvanceException,
DuplicateAdvanceAssignmentForOrderElementException,
InstanceNotFoundException {
Order order = createValidOrder();
OrderVersion orderVersion = order.getOrderVersionFor(scenarioManager
.getCurrent());
OrderElement orderElement = OrderElementTest
.givenOrderLineGroupWithTwoOrderLines(orderVersion,
2000, 3000);
order.add(orderElement);
List<OrderElement> children = orderElement.getChildren();
AdvanceType advanceType = PredefinedAdvancedTypes.UNITS.getType();
OrderElementTest.addAvanceAssignmentWithMeasurement(children.get(0),
advanceType,
new BigDecimal(1000), new BigDecimal(100), true);
OrderElementTest.addAvanceAssignmentWithMeasurement(children.get(1),
advanceType,
new BigDecimal(1000), new BigDecimal(300), true);
orderElementDAO.save(orderElement);
orderElementDAO.flush();
assertTrue(orderElementDAO.exists(orderElement.getId()));
OrderLineGroup found = (OrderLineGroup) orderElementDAO
.find(orderElement.getId());
assertThat(found.getDirectAdvanceAssignments().size(), equalTo(0));
assertThat(found.getIndirectAdvanceAssignments().size(), equalTo(2));
Set<DirectAdvanceAssignment> directAdvanceAssignments = found.getChildren().get(0).getDirectAdvanceAssignments();
assertThat(directAdvanceAssignments.size(), equalTo(1));
DirectAdvanceAssignment directAdvanceAssignment = directAdvanceAssignments
.iterator().next();
assertThat(directAdvanceAssignment.getMaxValue(),
equalTo(new BigDecimal(1000)));
SortedSet<AdvanceMeasurement> advanceMeasurements = directAdvanceAssignment.getAdvanceMeasurements();
assertThat(advanceMeasurements.size(), equalTo(1));
assertThat(advanceMeasurements.iterator().next().getValue(),
equalTo(new BigDecimal(100)));
}
@Test
@Transactional
public void testRemoveOrderLineWithAdvanceAssignments()
throws DuplicateValueTrueReportGlobalAdvanceException,
DuplicateAdvanceAssignmentForOrderElementException,
InstanceNotFoundException {
OrderLine orderLine = createValidOrderLine();
OrderElementTest.addAvanceAssignmentWithMeasurement(orderLine,
PredefinedAdvancedTypes.UNITS.getType(), new BigDecimal(1000),
new BigDecimal(400), true);
orderElementDAO.save(orderLine);
orderElementDAO.flush();
Long id = orderLine.getId();
OrderLine found = (OrderLine) orderElementDAO.find(id);
assertNotNull(found);
orderElementDAO.remove(orderLine.getOrder().getId());
orderElementDAO.flush();
try {
orderElementDAO.find(id);
fail("It should throw an exception");
} catch (InstanceNotFoundException e) {
// ok
}
}
@Test
@Transactional
public void testRemoveOrderLineGroupWithAdvanceAssignments()
throws DuplicateValueTrueReportGlobalAdvanceException,
DuplicateAdvanceAssignmentForOrderElementException,
InstanceNotFoundException {
Order order = createValidOrder();
OrderVersion orderVersion = order.getOrderVersionFor(scenarioManager
.getCurrent());
OrderElement orderElement = OrderElementTest
.givenOrderLineGroupWithTwoOrderLines(orderVersion, 2000, 3000);
order.add(orderElement);
order.useSchedulingDataFor(order.getOrderVersionFor(scenarioManager
.getCurrent()));
List<OrderElement> children = orderElement.getChildren();
AdvanceType advanceType = PredefinedAdvancedTypes.UNITS.getType();
OrderElementTest.addAvanceAssignmentWithMeasurement(children.get(0),
advanceType, new BigDecimal(1000), new BigDecimal(100), true);
OrderElementTest.addAvanceAssignmentWithMeasurement(children.get(1),
advanceType, new BigDecimal(1000), new BigDecimal(300), true);
orderElementDAO.save(orderElement);
orderElementDAO.flush();
Long id = orderElement.getId();
OrderLineGroup found = (OrderLineGroup) orderElementDAO.find(id);
assertNotNull(found);
orderElementDAO.remove(id);
orderElementDAO.flush();
try {
found = (OrderLineGroup) orderElementDAO.find(id);
fail("It should throw an exception");
} catch (InstanceNotFoundException e) {
// ok
}
}
@Test
@Transactional
public void testSaveAndRemoveTaskQualityForm() {
OrderElement orderElement = OrderElementTest
.givenOrderLineGroupWithTwoOrderLines(2000, 3000);
QualityForm qualityForm = QualityForm.create(UUID.randomUUID()
.toString(), UUID.randomUUID().toString());
TaskQualityForm taskQualityForm = orderElement
.addTaskQualityForm(qualityForm);
assertThat(orderElement.getTaskQualityForms().size(), equalTo(1));
orderElement.removeTaskQualityForm(taskQualityForm);
assertThat(orderElement.getTaskQualityForms().size(), equalTo(0));
}
@Test
@Transactional
public void testCheckUniqueQualityForm() {
OrderElement orderElement = OrderElementTest
.givenOrderLineGroupWithTwoOrderLines(2000, 3000);
QualityForm qualityForm = QualityForm.create(UUID.randomUUID()
.toString(), UUID.randomUUID().toString());
orderElement.addTaskQualityForm(qualityForm);
assertThat(orderElement.getTaskQualityForms().size(), equalTo(1));
try {
orderElement.addTaskQualityForm(null);
fail("It should throw an exception");
} catch (IllegalArgumentException e) {
// ok
}
try {
orderElement.addTaskQualityForm(qualityForm);
fail("It should throw an exception");
} catch (ValidationException e) {
// ok
}
}
@Test
@Transactional
public void testSumChargedHoursRelation() throws InstanceNotFoundException {
OrderLine orderLine = createValidOrderLine();
orderLine.setSumChargedEffort(SumChargedEffort.create(orderLine));
orderLine.getSumChargedEffort().addDirectChargedEffort(
EffortDuration.hours(8));
orderLine.getSumChargedEffort().addIndirectChargedEffort(
EffortDuration.hours(10));
orderElementDAO.save(orderLine);
OrderElement orderLineCopy = orderElementDAO.find(orderLine.getId());
assertEquals(orderLine.getSumChargedEffort().getId(),
orderLineCopy.getSumChargedEffort().getId());
assertEquals(orderLineCopy.getSumChargedEffort()
.getTotalChargedEffort(), EffortDuration.hours(18));
}
private Label givenStoredLabel() {
LabelType labelType = LabelType.create(UUID.randomUUID().toString());
labelTypeDAO.save(labelType);
Label label = Label.create(UUID.randomUUID().toString());
label.setType(labelType);
labelDAO.save(label);
return label;
}
@Test
@Transactional
public void testFindByLabels1() throws InstanceNotFoundException {
Label label = givenStoredLabel();
OrderLine orderLine1 = createValidOrderLine();
orderLine1.addLabel(label);
orderElementDAO.save(orderLine1);
OrderLine orderLine2 = createValidOrderLine();
orderElementDAO.save(orderLine2);
OrderLine orderLine3 = createValidOrderLine();
orderElementDAO.save(orderLine3);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(Collections.singleton(label), null);
assertEquals(1, orderELements.size());
assertEquals(orderLine1.getId(), orderELements.get(0).getId());
}
@Test
@Transactional
public void testFindByLabels2() throws InstanceNotFoundException {
Label label1 = givenStoredLabel();
Label label2 = givenStoredLabel();
OrderLine orderLine1 = createValidOrderLine();
orderLine1.addLabel(label1);
orderLine1.addLabel(label2);
orderElementDAO.save(orderLine1);
OrderLine orderLine2 = createValidOrderLine();
orderElementDAO.save(orderLine2);
OrderLine orderLine3 = createValidOrderLine();
orderLine3.addLabel(label2);
orderElementDAO.save(orderLine3);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(Collections.singleton(label1), null);
assertEquals(1, orderELements.size());
assertEquals(orderLine1.getId(), orderELements.get(0).getId());
orderELements = orderElementDAO.findByLabelsAndCriteria(
Collections.singleton(label2), null);
assertEquals(2, orderELements.size());
for (OrderElement each : orderELements) {
assertTrue(each.getId().equals(orderLine1.getId())
|| each.getId().equals(orderLine3.getId()));
}
orderELements = orderElementDAO.findByLabelsAndCriteria(
new HashSet<Label>(Arrays.asList(label1, label2)), null);
assertEquals(1, orderELements.size());
assertEquals(orderLine1.getId(), orderELements.get(0).getId());
}
@Test
@Transactional
public void testFindByLabelsOnTree() throws InstanceNotFoundException {
Label label1 = givenStoredLabel();
Label label2 = givenStoredLabel();
OrderLineGroup orderLineGroup = createValidOrderLineGroup();
orderLineGroup.addLabel(label1);
OrderElement child = orderLineGroup.getChildren().get(0);
child.addLabel(label2);
orderElementDAO.save(orderLineGroup);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(Collections.singleton(label1), null);
assertEquals(1, orderELements.size());
assertEquals(orderLineGroup.getId(), orderELements.get(0).getId());
orderELements = orderElementDAO.findByLabelsAndCriteria(
Collections.singleton(label2), null);
assertEquals(1, orderELements.size());
assertEquals(child.getId(), orderELements.get(0).getId());
orderELements = orderElementDAO.findByLabelsAndCriteria(
new HashSet<Label>(Arrays.asList(label1, label2)), null);
assertEquals(0, orderELements.size());
}
private Criterion givenStoredCriterion() {
String criterionTypeCode = UUID.randomUUID().toString();
CriterionType criterionType = CriterionType.create(criterionTypeCode);
criterionType.setName(criterionTypeCode);
criterionTypeDAO.save(criterionType);
String criterionCode = UUID.randomUUID().toString();
Criterion criterion = Criterion.create(criterionCode, criterionType);
criterion.setCode(criterionCode);
criterionDAO.save(criterion);
return criterion;
}
@Test
@Transactional
public void testFindByCriteria1() throws InstanceNotFoundException {
Criterion criterion = givenStoredCriterion();
OrderLine orderLine1 = createValidOrderLine();
orderLine1.addCriterionRequirement(new DirectCriterionRequirement(
criterion));
orderElementDAO.save(orderLine1);
OrderLine orderLine2 = createValidOrderLine();
orderElementDAO.save(orderLine2);
OrderLine orderLine3 = createValidOrderLine();
orderElementDAO.save(orderLine3);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(null, Collections.singleton(criterion));
assertEquals(1, orderELements.size());
assertEquals(orderLine1.getId(), orderELements.get(0).getId());
}
@Test
@Transactional
public void testFindByCriteria2() throws InstanceNotFoundException {
Criterion criterion1 = givenStoredCriterion();
Criterion criterion2 = givenStoredCriterion();
OrderLine orderLine1 = createValidOrderLine();
orderLine1.addCriterionRequirement(new DirectCriterionRequirement(
criterion1));
orderLine1.addCriterionRequirement(new DirectCriterionRequirement(
criterion2));
orderElementDAO.save(orderLine1);
OrderLine orderLine2 = createValidOrderLine();
orderElementDAO.save(orderLine2);
OrderLine orderLine3 = createValidOrderLine();
orderLine3.addCriterionRequirement(new DirectCriterionRequirement(
criterion2));
orderElementDAO.save(orderLine3);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(null, Collections.singleton(criterion1));
assertEquals(1, orderELements.size());
assertEquals(orderLine1.getId(), orderELements.get(0).getId());
orderELements = orderElementDAO.findByLabelsAndCriteria(null,
Collections.singleton(criterion2));
assertEquals(2, orderELements.size());
for (OrderElement each : orderELements) {
assertTrue(each.getId().equals(orderLine1.getId())
|| each.getId().equals(orderLine3.getId()));
}
orderELements = orderElementDAO.findByLabelsAndCriteria(null,
new HashSet<Criterion>(Arrays.asList(criterion1, criterion2)));
assertEquals(1, orderELements.size());
assertEquals(orderLine1.getId(), orderELements.get(0).getId());
}
@Test
@Transactional
public void testFindByCriteriaOnTree() throws InstanceNotFoundException {
Criterion criterion1 = givenStoredCriterion();
Criterion criterion2 = givenStoredCriterion();
OrderLineGroup orderLineGroup = createValidOrderLineGroup();
orderLineGroup.addCriterionRequirement(new DirectCriterionRequirement(
criterion1));
OrderElement child = orderLineGroup.getChildren().get(0);
child.addCriterionRequirement(new DirectCriterionRequirement(criterion2));
orderElementDAO.save(orderLineGroup);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(null,
Collections.singleton(criterion1));
assertEquals(1, orderELements.size());
assertEquals(orderLineGroup.getId(), orderELements.get(0).getId());
orderELements = orderElementDAO.findByLabelsAndCriteria(null,
Collections.singleton(criterion2));
assertEquals(1, orderELements.size());
assertEquals(child.getId(), orderELements.get(0).getId());
orderELements = orderElementDAO.findByLabelsAndCriteria(null,
new HashSet<Criterion>(Arrays.asList(criterion1, criterion2)));
assertEquals(0, orderELements.size());
}
@Test
@Transactional
public void testFindByLabelsAndCriteria1() throws InstanceNotFoundException {
Label label = givenStoredLabel();
Criterion criterion = givenStoredCriterion();
OrderLine orderLine1 = createValidOrderLine();
orderLine1.addLabel(label);
orderElementDAO.save(orderLine1);
OrderLine orderLine2 = createValidOrderLine();
orderLine2.addCriterionRequirement(new DirectCriterionRequirement(
criterion));
orderElementDAO.save(orderLine2);
OrderLine orderLine3 = createValidOrderLine();
orderLine3.addLabel(label);
orderLine3.addCriterionRequirement(new DirectCriterionRequirement(
criterion));
orderElementDAO.save(orderLine3);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(Collections.singleton(label),
Collections.singleton(criterion));
assertEquals(1, orderELements.size());
assertEquals(orderLine3.getId(), orderELements.get(0).getId());
}
@Test
@Transactional
public void testFindByLabelsAndCriteria2() throws InstanceNotFoundException {
Label label1 = givenStoredLabel();
Label label2 = givenStoredLabel();
Criterion criterion1 = givenStoredCriterion();
Criterion criterion2 = givenStoredCriterion();
OrderLine orderLine1 = createValidOrderLine();
orderLine1.addLabel(label1);
orderLine1.addCriterionRequirement(new DirectCriterionRequirement(
criterion2));
orderElementDAO.save(orderLine1);
OrderLine orderLine2 = createValidOrderLine();
orderLine2.addLabel(label2);
orderLine2.addCriterionRequirement(new DirectCriterionRequirement(
criterion1));
orderElementDAO.save(orderLine2);
OrderLine orderLine3 = createValidOrderLine();
orderLine3.addLabel(label1);
orderLine3.addLabel(label2);
orderLine3.addCriterionRequirement(new DirectCriterionRequirement(
criterion1));
orderLine3.addCriterionRequirement(new DirectCriterionRequirement(
criterion2));
orderElementDAO.save(orderLine3);
List<OrderElement> orderELements = orderElementDAO
.findByLabelsAndCriteria(Collections.singleton(label2),
Collections.singleton(criterion1));
assertEquals(2, orderELements.size());
for (OrderElement each : orderELements) {
assertTrue(each.getId().equals(orderLine2.getId())
|| each.getId().equals(orderLine3.getId()));
}
orderELements = orderElementDAO
.findByLabelsAndCriteria(Collections.singleton(label1),
Collections.singleton(criterion2));
assertEquals(2, orderELements.size());
for (OrderElement each : orderELements) {
assertTrue(each.getId().equals(orderLine1.getId())
|| each.getId().equals(orderLine3.getId()));
}
orderELements = orderElementDAO
.findByLabelsAndCriteria(
new HashSet<Label>(Arrays.asList(label1, label2)),
new HashSet<Criterion>(Arrays.asList(criterion1,
criterion2)));
assertEquals(1, orderELements.size());
assertEquals(orderLine3.getId(), orderELements.get(0).getId());
}
}