diff --git a/libreplan-webapp/src/test/java/org/libreplan/web/calendars/BaseCalendarModelTest.java b/libreplan-webapp/src/test/java/org/libreplan/web/calendars/BaseCalendarModelTest.java index 9dd6b3776..4ee47b93b 100644 --- a/libreplan-webapp/src/test/java/org/libreplan/web/calendars/BaseCalendarModelTest.java +++ b/libreplan-webapp/src/test/java/org/libreplan/web/calendars/BaseCalendarModelTest.java @@ -22,6 +22,7 @@ package org.libreplan.web.calendars; import static org.hamcrest.CoreMatchers.equalTo; +import static org.hamcrest.CoreMatchers.notNullValue; import static org.junit.Assert.assertThat; import static org.libreplan.business.BusinessGlobalNames.BUSINESS_SPRING_CONFIG_FILE; import static org.libreplan.business.workingday.EffortDuration.hours; @@ -31,6 +32,8 @@ import static org.libreplan.web.test.WebappGlobalNames.WEBAPP_SPRING_CONFIG_TEST import static org.libreplan.web.test.WebappGlobalNames.WEBAPP_SPRING_SECURITY_CONFIG_TEST_FILE; import java.util.List; +import java.util.Map; +import java.util.Set; import java.util.UUID; import javax.annotation.Resource; @@ -44,6 +47,7 @@ import org.libreplan.business.calendars.daos.IBaseCalendarDAO; import org.libreplan.business.calendars.entities.BaseCalendar; import org.libreplan.business.calendars.entities.CalendarData.Days; import org.libreplan.business.calendars.entities.Capacity; +import org.libreplan.business.common.BaseEntity; import org.libreplan.business.common.IAdHocTransactionService; import org.libreplan.business.common.IOnTransaction; import org.libreplan.business.common.exceptions.ValidationException; @@ -102,12 +106,13 @@ public class BaseCalendarModelTest { Capacity capacity = Capacity.create(hours(8)); setCapacity(baseCalendar, capacity); baseCalendarModel.confirmSave(); - List currentCalendars = baseCalendarModel - .getBaseCalendars(); + Map> currentCalendars = BaseEntity + .byId(baseCalendarModel.getBaseCalendars()); assertThat(currentCalendars.size(), equalTo(previous + 1)); - BaseCalendar createdCalendar = currentCalendars.get(previous); - assertThat(createdCalendar.getId(), equalTo(baseCalendar.getId())); + BaseCalendar createdCalendar = currentCalendars + .get(baseCalendar.getId()).iterator().next(); + assertThat(createdCalendar, notNullValue()); assertThat(createdCalendar.getCapacityConsideringCalendarDatasOn( new LocalDate(), Days.MONDAY), equalTo(capacity)); }