From c362b166918f2bbc36556b98a9d232ef7919f16c Mon Sep 17 00:00:00 2001 From: Nacho Barrientos Date: Wed, 2 Nov 2011 17:18:35 +0100 Subject: [PATCH] Turn ResourceCalendar mock creator static to reuse it from another test. FEA: ItEr75S27PerProjectDashboard --- .../SpecificResourceAllocationTest.java | 38 +++++++++++-------- .../test/planner/entities/TaskTest.java | 35 +---------------- 2 files changed, 24 insertions(+), 49 deletions(-) diff --git a/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/SpecificResourceAllocationTest.java b/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/SpecificResourceAllocationTest.java index 26900041e..8238f8776 100644 --- a/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/SpecificResourceAllocationTest.java +++ b/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/SpecificResourceAllocationTest.java @@ -79,24 +79,30 @@ public class SpecificResourceAllocationTest { private ResourceCalendar calendar; private void givenResourceCalendarAlwaysReturning(final int hours) { - this.calendar = createNiceMock(ResourceCalendar.class); - expect(this.calendar.getCapacityOn(isA(PartialDay.class))) - .andReturn(EffortDuration.hours(hours)).anyTimes(); - IAnswer asDurationAnswer = asDurationOnAnswer(hours(hours)); - expect( - this.calendar.asDurationOn(isA(PartialDay.class), - isA(ResourcesPerDay.class))) - .andAnswer(asDurationAnswer).anyTimes(); - expect(this.calendar.getCapacityWithOvertime(isA(LocalDate.class))) - .andReturn( - Capacity.create(hours(hours)) - .overAssignableWithoutLimit()).anyTimes(); - expect(this.calendar.getAvailability()).andReturn( - AvailabilityTimeLine.allValid()).anyTimes(); - replay(this.calendar); + this.calendar = createResourceCalendarAlwaysReturning(hours); } - private IAnswer asDurationOnAnswer( + public static ResourceCalendar createResourceCalendarAlwaysReturning(final int hours) { + ResourceCalendar workerCalendar; + workerCalendar = createNiceMock(ResourceCalendar.class); + expect(workerCalendar.getCapacityOn(isA(PartialDay.class))) + .andReturn(EffortDuration.hours(hours)).anyTimes(); + IAnswer asDurationAnswer = asDurationOnAnswer(hours(hours)); + expect( + workerCalendar.asDurationOn(isA(PartialDay.class), + isA(ResourcesPerDay.class))) + .andAnswer(asDurationAnswer).anyTimes(); + expect(workerCalendar.getCapacityWithOvertime(isA(LocalDate.class))) + .andReturn( + Capacity.create(hours(hours)) + .overAssignableWithoutLimit()).anyTimes(); + expect(workerCalendar.getAvailability()).andReturn( + AvailabilityTimeLine.allValid()).anyTimes(); + replay(workerCalendar); + return workerCalendar; + } + + private static IAnswer asDurationOnAnswer( final EffortDuration duration) { return new IAnswer() { diff --git a/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/TaskTest.java b/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/TaskTest.java index 59cb63ab1..ac3696c51 100644 --- a/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/TaskTest.java +++ b/libreplan-business/src/test/java/org/libreplan/business/test/planner/entities/TaskTest.java @@ -43,7 +43,6 @@ import java.util.Date; import javax.annotation.Resource; import org.easymock.IAnswer; -import org.easymock.classextension.EasyMock; import org.joda.time.LocalDate; import org.junit.Before; import org.junit.Test; @@ -51,7 +50,6 @@ import org.junit.runner.RunWith; import org.libreplan.business.IDataBootstrap; import org.libreplan.business.calendars.entities.AvailabilityTimeLine; import org.libreplan.business.calendars.entities.BaseCalendar; -import org.libreplan.business.calendars.entities.Capacity; import org.libreplan.business.calendars.entities.ResourceCalendar; import org.libreplan.business.orders.entities.HoursGroup; import org.libreplan.business.orders.entities.Order; @@ -397,37 +395,8 @@ public class TaskTest { replay(this.worker); } - /* Taken from: SpecificResourceAllocationTest (TODO: Refactor) */ private void givenResourceCalendarAlwaysReturning(final int hours) { - this.workerCalendar = createNiceMock(ResourceCalendar.class); - expect(this.workerCalendar.getCapacityOn(isA(PartialDay.class))) - .andReturn(EffortDuration.hours(hours)).anyTimes(); - IAnswer asDurationAnswer = asDurationOnAnswer(hours(hours)); - expect( - this.workerCalendar.asDurationOn(isA(PartialDay.class), - isA(ResourcesPerDay.class))) - .andAnswer(asDurationAnswer).anyTimes(); - expect(this.workerCalendar.getCapacityWithOvertime(isA(LocalDate.class))) - .andReturn( - Capacity.create(hours(hours)) - .overAssignableWithoutLimit()).anyTimes(); - expect(this.workerCalendar.getAvailability()).andReturn( - AvailabilityTimeLine.allValid()).anyTimes(); - replay(this.workerCalendar); + this.workerCalendar = SpecificResourceAllocationTest. + createResourceCalendarAlwaysReturning(hours); } - - /* Taken from: SpecificResourceAllocationTest (TODO: Refactor) */ - private IAnswer asDurationOnAnswer( - final EffortDuration duration) { - return new IAnswer() { - - @Override - public EffortDuration answer() throws Throwable { - ResourcesPerDay perDay = (ResourcesPerDay) EasyMock - .getCurrentArguments()[1]; - return perDay.asDurationGivenWorkingDayOf(duration); - } - }; - } - } \ No newline at end of file